xinb
/
ip2region_private
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

search with ip_string for 3-kinds-of cache-policy searcher

This commit is contained in:
lionsoul2014 2025-10-31 11:52:17 +08:00
parent c4113e5eae
commit bf0ded1171
1 changed files with 46 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
binding/python/util_test.py
View File

@ -8,7 +8,8 @@
import os
import sys
import time
from xdb import util
import xdb.util as util
import xdb.searcher as xdb
script_dir = os.path.dirname(__file__)
data_dir = os.path.join(script_dir, '../../data/')
@ -98,6 +99,50 @@ def test_ip_compare():
cmp = util.ip_compare(ip1, ip2)
print("compare({}, {}) -> {} ? {}".format(util.ip_to_string(ip1), util.ip_to_string(ip2), cmp, cmp == ip_pair[2]))
def _get_searcher_list(version: util.Version):
db_path = xdb_v4_path if version.id == util.XdbIPv4Id else xdb_v6_path
return [
["new_with_file_only", lambda: xdb.new_with_file_only(version, db_path)],
["new_with_vector_index", lambda: xdb.new_with_vector_index(version, db_path, util.load_vector_index_from_file(db_path))],
["new_with_buffer", lambda: xdb.new_with_buffer(version, util.load_content_from_file(db_path))]
]
def test_ip_search():
# ipv4 search test
print("---ipv4 search test:")
ip_str = "120.229.45.92"
s_list = _get_searcher_list(util.IPv4)
try:
b_region = None
for meta in s_list:
searcher = meta[1]()
region = searcher.search(ip_str)
if b_region != None:
assert b_region == region, f"region and b_region is not the same"
print(f"{meta[0]}.search({ip_str}): {region}")
# searcher close
searcher.close()
except Exception as e:
print(f"failed to search({ip_str}): {str(e)}")
# ipv6 search test
print("---ipv6 search test:")
ip_str = "240e:3b7:3272:d8d0:db09:c067:8d59:539e"
s_list = _get_searcher_list(util.IPv6)
try:
b_region = None
for meta in s_list:
searcher = meta[1]()
region = searcher.search(ip_str)
if b_region != None:
assert b_region == region, f"region and b_region is not the same"
print(f"{meta[0]}.search({ip_str}): {region}")
# searcher close
searcher.close()
except Exception as e:
print(f"failed to search({ip_str}): {str(e)}")
if __name__ == "__main__":
# check and call the specified function