blob: 6e0c3d1962ec4326921b1c373462f84f6b79a9df (
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
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.messagebus.network.local;
import com.yahoo.jrt.slobrok.api.IMirror;
import com.yahoo.jrt.slobrok.api.Mirror;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.regex.Pattern;
/**
* @author Simon Thoresen Hult
*/
public class LocalWire implements IMirror {
private final AtomicInteger serviceId = new AtomicInteger();
private final AtomicInteger updateCnt = new AtomicInteger();
private final ConcurrentHashMap<String, LocalNetwork> services = new ConcurrentHashMap<>();
public void registerService(String serviceName, LocalNetwork owner) {
if (services.putIfAbsent(serviceName, owner) != null) {
throw new IllegalStateException();
}
updateCnt.incrementAndGet();
}
public void unregisterService(String serviceName) {
services.remove(serviceName);
updateCnt.incrementAndGet();
}
public LocalServiceAddress resolveServiceAddress(String serviceName) {
final LocalNetwork owner = services.get(serviceName);
return owner != null ? new LocalServiceAddress(serviceName, owner) : null;
}
public String newHostId() {
return "tcp/local:" + serviceId.getAndIncrement();
}
@Override
public List<Mirror.Entry> lookup(String pattern) {
List<Mirror.Entry> out = new ArrayList<>();
Pattern regex = Pattern.compile(pattern.replace("*", "[a-zA-Z0-9_-]+"));
for (String key : services.keySet()) {
if (regex.matcher(key).matches()) {
out.add(new Mirror.Entry(key, key));
}
}
return out;
}
@Override
public int updates() {
return updateCnt.get();
}
}
|