module geobench;
import tools.time, tools.log, tools.base, std.string, std.socket;

class GeoIP {
  import std.file;
  static {
    struct entry {
      uint start;
      string country;
    }
    entry[] list;
    string lookup(uint what) {
      auto range = list;
      while (range.length > 1) {
        auto center = range[$/2];
        if (center.start == what) return center.country;
        if (center.start < what) range = range[$/2 .. $];
        else range = range[0 .. $/2];
      }
      return range[0].country;
    }
  }
  static this() {
    string[string] countries; // lookup is by string equality
    "GeoIPCountryWhois.csv".read().castLike("").split("\n") /map/ (string line) {
      line = line.strip();
      if (!line.length) return;
      auto parts = line[1 .. $-1].split("\",\""); // remove ""s
      auto ip = parts[0];
      entry e;
      // why parse IPs? Let the system do it for us.
      e.start = (new InternetAddress(ip, 0)).addr();
      e.country = parts[$-1];
      if (auto p = e.country in countries) e.country = *p;
      else countries[e.country] = e.country;
      list ~= e; // yes they're ordered
    };
    logln(list.length, " ranges added, using ", countries.values.length, " countries. ");
  }
}

import tools.mersenne;
void main() {
  long count;
  auto last = sec();
  while (true) {
    string country;
    for (int i = 0; i < 16; ++i) {
      auto rand_ip = rand();
      country = GeoIP.lookup(rand_ip);
      count ++;
    }
    auto s = sec();
    if (s > last + 1) {
      last = s;
      logln(count, " lookups per second, last country was ", country);
      count = 0;
    }
  }
}