blob: 56970d1807d7b63922fb6cde78c6d520664d8b47 (
plain)
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
|
package com.orbekk.same.http;
import com.googlecode.jsonrpc4j.JsonRpcServer;
import com.orbekk.net.HttpUtil;
import com.orbekk.same.UrlReceiver;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.eclipse.jetty.server.Request;
import org.eclipse.jetty.server.handler.AbstractHandler;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class RpcHandler extends AbstractHandler {
private Logger logger = LoggerFactory.getLogger(getClass());
private UrlReceiver urlReceiver;
private Map<String, JsonRpcServer> rpcServers =
new HashMap<String, JsonRpcServer>();
public RpcHandler(UrlReceiver urlReceiver) {
this.urlReceiver = urlReceiver;
}
/**
* Add an RpcServer to this Handler.
*
* @param url the base url of the service, e.g.
* /MyService.json
*/
public void addRpcServer(String url, JsonRpcServer rpcServer) {
rpcServers.put(url, rpcServer);
}
@Override
public synchronized void handle(String target, Request baseRequest,
HttpServletRequest request, HttpServletResponse response)
throws IOException, ServletException {
logger.info("Handling request to target: " + target);
if (urlReceiver != null) {
String sameServiceUrl = "http://" + request.getLocalAddr() +
":" + request.getLocalPort() + "/";
urlReceiver.setUrl(sameServiceUrl);
urlReceiver = null;
}
if (target.equals("/ping")) {
int remotePort = Integer.parseInt(request.getParameter("port"));
String pongUrl = "http://" + request.getRemoteAddr() + ":" +
remotePort + "/pong";
logger.info("Got ping. Sending pong to {}", pongUrl);
HttpUtil.sendHttpRequest(pongUrl);
baseRequest.setHandled(true);
} else if (target.equals("/pong")) {
logger.info("Received pong from {}", request.getRemoteAddr());
baseRequest.setHandled(true);
} else {
JsonRpcServer server = rpcServers.get(target);
if (server != null) {
server.handle(request, response);
baseRequest.setHandled(true);
}
}
}
}
|
