Relax City and Country database checks

The strict database checks broke the per-continent City databases,
and adding all of those to an array is a bit much (and slow).

Closes #23.
This commit is contained in:
Gregory Oschwald 2014-09-12 07:06:12 -07:00
parent 920a92f352
commit 2f9746c32b
2 changed files with 39 additions and 29 deletions

26
bin/benchmark.php Executable file
View File

@ -0,0 +1,26 @@
#!/usr/bin/env php
<?php
require_once '../vendor/autoload.php';
use GeoIp2\Database\Reader;
use GeoIp2\Exception\AddressNotFoundException;
$reader = new Reader('GeoLite2-City.mmdb');
$count = 40000;
$startTime = microtime(true);
for ($i = 0; $i < $count; $i++) {
$ip = long2ip(rand(0, pow(2, 32) -1));
try {
$t = $reader->city($ip);
} catch (AddressNotFoundException $e) {
}
if ($i % 1000 == 0) {
print($i . ' ' . $ip . "\n");
}
}
$endTime = microtime(true);
$duration = $endTime - $startTime;
print('Requests per second: ' . $count / $duration . "\n");

View File

@ -71,11 +71,7 @@ class Reader implements ProviderInterface
*/ */
public function city($ipAddress) public function city($ipAddress)
{ {
return $this->modelFor( return $this->modelFor('City', 'City', $ipAddress);
'City',
array('GeoLite2-City', 'GeoIP2-City'),
$ipAddress
);
} }
/** /**
@ -92,11 +88,7 @@ class Reader implements ProviderInterface
*/ */
public function country($ipAddress) public function country($ipAddress)
{ {
return $this->modelFor( return $this->modelFor('Country', 'Country', $ipAddress);
'Country',
array('GeoLite2-Country', 'GeoIP2-Country'),
$ipAddress
);
} }
public function connectionType($ipAddress) public function connectionType($ipAddress)
@ -126,16 +118,9 @@ class Reader implements ProviderInterface
); );
} }
private function modelFor($class, $types, $ipAddress) private function modelFor($class, $type, $ipAddress)
{ {
if (!in_array($this->metadata()->databaseType, $types)) { $record = $this->getRecord($class, $type, $ipAddress);
$method = lcfirst($class);
throw new \BadMethodCallException(
"The $method method cannot be used to open a "
. $this->metadata()->databaseType . " database"
);
}
$record = $this->getRecord($ipAddress);
$record['traits']['ip_address'] = $ipAddress; $record['traits']['ip_address'] = $ipAddress;
$class = "GeoIp2\\Model\\" . $class; $class = "GeoIp2\\Model\\" . $class;
@ -145,15 +130,7 @@ class Reader implements ProviderInterface
private function flatModelFor($class, $type, $ipAddress) private function flatModelFor($class, $type, $ipAddress)
{ {
if ($this->metadata()->databaseType !== $type) { $record = $this->getRecord($class, $type, $ipAddress);
$method = lcfirst($class);
throw new \BadMethodCallException(
"The $method method cannot be used to open a "
. $this->metadata()->databaseType . " database"
);
}
$record = $this->getRecord($ipAddress);
$record['ip_address'] = $ipAddress; $record['ip_address'] = $ipAddress;
$class = "GeoIp2\\Model\\" . $class; $class = "GeoIp2\\Model\\" . $class;
@ -161,8 +138,15 @@ class Reader implements ProviderInterface
return new $class($record); return new $class($record);
} }
private function getRecord($ipAddress) private function getRecord($class, $type, $ipAddress)
{ {
if (strpos($this->metadata()->databaseType, $type) === false) {
$method = lcfirst($class);
throw new \BadMethodCallException(
"The $method method cannot be used to open a "
. $this->metadata()->databaseType . " database"
);
}
$record = $this->dbReader->get($ipAddress); $record = $this->dbReader->get($ipAddress);
if ($record === null) { if ($record === null) {
throw new AddressNotFoundException( throw new AddressNotFoundException(