blob: 55968fe3ea5aa3c9119a3407c505c23d56c543bd (
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
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
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.model;
import static com.yahoo.config.provision.NetworkPorts.Allocation;
import java.util.Collection;
import java.util.HashMap;
import java.util.Map;
import java.util.TreeMap;
import java.util.logging.Logger;
import java.util.logging.Level;
public class PortFinder {
private static final Logger log = Logger.getLogger(PortFinder.class.getName());
private final Map<String, Allocation> byKeys = new HashMap<>();
private final Map<Integer, Allocation> byPorts = new TreeMap<>();
/** force add the given allocation, removing any conflicting ones */
public void use(Allocation allocation) {
String key = allocation.key();
if (byKeys.containsKey(key)) {
if (byKeys.get(key).port == allocation.port) {
return; // already OK
}
Allocation toRemove = byKeys.remove(key);
byPorts.remove(toRemove.port);
}
if (byPorts.containsKey(allocation.port)) {
Allocation toRemove = byPorts.remove(allocation.port);
byKeys.remove(toRemove.key());
}
byPorts.put(allocation.port, allocation);
byKeys.put(key, allocation);
}
public int findPort(Allocation request, String host) {
String key = request.key();
if (byKeys.containsKey(key)) {
int port = byKeys.get(key).port;
log.log(Level.FINE, () -> "Re-using port "+port+" for allocation " + request + " on " + host);
return port;
}
int port = request.port;
while (byPorts.containsKey(port)) {
++port;
}
return port;
}
public boolean isFree(int port) {
return !byPorts.containsKey(port);
}
public PortFinder(Collection<Allocation> allocations) {
for (Allocation a : allocations) {
use(a);
}
}
public Collection<Allocation> allocations() {
return byPorts.values();
}
}
|