Mirror/MarginaliaSearch
1
0
Fork 0
mirror of https://github.com/MarginaliaSearch/MarginaliaSearch.git synced 2025-02-23 13:09:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

(client) Fix handling of channel refreshes

Browse Source 
The previous code made an incorrect assumption that all routes refer to the same node, and would overwrite the route list on each update.  This lead to storms of closing and opening channels whenever an update was received.

The new code is correctly aware that we may talk to multiple nodes.
This commit is contained in:
Viktor Lofgren 2024-02-20 14:14:09 +01:00
parent f85ec28a16
commit 746a865106
1 changed files with 14 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
code/common/service-discovery/src/main/java/nu/marginalia/service/client/GrpcSingleNodeChannelPool.java
View File

@ -21,7 +21,7 @@ import java.util.function.Function;
* Manages unicast-style requests */
public class GrpcSingleNodeChannelPool<STUB> extends ServiceChangeMonitor {
private final Map<InstanceAddress<?>, ManagedChannel> channels = new ConcurrentHashMap<>();
private volatile Set<InstanceAddress<?>> routes = Set.of();
private final Map<Integer, Set<InstanceAddress<?>>> routes = new ConcurrentHashMap<>();
private static final Logger logger = LoggerFactory.getLogger(GrpcSingleNodeChannelPool.class);
@ -68,31 +68,25 @@ public class GrpcSingleNodeChannelPool<STUB> extends ServiceChangeMonitor {
private void refreshNode(int node) {
Set<InstanceAddress<?>> newRoutes = serviceRegistryIf.getEndpoints(ApiSchema.GRPC, serviceId, node);
Set<InstanceAddress<?>> oldRoutes = routes;
Set<InstanceAddress<?>> oldRoutes = routes.getOrDefault(node, Set.of());
if (!oldRoutes.equals(newRoutes)) {
// Find the routes that have been added or removed
for (var route : Sets.symmetricDifference(oldRoutes, newRoutes)) {
// Find the routes that have been added or removed
for (var route : Sets.symmetricDifference(oldRoutes, newRoutes)) {
ManagedChannel oldChannel;
ManagedChannel oldChannel;
if (newRoutes.contains(route)) {
logger.info(STR."Adding channel for \{serviceId.serviceName}-\{node} \{route.host()}:\{route.port()}");
var newChannel = channelConstructor.apply(route);
oldChannel = channels.put(route, newChannel);
} else {
logger.info(STR."Removing channel for \{serviceId.serviceName}-\{node} \{route.host()}:\{route.port()}");
oldChannel = channels.remove(route);
}
if (oldChannel != null)
oldChannel.shutdown();
if (newRoutes.contains(route)) {
var newChannel = channelConstructor.apply(route);
oldChannel = channels.put(route, newChannel);
} else {
oldChannel = channels.remove(route);
}
routes = newRoutes;
if (oldChannel != null)
oldChannel.shutdown();
}
routes.put(node, newRoutes);
}
public boolean hasChannel() {