-
Notifications
You must be signed in to change notification settings - Fork 7
/
KeyValueServer.java
119 lines (92 loc) · 3.87 KB
/
KeyValueServer.java
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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
package com.cloudata.keyvalue;
import java.io.File;
import java.net.InetSocketAddress;
import java.net.SocketAddress;
import java.util.List;
import org.robotninjas.barge.ClusterConfig;
import org.robotninjas.barge.RaftService;
import org.robotninjas.barge.Replica;
import com.cloudata.keyvalue.redis.RedisEndpoint;
import com.cloudata.keyvalue.web.WebModule;
import com.google.common.collect.Lists;
import com.google.inject.Guice;
import com.google.inject.Injector;
import com.sun.grizzly.http.SelectorThread;
import com.sun.jersey.api.container.grizzly.GrizzlyServerFactory;
import com.sun.jersey.api.core.PackagesResourceConfig;
import com.sun.jersey.api.core.ResourceConfig;
import com.sun.jersey.core.spi.component.ioc.IoCComponentProviderFactory;
import com.sun.jersey.guice.spi.container.GuiceComponentProviderFactory;
public class KeyValueServer {
final File baseDir;
final int httpPort;
private final Replica local;
private final List<Replica> peers;
private RaftService raft;
private SelectorThread selector;
private final SocketAddress redisSocketAddress;
private RedisEndpoint redisEndpoint;
public KeyValueServer(File baseDir, Replica local, List<Replica> peers, int httpPort,
SocketAddress redisSocketAddress) {
this.baseDir = baseDir;
this.local = local;
this.peers = peers;
this.httpPort = httpPort;
this.redisSocketAddress = redisSocketAddress;
}
public synchronized void start() throws Exception {
if (raft != null || selector != null) {
throw new IllegalStateException();
}
File logDir = new File(baseDir, "logs");
File stateDir = new File(baseDir, "state");
logDir.mkdirs();
stateDir.mkdirs();
KeyValueStateMachine stateMachine = new KeyValueStateMachine();
ClusterConfig config = ClusterConfig.from(local, peers);
this.raft = RaftService.newBuilder(config).logDir(logDir).timeout(300).build(stateMachine);
stateMachine.init(raft, stateDir);
raft.startAsync().awaitRunning();
final String baseUri = getHttpUrl();
Injector injector = Guice.createInjector(new KeyValueModule(stateMachine), new WebModule());
ResourceConfig rc = new PackagesResourceConfig(WebModule.class.getPackage().getName());
IoCComponentProviderFactory ioc = new GuiceComponentProviderFactory(rc, injector);
this.selector = GrizzlyServerFactory.create(baseUri, rc, ioc);
if (redisSocketAddress != null) {
this.redisEndpoint = new RedisEndpoint(redisSocketAddress);
this.redisEndpoint.start();
}
}
public String getHttpUrl() {
return "http://localhost:" + httpPort + "/";
}
public static void main(String... args) throws Exception {
final int port = Integer.parseInt(args[0]);
Replica local = Replica.fromString("localhost:" + (10000 + port));
List<Replica> members = Lists.newArrayList(Replica.fromString("localhost:10001"),
Replica.fromString("localhost:10002"), Replica.fromString("localhost:10003"));
members.remove(local);
File baseDir = new File(args[0]);
int httpPort = (9990 + port);
int redisPort = 6379 + port - 1;
SocketAddress redisSocketAddress = new InetSocketAddress(redisPort);
final KeyValueServer server = new KeyValueServer(baseDir, local, members, httpPort, redisSocketAddress);
server.start();
Runtime.getRuntime().addShutdownHook(new Thread() {
@Override
public void run() {
server.stop();
}
});
}
public synchronized void stop() {
if (selector != null) {
selector.stopEndpoint();
selector = null;
}
if (raft != null) {
raft.stopAsync().awaitTerminated();
raft = null;
}
}
}