写点什么

Java 自定义 DNS 解析器三种实践

作者:FunTester
  • 2022 年 1 月 25 日
  • 本文字数:2545 字

    阅读完需:约 8 分钟

Java自定义DNS解析器三种实践

最近终于用上了高性能的测试机(54C96G * 3),相较之前的单机性能提升了三倍,数量提升了三倍,更关键的宽带提单机升了 30 倍不止,总体讲提升了 100 多倍,这下再也不用担心单机压力机瓶颈,直接原地起飞。


不过没高兴 5 分钟,我发现接口居然请求不通,经过一阵拨乱反正终于找到原因:域名无法解析,IP 无法直接访问。


自然而然,解决方案呼之欲出:自定义 Java DNS 解析器。


经过同事指点、资料搜索和探索实践。终于锁定了两个核心类:org.apache.http.impl.conn.InMemoryDnsResolverorg.apache.http.impl.conn.SystemDefaultDnsResolver,下面我会演示一下这两个类的使用实践,其中主要区别还是在负载均衡的实现上,这个有空再分享。

InMemoryDnsResolver

这个类使用比较简单,先写一个 Demo,实现一个简单的域名解析。


    /**     * 重写Java自定义DNS解析器,非负载均衡     *     * @return     */    private static DnsResolver getDnsResolver2() {        InMemoryDnsResolver dnsResolver = new InMemoryDnsResolver();
try { dnsResolver.add("fun.tester", InetAddress.getByName("127.0.0.1")); } catch (Exception e) { e.printStackTrace(); }
return dnsResolver; }
复制代码


这样我们就可以把fun.tester解析到127.0.0.1上了,后面我会进行一个简单的测试。

SystemDefaultDnsResolver

这个看名字是系统默认 DNS 解析器,但默认在哪,我也没看出来,唯一可以查到的引用就是异步线程池管理器使用org.apache.http.impl.nio.conn.PoolingNHttpClientConnectionManager#PoolingNHttpClientConnectionManager(org.apache.http.nio.reactor.ConnectingIOReactor, org.apache.http.nio.conn.NHttpConnectionFactory<org.apache.http.nio.conn.ManagedNHttpClientConnection>, org.apache.http.config.Registry<org.apache.http.nio.conn.SchemeIOSessionStrategy>, org.apache.http.conn.SchemePortResolver, org.apache.http.conn.DnsResolver, long, java.util.concurrent.TimeUnit),这里面还涉及到其他参数,这里就不讲了。


我们先看这个 Demo。


    /**     * 重写Java自定义DNS解析器,负载均衡     *     * @return     */    private static DnsResolver getDnsResolver() {        return new SystemDefaultDnsResolver() {            @Override            public InetAddress[] resolve(final String host) throws UnknownHostException {                if (host.equalsIgnoreCase("fun.tester")) {                    return new InetAddress[]{InetAddress.getByName("127.0.0.1")};                } else {                    return super.resolve(host);                }            }        };    }
复制代码

自定义 DnsResolver

通过源码可以看出,两个实现类都是通过实现org.apache.http.conn.DnsResolver这个接口中org.apache.http.conn.DnsResolver#resolve方法。我们自己可以完全自己实现。


    /**     * 自定义本地DNS解析器实现     *     * @return     */    private static DnsResolver getDnsResolver3() {        return new DnsResolver() {            @Override            public InetAddress[] resolve(final String host) throws UnknownHostException {                if (host.equalsIgnoreCase("fun.tester")) {                    return new InetAddress[]{InetAddress.getByName("127.0.0.1")};                } else {                    return InetAddress.getAllByName(host);                }            }        };    }
复制代码


仔细看不难发现,其实就是代码缝合怪。

连接池管理器

下面分享一下如何使用自定义的org.apache.http.conn.DnsResolver,就是在创建连接池管理器的时候设置一下就可以。

测试

首先我在本地起一个 HTTP 服务,端口 12345,非常简单。代码如下:


    static void main(String[] args) {        def util = new ArgsUtil(args)        def server = getServerNoLog(util.getIntOrdefault(0, 12345))        server.response("Have Fun ~ Tester !")        def run = run(server)        waitForKey("fan")        run.stop()    }
复制代码


然后我准备一个测试脚本:


    public static void main(String[] args) {        String url = "http://fun.tester:12345/"        def get = getHttpGet(url)        def funtester = {            fun {                getHttpResponse(get)            }        }        10.times {            funtester()        }    }
复制代码


控制台日志输出:


INFO-> 27.214 F-1  请求uri:http://fun.tester:12345/ , 耗时:304 ms , HTTPcode: 200INFO-> 27.214 F-4  请求uri:http://fun.tester:12345/ , 耗时:304 ms , HTTPcode: 200INFO-> 27.214 F-10 请求uri:http://fun.tester:12345/ , 耗时:305 ms , HTTPcode: 200INFO-> 27.214 F-5  请求uri:http://fun.tester:12345/ , 耗时:305 ms , HTTPcode: 200INFO-> 27.214 F-2  请求uri:http://fun.tester:12345/ , 耗时:305 ms , HTTPcode: 200INFO-> 27.214 F-8  请求uri:http://fun.tester:12345/ , 耗时:305 ms , HTTPcode: 200INFO-> 27.214 F-3  请求uri:http://fun.tester:12345/ , 耗时:305 ms , HTTPcode: 200INFO-> 27.214 F-7  请求uri:http://fun.tester:12345/ , 耗时:305 ms , HTTPcode: 200INFO-> 27.214 F-6  请求uri:http://fun.tester:12345/ , 耗时:305 ms , HTTPcode: 200INFO-> 27.214 F-9  请求uri:http://fun.tester:12345/ , 耗时:305 ms , HTTPcode: 200
复制代码


三种实现方式控制台输出大同小异,都能满足我们的需求,当然仅仅是功能测试场景下。下期会结合源码分析如何实现负载均衡。

Have Fun ~ Tester !

发布于: 2022 年 01 月 25 日阅读数: 96
用户头像

FunTester

关注

公众号:FunTester,750篇原创,欢迎关注 2020.10.20 加入

公众号FunTester,坚持原创文章的测试人,一个有趣的灵魂。

评论

发布
暂无评论
Java自定义DNS解析器三种实践_Java_FunTester_InfoQ写作社区