{"id":201444,"date":"2025-05-10T10:59:58","date_gmt":"2025-05-10T02:59:58","guid":{"rendered":"https:\/\/server.hk\/cnblog\/201444\/"},"modified":"2025-05-10T10:59:58","modified_gmt":"2025-05-10T02:59:58","slug":"%e4%bd%bf%e7%94%a8redis%e6%9c%89%e5%ba%8f%e9%9b%86%e5%90%88%e5%ae%9e%e7%8e%b0ip%e5%bd%92%e5%b1%9e%e5%9c%b0%e6%9f%a5%e8%af%a2%e8%af%a6%e8%a7%a3","status":"publish","type":"post","link":"https:\/\/server.hk\/cnblog\/201444\/","title":{"rendered":"\u4f7f\u7528Redis\u6709\u5e8f\u96c6\u5408\u5b9e\u73b0IP\u5f52\u5c5e\u5730\u67e5\u8be2\u8be6\u89e3"},"content":{"rendered":"

<\/b> <\/p>\n

\u4f7f\u7528Redis\u6709\u5e8f\u96c6\u5408\u5b9e\u73b0IP\u5f52\u5c5e\u5730\u67e5\u8be2\u8be6\u89e3<\/h1>\n

<\/i>\u6536\u85cf<\/span> <\/p>\n

\u4eb2\u7231\u7684\u7f16\u7a0b\u5b66\u4e60\u7231\u597d\u8005\uff0c\u5982\u679c\u4f60\u70b9\u5f00\u4e86\u8fd9\u7bc7\u6587\u7ae0\uff0c\u8bf4\u660e\u4f60\u5bf9\u300a\u4f7f\u7528Redis\u6709\u5e8f\u96c6\u5408\u5b9e\u73b0IP\u5f52\u5c5e\u5730\u67e5\u8be2\u8be6\u89e3\u300b\u5f88\u611f\u5174\u8da3\u3002\u672c\u7bc7\u6587\u7ae0\u5c31\u6765\u7ed9\u5927\u5bb6\u8be6\u7ec6\u89e3\u6790\u4e00\u4e0b\uff0c\u4e3b\u8981\u4ecb\u7ecd\u4e00\u4e0bIP\u3001Redis\u6709\u5e8f\u96c6\u5408\uff0c\u5e0c\u671b\u6240\u6709\u8ba4\u771f\u8bfb\u5b8c\u7684\u7ae5\u978b\u4eec\uff0c\u90fd\u6709\u5b9e\u8d28\u6027\u7684\u63d0\u9ad8\u3002<\/p>\n

\u5de5\u4f5c\u4e2d\u7ecf\u5e38\u9047\u5230\u4e00\u7c7b\u9700\u6c42\uff0c\u6839\u636e IP \u5730\u5740\u6bb5\u6765\u67e5\u627e IP \u5bf9\u5e94\u7684\u5f52\u5c5e\u5730\u4fe1\u606f\u3002\u5982\u679c\u628a\u67e5\u8be2\u8fc7\u7a0b\u653e\u5230\u5173\u7cfb\u578b\u6570\u636e\u5e93\u4e2d\uff0c\u4f1a\u5e26\u6765\u5f88\u5927\u7684 IO \u6d88\u8017\uff0c\u901f\u5ea6\u4e5f\u4e0d\u80fd\u6ee1\u8db3\uff0c\u663e\u7136\u662f\u4e0d\u5408\u9002\u7684\u3002<\/p>\n

\u90a3\u6709\u54ea\u4e9b\u66f4\u597d\u7684\u529e\u6cd5\u5462\uff1f\u4e3a\u6b64\u505a\u4e86\u4e00\u4e9b\u5c1d\u8bd5\uff0c\u4e0b\u9762\u6765\u8be6\u7ec6\u8bf4\u660e\u3002<\/p>\n

\u6784\u5efa\u7d22\u5f15\u6587\u4ef6<\/strong><\/span><\/p>\n

\u5728 GitHub \u4e0a\u770b\u5230\u4e00\u4e2a \u9879\u76ee\uff0c\u4f5c\u8005\u901a\u8fc7\u751f\u6210\u4e00\u4e2a\u5305\u542b\u6709\u4e8c\u7ea7\u7d22\u5f15\u7684\u6587\u4ef6\u6765\u5b9e\u73b0\u5feb\u901f\u67e5\u8be2\uff0c\u67e5\u8be2\u901f\u5ea6\u8db3\u591f\u5feb\uff0c\u6beb\u79d2\u7ea7\u522b\u3002\u4f46\u5982\u679c\u60f3\u66f4\u65b0\u5730\u5740\u6bb5\u6216\u5f52\u5c5e\u5730\u4fe1\u606f\uff0c\u6bcf\u6b21\u90fd\u8981\u91cd\u65b0\u751f\u6210\u6587\u4ef6\uff0c\u5e76\u4e0d\u662f\u5f88\u65b9\u4fbf\u3002
\u4e0d\u8fc7\u8fd8\u662f\u63a8\u8350\u5927\u5bb6\u770b\u770b\u8fd9\u4e2a\u9879\u76ee\uff0c\u5176\u4e2d\u5efa\u7d22\u5f15\u7684\u601d\u60f3\u8fd8\u662f\u5f88\u503c\u5f97\u5b66\u4e60\u7684\u3002\u4f5c\u8005\u7684\u5f00\u6e90\u9879\u76ee\u4e2d\u53ea\u6709\u67e5\u8be2\u7684\u76f8\u5173\u4ee3\u7801\uff0c\u5e76\u6ca1\u6709\u751f\u6210\u7d22\u5f15\u6587\u4ef6\u7684\u4ee3\u7801\uff0c\u6211\u4f9d\u7167\u539f\u7406\u56fe\u5199\u4e86\u4e00\u6bb5\u751f\u6210\u7d22\u5f15\u6587\u4ef6\u7684\u4ee3\u7801\uff0c\u5982\u4e0b\uff1a<\/p>\n

\n# -*- coding:utf-8 -*-\n\n\nimport time\nimport socket\nimport struct\n\nIP_REGION_FILE = '.\/data\/ip_to_region.db'\n\nSUPER_BLOCK_LENGTH = 8\nINDEX_BLOCK_LENGTH = 12\nHEADER_INDEX_LENGTH = 8192\n\n\ndef generate_db_file():\n  pointer = SUPER_BLOCK_LENGTH + HEADER_INDEX_LENGTH\n\n  region, index = '', ''\n\n  # \u6587\u4ef6\u683c\u5f0f\n  # 1.0.0.0|1.0.0.255|\u6fb3\u5927\u5229\u4e9a|0|0|0|0\n  # 1.0.1.0|1.0.3.255|\u4e2d\u56fd|0|\u798f\u5efa\u7701|\u798f\u5dde\u5e02|\u7535\u4fe1\n  with open('.\/ip.merge.txt', 'r') as f:\n    for line in f.readlines():\n      item = line.strip().split('|')\n      print item[0], item[1], item[2], item[3], item[4], item[5], item[6]\n      start_ip = struct.pack('I', struct.unpack('!L', socket.inet_aton(item[0]))[0])\n      end_ip = struct.pack('I', struct.unpack('!L', socket.inet_aton(item[1]))[0])\n      region_item = '|'.join([item[2], item[3], item[4], item[5], item[6]])\n      region += region_item\n\n      ptr = struct.pack('I', int(bin(len(region_item))[2:].zfill(8) + bin(pointer)[2:].zfill(24), 2))\n      index += start_ip + end_ip + ptr\n      pointer += len(region_item)\n\n  index_start_ptr = pointer\n  index_end_ptr = pointer + len(index) - 12\n  super_block = struct.pack('I', index_start_ptr) + struct.pack('I', index_end_ptr)\n\n  n = 0\n  header_index = ''\n  for index_block in range(pointer, index_end_ptr, 8184):\n    header_index_block_ip = index[n * 8184:n * 8184 + 4]\n    header_index_block_ptr = index_block\n    header_index += header_index_block_ip + struct.pack('I', header_index_block_ptr)\n\n    n += 1\n\n  header_index += index[len(index) - 12: len(index) - 8] + struct.pack('I', index_end_ptr)\n\n  with open(IP_REGION_FILE, 'wb') as f:\n    f.write(super_block)\n    f.write(header_index)\n    f.seek(SUPER_BLOCK_LENGTH + HEADER_INDEX_LENGTH, 0)\n    f.write(region)\n    f.write(index)\n\n\nif __name__ == '__main__':\n  start_time = time.time()\n  generate_db_file()\n\n  print 'cost time: ', time.time() - start_time\n\n<\/pre>\n
\u4f7f\u7528 Redis \u7f13\u5b58<\/strong><\/span><\/p>\n
\u76ee\u524d\u6709\u4e24\u79cd\u65b9\u5f0f\u5bf9 IP \u4ee5\u53ca\u5f52\u5c5e\u5730\u4fe1\u606f\u8fdb\u884c\u7f13\u5b58\uff1a<\/p>\n
\u7b2c\u4e00\u79cd\u662f\u5c06\u8d77\u59cb IP\uff0c\u7ed3\u675f IP \u4ee5\u53ca\u4e2d\u95f4\u6240\u6709 IP \u8f6c\u6362\u6210\u6574\u578b\uff0c\u7136\u540e\u4ee5\u5b57\u7b26\u4e32\u65b9\u5f0f\uff0c\u7528\u8f6c\u6362\u540e\u7684 IP \u4f5c\u4e3a key\uff0c\u5f52\u5c5e\u5730\u4fe1\u606f\u4f5c\u4e3a value \u5b58\u5165 Redis\uff1b<\/p>\n
\u7b2c\u4e8c\u79cd\u662f\u91c7\u7528\u6709\u5e8f\u96c6\u5408\u548c\u6563\u5217\u65b9\u5f0f\uff0c\u9996\u5148\u5c06\u8d77\u59cb IP \u548c\u7ed3\u675f IP \u6dfb\u52a0\u5230\u6709\u5e8f\u96c6\u5408 ip2cityid\uff0c\u57ce\u5e02 ID \u4f5c\u4e3a\u6210\u5458\uff0c\u8f6c\u6362\u540e\u7684 IP \u4f5c\u4e3a\u5206\u503c\uff0c\u7136\u540e\u518d\u5c06\u57ce\u5e02 ID \u548c\u5f52\u5c5e\u5730\u4fe1\u606f\u6dfb\u52a0\u5230\u6563\u5217 cityid2city\uff0c\u57ce\u5e02 ID \u4f5c\u4e3a key\uff0c\u5f52\u5c5e\u5730\u4fe1\u606f\u4f5c\u4e3a value\u3002<\/p>\n
\u7b2c\u4e00\u79cd\u65b9\u5f0f\u5c31\u4e0d\u591a\u505a\u4ecb\u7ecd\u4e86\uff0c\u7b80\u5355\u7c97\u66b4\uff0c\u975e\u5e38\u4e0d\u63a8\u8350\u3002\u67e5\u8be2\u901f\u5ea6\u5f53\u7136\u5f88\u5feb\uff0c\u6beb\u79d2\u7ea7\u522b\uff0c\u4f46\u7f3a\u70b9\u4e5f\u5341\u5206\u660e\u663e\uff0c\u6211\u7528 1000 \u6761\u6570\u636e\u505a\u4e86\u6d4b\u8bd5\uff0c\u7f13\u5b58\u65f6\u95f4\u957f\uff0c\u5927\u6982 20 \u5206\u949f\uff0c\u5360\u7528\u7a7a\u95f4\u5927\uff0c\u5c06\u8fd1 1G\u3002<\/p>\n
\u4e0b\u9762\u4ecb\u7ecd\u7b2c\u4e8c\u79cd\u65b9\u5f0f\uff0c\u76f4\u63a5\u770b\u4ee3\u7801\uff1a<\/p>\n
\n# generate_to_redis.py\n# -*- coding:utf-8 -*-\n\nimport time\nimport json\nfrom redis import Redis\n\n\ndef ip_to_num(x):\n  return sum([256 ** j * int(i) for j, i in enumerate(x.split('.')[::-1])])\n\n\n# \u8fde\u63a5 Redis\nconn = Redis(host='127.0.0.1', port=6379, db=10)\n\nstart_time = time.time()\n\n# \u6587\u4ef6\u683c\u5f0f\n# 1.0.0.0|1.0.0.255|\u6fb3\u5927\u5229\u4e9a|0|0|0|0\n# 1.0.1.0|1.0.3.255|\u4e2d\u56fd|0|\u798f\u5efa\u7701|\u798f\u5dde\u5e02|\u7535\u4fe1\nwith open('.\/ip.merge.txt', 'r') as f:\n  i = 1\n  for line in f.readlines():\n    item = line.strip().split('|')\n    # \u5c06\u8d77\u59cb IP \u548c\u7ed3\u675f IP \u6dfb\u52a0\u5230\u6709\u5e8f\u96c6\u5408 ip2cityid\n    # \u6210\u5458\u5206\u522b\u662f\u57ce\u5e02 ID \u548c ID + #, \u5206\u503c\u662f\u6839\u636e IP \u8ba1\u7b97\u7684\u6574\u6570\u503c\n    conn.zadd('ip2cityid', str(i), ip_to_num(item[0]), str(i) + '#', ip_to_num(item[1]) + 1)\n    # \u5c06\u57ce\u5e02\u4fe1\u606f\u6dfb\u52a0\u5230\u6563\u5217 cityid2city\uff0ckey \u662f\u57ce\u5e02 ID\uff0c\u503c\u662f\u57ce\u5e02\u4fe1\u606f\u7684 json \u5e8f\u5217\n    conn.hset('cityid2city', str(i), json.dumps([item[2], item[3], item[4], item[5]]))\n\n    i += 1\n\nend_time = time.time()\n\nprint 'start_time: ' + str(start_time) + ', end_time: ' + str(end_time) + ', cost time: ' + str(end_time - start_time)\n\n<\/pre>\n
\n# test.py\n# -*- coding:utf-8 -*-\n\nimport sys\nimport time\nimport json\nimport socket\nimport struct\nfrom redis import Redis\n\n# \u8fde\u63a5 Redis\nconn = Redis(host='127.0.0.1', port=6379, db=10)\n\n# \u5c06 IP \u8f6c\u6362\u6210\u6574\u6570\nip = struct.unpack(\"!L\", socket.inet_aton(sys.argv[1]))[0]\n\nstart_time = time.time()\n# \u5c06\u6709\u5e8f\u96c6\u5408\u4ece\u5927\u5230\u5c0f\u6392\u5e8f\uff0c\u53d6\u5c0f\u4e8e\u8f93\u5165 IP \u503c\u7684\u7b2c\u4e00\u6761\u6570\u636e\ncityid = conn.zrevrangebyscore('ip2cityid', ip, 0, start=0, num=1)\n# \u5982\u679c\u8fd4\u56de cityid \u662f\u7a7a\uff0c\u6216\u8005\u5339\u914d\u5230\u4e86 # \u53f7\uff0c\u8bf4\u660e\u6ca1\u6709\u627e\u5230\u5bf9\u5e94\u5730\u5740\u6bb5\nif not cityid or cityid[0].endswith('#'):\n  print 'no city info...'\nelse:\n  # \u6839\u636e\u57ce\u5e02 ID \u5230\u6563\u5217\u8868\u53d6\u51fa\u57ce\u5e02\u4fe1\u606f\n  ret = json.loads(conn.hget('cityid2city', cityid[0]))\n  print ret[0], ret[1], ret[2]\n\nend_time = time.time()\nprint 'start_time: ' + str(start_time) + ', end_time: ' + str(end_time) + ', cost time: ' + str(end_time - start_time)\n\n<\/pre>\n
\n# python generate_to_redis.py \nstart_time: 1554300310.31, end_time: 1554300425.65, cost time: 115.333260059<\/pre>\n
\n# python test_2.py 1.0.16.0\n\u65e5\u672c 0 0\nstart_time: 1555081532.44, end_time: 1555081532.45, cost time: 0.000912189483643<\/pre>\n
\u6d4b\u8bd5\u6570\u636e\u5927\u6982 50 \u4e07\u6761\uff0c\u7f13\u5b58\u6240\u7528\u65f6\u95f4\u4e0d\u5230 2 \u5206\u949f\uff0c\u5360\u7528\u5185\u5b58 182M\uff0c\u67e5\u8be2\u901f\u5ea6\u6beb\u79d2\u7ea7\u522b\u3002\u663e\u800c\u6613\u89c1\uff0c\u8fd9\u79cd\u65b9\u5f0f\u66f4\u503c\u5f97\u5c1d\u8bd5\u3002<\/p>\n
zrevrangebyscore \u65b9\u6cd5\u7684\u65f6\u95f4\u590d\u6742\u5ea6\u662f O(log(N)+M)\uff0c N \u4e3a\u6709\u5e8f\u96c6\u7684\u57fa\u6570\uff0c M \u4e3a\u7ed3\u679c\u96c6\u7684\u57fa\u6570\u3002\u53ef\u89c1\u5f53 N \u7684\u503c\u8d8a\u5927\uff0c\u67e5\u8be2\u6548\u7387\u8d8a\u6162\uff0c\u5177\u4f53\u5728\u591a\u5927\u7684\u6570\u636e\u91cf\u8fd8\u53ef\u4ee5\u9ad8\u6548\u67e5\u8be2\uff0c\u8fd9\u4e2a\u6709\u5f85\u9a8c\u8bc1\u3002\u4e0d\u8fc7\u8fd9\u4e2a\u95ee\u9898\u6211\u89c9\u5f97\u5e76\u4e0d\u7528\u62c5\u5fc3\uff0c\u9047\u5230\u4e86\u518d\u8bf4\u5427\u3002<\/p>\n
\u4ee5\u4e0a\u6240\u8ff0\u662f\u5c0f\u7f16\u7ed9\u5927\u5bb6\u4ecb\u7ecd\u7684\u4f7f\u7528Redis\u6709\u5e8f\u96c6\u5408\u5b9e\u73b0IP\u5f52\u5c5e\u5730\u67e5\u8be2\u8be6\u89e3\u6574\u5408\uff0c\u5e0c\u671b\u5bf9\u5927\u5bb6\u6709\u6240\u5e2e\u52a9\uff0c\u5982\u679c\u5927\u5bb6\u6709\u4efb\u4f55\u7591\u95ee\u8bf7\u7ed9\u6211\u7559\u8a00\uff0c\u5c0f\u7f16\u4f1a\u53ca\u65f6\u56de\u590d\u5927\u5bb6\u7684\u3002\u5728\u6b64\u4e5f\u975e\u5e38\u611f\u8c22\u5927\u5bb6\u5bf9golang\u5b66\u4e60\u7f51\u7f51\u7ad9\u7684\u652f\u6301\uff01<\/p>\n","protected":false},"excerpt":{"rendered":"
\u4f7f\u7528Redis\u6709\u5e8f\u96c6\u5408\u5b9e\u73b0IP\u5f52...<\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"closed","ping_status":"","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[101],"tags":[],"class_list":["post-201444","post","type-post","status-publish","format-standard","hentry","category-database"],"_links":{"self":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts\/201444","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/comments?post=201444"}],"version-history":[{"count":0,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts\/201444\/revisions"}],"wp:attachment":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/media?parent=201444"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/categories?post=201444"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/tags?post=201444"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}