新来了个技术总监,居然要我做一个ip属地功能。。不服就干!(代码片段)

Java技术栈 Java技术栈     2022-12-07     142

关键词:

作者:ThinkingKeep
链接:https://juejin.cn/post/7118954784853327903

细心的朋友应该会发现,最近,继新浪微博之后,头条、腾讯、抖音、知乎、快手、小红书等各大平台陆陆续续都上线了“网络用户IP地址显示功能”,境外用户显示的是国家,国内的用户显示的省份,而且此项显示无法关闭,归属地强制显示。

作为技术人,那!这个功能要怎么实现呢?

下面,我就来讲讲,Java中是如何获取IP属地的,主要分为以下几步:

  • 通过 HttpServletRequest 对象,获取用户的 IP 地址
  • 通过 IP 地址,获取对应的省份、城市

首先需要写一个 IP 获取的工具类,因为每一次用户的 Request 请求,都会携带上请求的 IP 地址放到请求头中

通过此方法,从请求Header中获取到用户的IP地址

目前本人在做的项目中,也有获取IP地址归属地省份、城市的需求,用的是:淘宝IP库

地址:ip.taobao.com/

原来的请求源码如下:

可以看到日志log文件中,大量的the request over max qps for user问题


下面,给大家介绍下之前在Github冲浪时发现的今天的主角:

目前最新已更新到了v2.0版本,ip2region v2.0是一个离线IP地址定位库和IP定位数据管理框架,10微秒级别的查询效率,准提供了众多主流编程语言的 xdb 数据生成和查询客户端实现。

99.9%准确率:

数据聚合了一些知名ip到地名查询提供商的数据,这些是他们官方的的准确率,经测试着实比经典的纯真IP定位准确一些。

多查询客户端的支持

已经集成的客户端有:java、C#、php、c、python、nodejs、php扩展(php5和php7)、golang、rust、lua、lua_c, nginx。

binding描述开发状态binary查询耗时b-tree查询耗时memory查询耗时
cANSC c binding已完成0.0x毫秒0.0x毫秒0.00x毫秒
c#c# binding已完成0.x毫秒0.x毫秒0.1x毫秒
golanggolang binding已完成0.x毫秒0.x毫秒0.1x毫秒
javajava binding已完成0.x毫秒0.x毫秒0.1x毫秒
lualua实现的binding已完成0.x毫秒0.x毫秒0.x毫秒
lua_clua的c扩展已完成0.0x毫秒0.0x毫秒0.00x毫秒
nginxnginx的c扩展已完成0.0x毫秒0.0x毫秒0.00x毫秒
nodejsnodejs已完成0.x毫秒0.x毫秒0.1x毫秒
phpphp实现的binding已完成0.x毫秒0.1x毫秒0.1x毫秒
php5_extphp5的c扩展已完成0.0x毫秒0.0x毫秒0.00x毫秒
php7_extphp7的c扩展已完成0.0毫秒0.0x毫秒0.00x毫秒
pythonpython bindng已完成0.x毫秒0.x毫秒0.x毫秒
rustrust binding已完成0.x毫秒0.x毫秒0.x毫秒

Ip2region V2.0 特性

1、标准化的数据格式

每个 ip 数据段的 region 信息都固定了格式:国家|区域|省份|城市|ISP,只有中国的数据绝大部分精确到了城市,其他国家部分数据只能定位到国家,后前的选项全部是0。

2、数据去重和压缩

xdb 格式生成程序会自动去重和压缩部分数据,默认的全部 IP 数据,生成的 ip2region.xdb 数据库是 11MiB,随着数据的详细度增加数据库的大小也慢慢增大。

3、极速查询响应

即使是完全基于 xdb 文件的查询,单次查询响应时间在十微秒级别,可通过如下两种方式开启内存加速查询:

  1. vIndex 索引缓存 :使用固定的 512KiB 的内存空间缓存 vector index 数据,减少一次 IO 磁盘操作,保持平均查询效率稳定在10-20微秒之间。
  2. xdb 整个文件缓存:将整个 xdb 文件全部加载到内存,内存占用等同于 xdb 文件大小,无磁盘 IO 操作,保持微秒级别的查询效率。

4、极速查询响应

v2.0 格式的 xdb 支持亿级别的 IP 数据段行数,region 信息也可以完全自定义,例如:你可以在 region 中追加特定业务需求的数据,例如:GPS信息/国际统一地域信息编码/邮编等。也就是你完全可以使用 ip2region 来管理你自己的 IP 定位数据。

ip2region xdb java 查询客户端实现

  • 使用方式

引入maven仓库:

<dependency>
    <groupId>org.lionsoul</groupId>
    <artifactId>ip2region</artifactId>
    <version>2.6.4</version>
</dependency>
  • 完全基于文件的查询
import org.lionsoul.ip2region.xdb.Searcher;
import java.io.*;
import java.util.concurrent.TimeUnit;

public class SearcherTest 
    public static void main(String[] args) 
        // 1、创建 searcher 对象
        String dbPath = "ip2region.xdb file path";
        Searcher searcher = null;
        try 
            searcher = Searcher.newWithFileOnly(dbPath);
         catch (IOException e) 
            System.out.printf("failed to create searcher with `%s`: %s\\n", dbPath, e);
            return;
        

        // 2、查询
        try 
            String ip = "1.2.3.4";
            long sTime = System.nanoTime();
            String region = searcher.search(ip);
            long cost = TimeUnit.NANOSECONDS.toMicros((long) (System.nanoTime() - sTime));
            System.out.printf("region: %s, ioCount: %d, took: %d μs\\n", region, searcher.getIOCount(), cost);
         catch (Exception e) 
            System.out.printf("failed to search(%s): %s\\n", ip, e);
        

        // 3、备注:并发使用,每个线程需要创建一个独立的 searcher 对象单独使用。
    

  • 缓存VectorIndex索引

我们可以提前从 xdb 文件中加载出来 VectorIndex 数据,然后全局缓存,每次创建 Searcher 对象的时候使用全局的 VectorIndex 缓存可以减少一次固定的 IO 操作,从而加速查询,减少 IO 压力。

import org.lionsoul.ip2region.xdb.Searcher;
import java.io.*;
import java.util.concurrent.TimeUnit;

public class SearcherTest 
    public static void main(String[] args) 
        String dbPath = "ip2region.xdb file path";

        // 1、从 dbPath 中预先加载 VectorIndex 缓存,并且把这个得到的数据作为全局变量,后续反复使用。
        byte[] vIndex;
        try 
            vIndex = Searcher.loadVectorIndexFromFile(dbPath);
         catch (Exception e) 
            System.out.printf("failed to load vector index from `%s`: %s\\n", dbPath, e);
            return;
        

        // 2、使用全局的 vIndex 创建带 VectorIndex 缓存的查询对象。
        Searcher searcher;
        try 
            searcher = Searcher.newWithVectorIndex(dbPath, vIndex);
         catch (Exception e) 
            System.out.printf("failed to create vectorIndex cached searcher with `%s`: %s\\n", dbPath, e);
            return;
        

        // 3、查询
        try 
            String ip = "1.2.3.4";
            long sTime = System.nanoTime();
            String region = searcher.search(ip);
            long cost = TimeUnit.NANOSECONDS.toMicros((long) (System.nanoTime() - sTime));
            System.out.printf("region: %s, ioCount: %d, took: %d μs\\n", region, searcher.getIOCount(), cost);
         catch (Exception e) 
            System.out.printf("failed to search(%s): %s\\n", ip, e);
        

        // 备注:每个线程需要单独创建一个独立的 Searcher 对象,但是都共享全局的制度 vIndex 缓存。
    

  • 缓存整个xdb数据

我们也可以预先加载整个 ip2region.xdb 的数据到内存,然后基于这个数据创建查询对象来实现完全基于文件的查询,类似之前的 memory search。

import org.lionsoul.ip2region.xdb.Searcher;
import java.io.*;
import java.util.concurrent.TimeUnit;

public class SearcherTest 
    public static void main(String[] args) 
        String dbPath = "ip2region.xdb file path";

        // 1、从 dbPath 加载整个 xdb 到内存。
        byte[] cBuff;
        try 
            cBuff = Searcher.loadContentFromFile(dbPath);
         catch (Exception e) 
            System.out.printf("failed to load content from `%s`: %s\\n", dbPath, e);
            return;
        

        // 2、使用上述的 cBuff 创建一个完全基于内存的查询对象。
        Searcher searcher;
        try 
            searcher = Searcher.newWithBuffer(cBuff);
         catch (Exception e) 
            System.out.printf("failed to create content cached searcher: %s\\n", e);
            return;
        

        // 3、查询
        try 
            String ip = "1.2.3.4";
            long sTime = System.nanoTime();
            String region = searcher.search(ip);
            long cost = TimeUnit.NANOSECONDS.toMicros((long) (System.nanoTime() - sTime));
            System.out.printf("region: %s, ioCount: %d, took: %d μs\\n", region, searcher.getIOCount(), cost);
         catch (Exception e) 
            System.out.printf("failed to search(%s): %s\\n", ip, e);
        

        // 备注:并发使用,用整个 xdb 数据缓存创建的查询对象可以安全的用于并发,也就是你可以把这个 searcher 对象做成全局对象去跨线程访问。
    

IDEA中做个测试

完全基于文件的查询

ip属地国内的话,会展示省份,国外的话,只会展示国家。可以通过如下图这个方法进行进一步封装,得到获取IP属地的信息。


下面是官网给出的命令运行jar方式给出的测试demo,可以理解下

编译测试程序

通过 maven 来编译测试程序。

# cd 到 java binding 的根目录
cd binding/java/
mvn compile package

然后会在当前目录的 target 目录下得到一个 ip2region-version.jar 的打包文件。

查询测试

可以通过 java -jar ip2region-version.jar search 命令来测试查询:

➜  java git:(v2.0_xdb) ✗ java -jar target/ip2region-2.6.0.jar search
java -jar ip2region-version.jar search [command options]
options:
 --db string              ip2region binary xdb file path
 --cache-policy string    cache policy: file/vectorIndex/content

例如:使用默认的 data/ip2region.xdb 文件进行查询测试:

➜  java git:(v2.0_xdb) ✗ java -jar target/ip2region-2.6.0.jar search --db=../../data/ip2region.xdb
ip2region xdb searcher test program, cachePolicy: vectorIndex
type 'quit' to exit
ip2region>> 1.2.3.4
region: 美国|0|华盛顿|0|谷歌, ioCount: 7, took: 82 μs
ip2region>>

输入 ip 即可进行查询测试,也可以分别设置 cache-policy 为 file/vectorIndex/content 来测试三种不同缓存实现的查询效果。

bench 测试

可以通过 java -jar ip2region-version.jar bench 命令来进行 bench 测试,一方面确保 xdb 文件没有错误,一方面可以评估查询性能:

➜  java git:(v2.0_xdb) ✗ java -jar target/ip2region-2.6.0.jar bench
java -jar ip2region-version.jar bench [command options]
options:
 --db string              ip2region binary xdb file path
 --src string             source ip text file path
 --cache-policy string    cache policy: file/vectorIndex/content

例如:通过默认的 data/ip2region.xdb 和 data/ip.merge.txt 文件进行 bench 测试:

➜  java git:(v2.0_xdb) ✗ java -jar target/ip2region-2.6.0.jar bench --db=../../data/ip2region.xdb --src=../../data/ip.merge.txt
Bench finished, cachePolicy: vectorIndex, total: 3417955, took: 8s, cost: 2 μs/op

可以通过分别设置 cache-policy 为 file/vectorIndex/content 来测试三种不同缓存实现的效果。 @Note: 注意 bench 使用的 src 文件要是生成对应 xdb 文件相同的源文件。

到这里获取用户IP属地已经完成啦,这篇文章介绍的v2.0版本,有兴趣的小伙伴可以登录上门的github地址了解下v1.0版本

如若觉得有用,欢迎收藏+点赞,如遇到什么问题,欢迎留言讨论

近期热文推荐:

1.1,000+ 道 Java面试题及答案整理(2022最新版)

2.劲爆!Java 协程要来了。。。

3.Spring Boot 2.x 教程,太全了!

4.别再写满屏的爆爆爆炸类了,试试装饰器模式,这才是优雅的方式!!

5.《Java开发手册(嵩山版)》最新发布,速速下载!

觉得不错,别忘了随手点赞+转发哦!

新来个技术总监要我做一个ip属地功能~

Hollis的新书限时折扣中,一本深入讲解Java基础的干货笔记!文章来源:juejin.cn/post/7118954784853327903目录背景HttpServletRequest获取IPIp2regionIp2regionV2.0特性ip2regionxdbjava查询客户端实现IDEA中做个测试背景细心的朋友应该会发... 查看详情

新来个技术总监,禁止我们使用lombok!(代码片段)

...,在一家小型互联网公司做Java后端开发,最近他们公司新来了一个技术总监,这位技术总监对技术细节很看重,一来公司之后就推出了很多"政策",比如定义了很多开发规范、日志规范、甚至是要求大家统一使用某一款ID... 查看详情

新来了个同事,设计模式用的是真优雅呀!代码如诗!!

之前有小伙伴私信我说看源码的时候感觉源码很难,不知道该怎么看,其实这有部分原因是因为没有弄懂一些源码实现的套路,也就是设计模式,所以本文我就总结了9种在源码中非常常见的设计模式,并列举... 查看详情

新来了个同事,设计模式用的是真优雅呀!代码如诗!!

之前有小伙伴私信我说看源码的时候感觉源码很难,不知道该怎么看,其实这有部分原因是因为没有弄懂一些源码实现的套路,也就是设计模式,所以本文我就总结了9种在源码中非常常见的设计模式,并列举... 查看详情

公司新来了个软件测试工程师,一副毛头小子的样儿,哪想到是新一代卷王...

内卷,是现在热度非常高的一个词汇,随着热度不断攀升,隐隐到了“万物皆可卷”的程度。在程序员职场上,什么样的人最让人反感呢?是技术不好的人吗?并不是。技术不好的同事,我们可以帮他。是技术太... 查看详情

新来一个技术总监:禁止戴耳机写代码,发现就扣绩效。。

...时送达前言是的,小李(化名)上班戴耳机被新来的技术总监批了。。事情是这样的,本来小李所在的公司也没有规定上班时间不能戴耳机写代码,之前都戴的好好的,某次开技术会议,时间到了,... 查看详情

部门新来了个软件测试工程师,一副毛头小子的样儿,哪想到是新一代卷王...

内卷,是现在热度非常高的一个词汇,随着热度不断攀升,隐隐到了“万物皆可卷”的程度。在程序员职场上,什么样的人最让人反感呢?是技术不好的人吗?并不是。技术不好的同事,我们可以帮他。是技术太... 查看详情

部门新来了个软件测试工程师,看着一副毛头小子的样儿,哪想到是新一代卷王...

内卷,是现在热度非常高的一个词汇,随着热度不断攀升,隐隐到了“万物皆可卷”的程度。在程序员职场上,什么样的人最让人反感呢?是技术不好的人吗?并不是。技术不好的同事,我们可以帮他。是技术太... 查看详情

新来个技术总监,给团队引入了这款开发神器,同事直呼哇塞!

来自读者投稿:我在一家做微信营销的公司干技术leader,带40多个人,公司名就不说了。在这个位置上做了好几年,把团队从小带大,公司虽然不算风口浪尖上的高增长业务,但技术这块儿也从来没出过什... 查看详情

新来个技术总监,给团队引入了这款开发神器,同事直呼哇塞!

来自读者投稿:我在一家做微信营销的公司干技术leader,带40多个人,公司名就不说了。在这个位置上做了好几年,把团队从小带大,公司虽然不算风口浪尖上的高增长业务,但技术这块儿也从来没出过什... 查看详情

新来个技术总监,把rabbitmq讲的那叫一个透彻,佩服!

整体阅读时间,在40分钟左右。常见的消息队列很多,主要包括RabbitMQ、Kafka、RocketMQ和ActiveMQ,这篇文章只讲RabbitMQ,先讲原理,后搞实战。文章很长,如果你能一次性看完,“大哥,请收下我的膝... 查看详情

公司新来了个腾讯拿25k出来的,真是砂纸擦屁股,给我露了一手啊

一进公司,看见门外有几个带着行李在外面等待的人,应该就是来入职的,果不其然,在进公司后,召开了新人见面会,让他们做了自我介绍和职业规划以及部门,令我印象最深刻的就是一个刚刚从腾... 查看详情

公司新来了个腾讯拿30k出来的,让我见识到了什么叫真正的软件测试天花板...

今天公司新来个腾讯30k出来的,让我见识到了什么叫真正的基础天花板…最近内卷严重,各种跳槽裁员,分享一套学习笔记/面试手册,年后跳槽的朋友可以好好刷一刷,还是挺有必要的,它几乎涵盖了所... 查看详情

新来的"大神"用策略模式把ifelse给"优化"了,技术总监说:能不能想好了再改?(代码片段)

新来的"大神"用策略模式把ifelse给"优化"了,技术总监说:能不能想好了再改?收录于话题#代码优化4个△Hollis,一个对Coding有着独特追求的人△这是Hollis的第285篇原创分享作者l上帝爱吃苹果来源lHollis(ID:hollischuan... 查看详情

如何看待快手知乎等平台将上线显示用户ip属地功能?

...、知乎、百家号等多家平台宣布,将上线显示用户IP属地功能。此前,新浪微博也宣布相关功能,将在用户个人主页展示/P1日属地域信息,境内展示到省(区、市),境外展示到国家(地区),信息以网络... 查看详情

如何胜任一个小型公司的技术总监?我的感想

...,深有感触。对照自身在公司中遇到的情况,我给自己做了个现状分析。作为程序员,长久的职业价值是什么?如何让自己多年的经验不白费?如何跟精力更好的年轻人竞争?我特别赞同文中一句话:”软件开发是类似外科医生... 查看详情

新来个技术总监,这redis分布式锁设计的真漂亮!(代码片段)

前言在分布式系统中,由于redis分布式锁相对于更简单和高效,成为了分布式锁的首先,被我们用到了很多实际业务场景当中。但不是说用了redis分布式锁,就可以高枕无忧了,如果没有用好或者用对,也... 查看详情

公司新来了个腾讯拿30k出来的,让我见识到了什么叫真正的软件测试天花板...

今天公司新来个腾讯30k出来的,让我见识到了什么叫真正的基础天花板…最近内卷严重,各种跳槽裁员,分享一套学习笔记/面试手册,年后跳槽的朋友可以好好刷一刷,还是挺有必要的,它几乎涵盖了所... 查看详情