Skip to content

Commit abcf179

Browse files
authored
Use RoundRobinInetAddressResolver for DNS resolution
[#338]
1 parent 3745841 commit abcf179

File tree

2 files changed

+44
-0
lines changed

2 files changed

+44
-0
lines changed
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,42 @@
1+
/*
2+
* Copyright 2020 the original author or authors.
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* https://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
17+
package io.r2dbc.postgresql.client;
18+
19+
import io.netty.resolver.AddressResolver;
20+
import io.netty.resolver.AddressResolverGroup;
21+
import io.netty.resolver.DefaultNameResolver;
22+
import io.netty.resolver.RoundRobinInetAddressResolver;
23+
import io.netty.util.concurrent.EventExecutor;
24+
25+
import java.net.InetSocketAddress;
26+
27+
/**
28+
* When the {@link InetSocketAddress} resolves to multiple IP addresses, pick one randomly.
29+
*
30+
* @since 0.8.6
31+
*/
32+
final class BalancedResolverGroup extends AddressResolverGroup<InetSocketAddress> {
33+
BalancedResolverGroup() {
34+
}
35+
36+
public static final BalancedResolverGroup INSTANCE = new BalancedResolverGroup();
37+
38+
@Override
39+
protected AddressResolver<InetSocketAddress> newResolver(EventExecutor executor) throws Exception {
40+
return new RoundRobinInetAddressResolver(executor, new DefaultNameResolver(executor)).asAddressResolver();
41+
}
42+
}

src/main/java/io/r2dbc/postgresql/client/ReactorNettyClient.java

+2
Original file line numberDiff line numberDiff line change
@@ -374,6 +374,8 @@ public static Mono<ReactorNettyClient> connect(SocketAddress socketAddress, Conn
374374
tcpClient = tcpClient.runOn(settings.getRequiredLoopResources());
375375
}
376376

377+
tcpClient = tcpClient.resolver(new BalancedResolverGroup());
378+
377379
tcpClient = tcpClient.option(ChannelOption.SO_KEEPALIVE, settings.isTcpKeepAlive());
378380
tcpClient = tcpClient.option(ChannelOption.TCP_NODELAY, settings.isTcpNoDelay());
379381
}

0 commit comments

Comments
 (0)