add batch test script

2026-02-27 09:44:31 +08:00 · 2025-06-19 15:08:52 +08:00
parent 7a9a53a058
commit 73e4e6047f
1 changed files with 133 additions and 0 deletions
--- a/binding/php/batch_test.php
+++ b/binding/php/batch_test.php
@@ -0,0 +1,133 @@
+<?php
+// Copyright 2022 The Ip2Region Authors. All rights reserved.
+// Use of this source code is governed by a Apache2.0-style
+// license that can be found in the LICENSE file.
+//
+// @Author Lion <chenxin619315@gmail.com>
+// @Date   2022/06/22
+
+require dirname(__FILE__) . '/XdbSearcher.class.php';
+
+function printHelp($argv) {
+    printf("php %s [command options]\n", $argv[0]);
+    printf("options: \n");
+    printf(" --db string             ip2region binary xdb file path\n");
+    printf(" --src string            source ip text file path\n");
+    printf(" --cache-policy string   cache policy: file/vectorIndex/content\n");
+}
+
+if($argc < 2) {
+    printHelp($argv);
+    return;
+}
+
+$dbFile = "";
+$srcFile = "";
+$cachePolicy = 'vectorIndex';
+array_shift($argv);
+foreach ($argv as $r) {
+    if (strlen($r) < 5) {
+        continue;
+    }
+
+    if (strpos($r, '--') != 0) {
+        continue;
+    }
+
+    $sIdx = strpos($r, "=");
+    if ($sIdx < 0) {
+        printf("missing = for args pair %s\n", $r);
+        return;
+    }
+
+    $key = substr($r, 2, $sIdx - 2);
+    $val = substr($r, $sIdx + 1);
+    if ($key == 'db') {
+        $dbFile = $val;
+    } else if ($key == 'src') {
+        $srcFile = $val;
+    } else if ($key == 'cache-policy') {
+        $cachePolicy = $val;
+    } else {
+        printf("undefined option `%s`\n", $r);
+        return;
+    }
+}
+
+if (strlen($dbFile) < 1 || strlen($srcFile) < 1) {
+    printHelp($argv);
+    return;
+}
+
+// printf("debug: dbFile: %s, cachePolicy: %s\n", $dbFile, $cachePolicy);
+// create the xdb searcher by the cache-policy
+switch ( $cachePolicy ) {
+    case 'file':
+        try {
+            $searcher = XdbSearcher::newWithFileOnly($dbFile);
+        } catch (Exception $e) {
+            printf("failed to create searcher with '%s': %s\n", $dbFile, $e);
+            return;
+        }
+        break;
+    case 'vectorIndex':
+        $vIndex = XdbSearcher::loadVectorIndexFromFile($dbFile);
+        if ($vIndex == null) {
+            printf("failed to load vector index from '%s'\n", $dbFile);
+            return;
+        }
+
+        try {
+            $searcher = XdbSearcher::newWithVectorIndex($dbFile, $vIndex);
+        } catch (Exception $e) {
+            printf("failed to create vector index cached searcher with '%s': %s\n", $dbFile, $e);
+            return;
+        }
+        break;
+    case 'content':
+        $cBuff = XdbSearcher::loadContentFromFile($dbFile);
+        if ($cBuff == null) {
+            printf("failed to load xdb content from '%s'\n", $dbFile);
+            return;
+        }
+
+        try {
+            $searcher = XdbSearcher::newWithBuffer($cBuff);
+        } catch (Exception $e) {
+            printf("failed to create content cached searcher: %s", $e);
+            return;
+        }
+        break;
+    default:
+        printf("undefined cache-policy `%s`\n", $cachePolicy);
+        return;
+}
+
+
+// do the bench test
+$handle = fopen($srcFile, "r");
+if ($handle === false) {
+    printf("failed to open source text file `%s`\n", $srcFile);
+    return null;
+}
+
+$count = 0;
+while (!feof($handle)) {
+    $line = trim(fgets($handle, 1024));
+    if (strlen($line) < 1) {
+        continue;
+    }
+
+    $ip = XdbSearcher::ip2long($line);
+    if ($ip === null) {
+        printf("invalid ip `%s`\n", $line);
+        return;
+    }
+
+    $count++;
+    $region = $searcher->search($ip);
+    echo $line, ", ", $region, "\n";
+}
+
+fclose($handle);
+echo "Done, with {$count} IPs\n";