blob: 00c59fbc979c07bac43ae285fd9b8b280d36f589 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
// Copyright 2018 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.search.dispatch;
import com.yahoo.prelude.fastsearch.FS4ResourcePool;
import com.yahoo.search.dispatch.SearchCluster.Group;
import com.yahoo.search.dispatch.SearchCluster.Node;
import java.util.Optional;
/**
* An extension to CloseableChannel that encapsulates the release of a LoadBalancer group allocation.
*
* @author ollivir
*/
public class DispatchedChannel extends CloseableChannel {
private final SearchCluster.Group group;
private final LoadBalancer loadBalancer;
private boolean groupAllocated = true;
public DispatchedChannel(FS4ResourcePool fs4ResourcePool, LoadBalancer loadBalancer, Group group, Node node) {
super(fs4ResourcePool.getBackend(node.hostname(), node.fs4port(), Optional.of(node.key())));
this.loadBalancer = loadBalancer;
this.group = group;
}
public DispatchedChannel(FS4ResourcePool fs4ResourcePool, LoadBalancer loadBalancer, Group group) {
this(fs4ResourcePool, loadBalancer, group, group.nodes().iterator().next());
}
public void close() {
if (groupAllocated) {
groupAllocated = false;
loadBalancer.releaseGroup(group);
}
super.close();
}
}
|