You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
87 lines
2.4 KiB
Lua
87 lines
2.4 KiB
Lua
--[[
|
|
ip2region lua client test script
|
|
|
|
@author chenxin<chenxin619315@gmail.com>
|
|
]]--
|
|
|
|
-- check the command line arguments
|
|
if ( not arg[1] ) then
|
|
print([[
|
|
Usage: lua testSearcher.lua [ip2region db file] [algorithm]
|
|
+-Optional Algorithm: binary, b-tree, memory]]);
|
|
os.exit();
|
|
end
|
|
|
|
local Ip2region = require "Ip2region";
|
|
-- local cjson = require "cjson";
|
|
-- local socket = require "socket";
|
|
|
|
|
|
-- check and parse the dbFile and the method algorithm
|
|
-- Create a new ip2region object by the new interface
|
|
local ip2region = Ip2region.new(arg[1]);
|
|
|
|
-- reset the dbFile by the follow two ways:
|
|
-- ip2region.dbFile = arg[1];
|
|
-- ip2region:setDbFile(arg[1]);
|
|
|
|
|
|
local algorithm = "btree";
|
|
if ( arg[2] ~= nil ) then
|
|
local arg_2 = string.lower(arg[2]);
|
|
if ( arg_2 ~= "binary" and arg_2 ~= "memory" ) then
|
|
algorithm = "binary";
|
|
elseif ( arg_2 == "memory" ) then
|
|
algorithm = "memory";
|
|
end
|
|
end
|
|
|
|
|
|
|
|
-- local data = searcher:memorySearch("120.79.17.142");
|
|
-- local data = searcher:binarySearch("120.79.17.142");
|
|
-- local data = searcher:btreeSearch("120.79.17.142");
|
|
print("initializing " .. algorithm ..[[
|
|
+----------------------------------+
|
|
| ip2region test script |
|
|
| Author: chenxin619315@gmail.com |
|
|
| Type 'quit' to exit program |
|
|
+----------------------------------+]]
|
|
);
|
|
|
|
while ( true ) do
|
|
io.write("ip2region>> ");
|
|
io.input(io.stdin);
|
|
local line = io.read();
|
|
if ( line == nil ) then
|
|
-- do nothing
|
|
break;
|
|
elseif ( line == "quit" ) then
|
|
break;
|
|
elseif ( Ip2region.ip2long(line) == nil ) then
|
|
print("Invalid ip address=", line);
|
|
else
|
|
local data;
|
|
local s_time = os.clock();
|
|
if ( algorithm == "btree" ) then
|
|
data = ip2region:btreeSearch(line);
|
|
elseif ( algorithm == "binary" ) then
|
|
data = ip2region:binarySearch(line);
|
|
elseif ( algorithm == "memory" ) then
|
|
data = ip2region:memorySearch(line);
|
|
end
|
|
|
|
local cost_time = (os.clock() - s_time) * 1000; -- to millseconds
|
|
if ( data == nil ) then
|
|
io.write("Failed for ip=", line, " is it a valid ip address ?");
|
|
else
|
|
io.write(string.format("%u|%s in %5f millseconds\n", data.city_id, data.region, cost_time));
|
|
end
|
|
end
|
|
end
|
|
|
|
-- close the object
|
|
-- Also the lua gc will invoke it automatically
|
|
-- print(ip2region);
|
|
ip2region:close();
|