Skip to content

Commit 6a9b6b3

Browse files
johnnymp911de
authored andcommitted
Use RoundRobinInetAddressResolver for DNS resolution
[#338]
1 parent 9518780 commit 6a9b6b3

File tree

2 files changed

+43
-0
lines changed

2 files changed

+43
-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

+1
Original file line numberDiff line numberDiff line change
@@ -389,6 +389,7 @@ public static Mono<ReactorNettyClient> connect(ConnectionProvider connectionProv
389389
tcpClient = tcpClient.runOn(connectionSettings.getRequiredLoopResources());
390390
}
391391

392+
tcpClient = tcpClient.resolver(new BalancedResolverGroup());
392393
tcpClient = tcpClient.option(ChannelOption.SO_KEEPALIVE, connectionSettings.isTcpKeepAlive());
393394
tcpClient = tcpClient.option(ChannelOption.TCP_NODELAY, connectionSettings.isTcpNoDelay());
394395
}

0 commit comments

Comments
 (0)