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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.curator;
import com.yahoo.net.HostName;
import java.util.List;
import java.util.Objects;
import java.util.function.Function;
/**
* A connection spec for Curator.
*
* @author mpolden
*/
class ConnectionSpec {
private final String local;
private final String ensemble;
private final int ensembleSize;
private ConnectionSpec(String local, String ensemble, int ensembleSize) {
this.local = requireNonEmpty(local, "local spec");
this.ensemble = requireNonEmpty(ensemble, "ensemble spec");
this.ensembleSize = ensembleSize;
}
/** Returns the local spec. This may be a subset of the ensemble spec */
public String local() {
return local;
}
/** Returns the ensemble spec. This always contains all nodes in the ensemble */
public String ensemble() {
return ensemble;
}
/** Returns the number of servers in the ensemble */
public int ensembleSize() {
return ensembleSize;
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
ConnectionSpec that = (ConnectionSpec) o;
return ensembleSize == that.ensembleSize &&
local.equals(that.local) &&
ensemble.equals(that.ensemble);
}
@Override
public int hashCode() {
return Objects.hash(local, ensemble, ensembleSize);
}
public static ConnectionSpec create(String spec) {
return create(spec, spec);
}
public static ConnectionSpec create(String localSpec, String ensembleSpec) {
return new ConnectionSpec(localSpec, ensembleSpec, ensembleSpec.split(",").length);
}
public static <T> ConnectionSpec create(List<T> servers,
Function<T, String> hostnameGetter,
Function<T, Integer> portGetter,
boolean localhostAffinity) {
String localSpec = createSpec(servers, hostnameGetter, portGetter, localhostAffinity);
String ensembleSpec = localhostAffinity ? createSpec(servers, hostnameGetter, portGetter, false) : localSpec;
return new ConnectionSpec(localSpec, ensembleSpec, servers.size());
}
private static <T> String createSpec(List<T> servers,
Function<T, String> hostnameGetter,
Function<T, Integer> portGetter,
boolean localhostAffinity) {
String thisServer = HostName.getLocalhost();
StringBuilder connectionSpec = new StringBuilder();
for (var server : servers) {
if (localhostAffinity && !thisServer.equals(hostnameGetter.apply(server))) continue;
connectionSpec.append(hostnameGetter.apply(server));
connectionSpec.append(':');
connectionSpec.append(portGetter.apply(server));
connectionSpec.append(',');
}
if (localhostAffinity && connectionSpec.length() == 0) {
throw new IllegalArgumentException("Unable to create connect string to localhost: " +
"There is no localhost server specified in config");
}
if (connectionSpec.length() > 0) {
connectionSpec.setLength(connectionSpec.length() - 1); // Remove trailing comma
}
return connectionSpec.toString();
}
private static String requireNonEmpty(String s, String field) {
if (Objects.requireNonNull(s).isEmpty()) throw new IllegalArgumentException(field + " must be non-empty");
return s;
}
}
|
