public static RpcServer createEmulatorRpcServer() {
RpcServer rpcServer = new RpcServer();
rpcServer.rpcUser = Defaults.DEFAULT_USER;
rpcServer.rpcHost = "10.0.2.2";
rpcServer.rpcPassword = Defaults.DEFAULT_PASSWORD;
rpcServer.rpcPort = Integer.valueOf(Defaults.DEFAULT_PORT);
return rpcServer;
}
public static RpcServer createDefaultRpcServer(String host) {
RpcServer rpcServer = new RpcServer();
rpcServer.rpcUser = Defaults.DEFAULT_USER;
rpcServer.rpcHost = host;
rpcServer.rpcPassword = Defaults.DEFAULT_PASSWORD;
rpcServer.rpcPort = Integer.valueOf(Defaults.DEFAULT_PORT);
return rpcServer;
}
public static RpcServer createNetHunterRpcServer() {
RpcServer rpcServer = new RpcServer();
rpcServer.rpcUser = "******";
rpcServer.rpcHost = "localhost";
rpcServer.rpcPassword = "******";
rpcServer.rpcPort = 55553;
return rpcServer;
}
/**
* Function to Launch the PathVisioRPC server
*
* @param args the port for the PathVisioRPC server to start on
*/
public static void main(String[] args) {
int port = 7777;
if (args.length > 0) {
port = Integer.parseInt(args[0]);
}
RpcServer service = new RpcServer();
service.startServer(port);
}
public static RpcServer createDefaultRpcServer() {
if (false) {
return createNetHunterRpcServer();
}
RpcServer rpcServer = new RpcServer();
rpcServer.rpcUser = Defaults.DEFAULT_USER;
rpcServer.rpcHost = Defaults.DEFAULT_HOST;
rpcServer.rpcPassword = Defaults.DEFAULT_PASSWORD;
rpcServer.rpcPort = Integer.valueOf(Defaults.DEFAULT_PORT);
rpcServer.ssl = true;
return rpcServer;
}
@Override
public void start() throws IOException {
String exchange = getApiDefinition().getName();
String queue = getApiDefinition().getName() + ".request";
Channel channel = connection.createChannel();
channel.exchangeDeclare(exchange, "fanout");
channel.queueDeclare(queue, false, false, true, null);
channel.queueBind(queue, exchange, "");
server =
new RpcServer(channel, queue) {
@Override
public byte[] handleCall(byte[] requestBody, BasicProperties replyProperties) {
Response res = rpc((Request) Serializer.deserialize(requestBody));
return Serializer.serialize(res);
}
};
server.start();
}
@Test
public void testAllocateByteBuffToReadInto() throws Exception {
int maxBuffersInPool = 10;
ByteBufferPool pool = new ByteBufferPool(6 * 1024, maxBuffersInPool);
initPoolWithAllBuffers(pool, maxBuffersInPool);
ByteBuff buff = null;
Pair<ByteBuff, CallCleanup> pair;
// When the request size is less than 1/6th of the pool buffer size. We should use on demand
// created on heap Buffer
pair =
RpcServer.allocateByteBuffToReadInto(pool, RpcServer.getMinSizeForReservoirUse(pool), 200);
buff = pair.getFirst();
assertTrue(buff.hasArray());
assertEquals(maxBuffersInPool, pool.getQueueSize());
assertNull(pair.getSecond());
// When the request size is > 1/6th of the pool buffer size.
pair =
RpcServer.allocateByteBuffToReadInto(pool, RpcServer.getMinSizeForReservoirUse(pool), 1024);
buff = pair.getFirst();
assertFalse(buff.hasArray());
assertEquals(maxBuffersInPool - 1, pool.getQueueSize());
assertNotNull(pair.getSecond());
pair.getSecond().run(); // CallCleanup#run should put back the BB to pool.
assertEquals(maxBuffersInPool, pool.getQueueSize());
// Request size> pool buffer size
pair =
RpcServer.allocateByteBuffToReadInto(
pool, RpcServer.getMinSizeForReservoirUse(pool), 7 * 1024);
buff = pair.getFirst();
assertFalse(buff.hasArray());
assertTrue(buff instanceof MultiByteBuff);
ByteBuffer[] bbs = ((MultiByteBuff) buff).getEnclosingByteBuffers();
assertEquals(2, bbs.length);
assertTrue(bbs[0].isDirect());
assertTrue(bbs[1].isDirect());
assertEquals(6 * 1024, bbs[0].limit());
assertEquals(1024, bbs[1].limit());
assertEquals(maxBuffersInPool - 2, pool.getQueueSize());
assertNotNull(pair.getSecond());
pair.getSecond().run(); // CallCleanup#run should put back the BB to pool.
assertEquals(maxBuffersInPool, pool.getQueueSize());
pair =
RpcServer.allocateByteBuffToReadInto(
pool, RpcServer.getMinSizeForReservoirUse(pool), 6 * 1024 + 200);
buff = pair.getFirst();
assertFalse(buff.hasArray());
assertTrue(buff instanceof MultiByteBuff);
bbs = ((MultiByteBuff) buff).getEnclosingByteBuffers();
assertEquals(2, bbs.length);
assertTrue(bbs[0].isDirect());
assertFalse(bbs[1].isDirect());
assertEquals(6 * 1024, bbs[0].limit());
assertEquals(200, bbs[1].limit());
assertEquals(maxBuffersInPool - 1, pool.getQueueSize());
assertNotNull(pair.getSecond());
pair.getSecond().run(); // CallCleanup#run should put back the BB to pool.
assertEquals(maxBuffersInPool, pool.getQueueSize());
ByteBuffer[] buffers = new ByteBuffer[maxBuffersInPool - 1];
for (int i = 0; i < maxBuffersInPool - 1; i++) {
buffers[i] = pool.getBuffer();
}
pair =
RpcServer.allocateByteBuffToReadInto(
pool, RpcServer.getMinSizeForReservoirUse(pool), 20 * 1024);
buff = pair.getFirst();
assertFalse(buff.hasArray());
assertTrue(buff instanceof MultiByteBuff);
bbs = ((MultiByteBuff) buff).getEnclosingByteBuffers();
assertEquals(2, bbs.length);
assertTrue(bbs[0].isDirect());
assertFalse(bbs[1].isDirect());
assertEquals(6 * 1024, bbs[0].limit());
assertEquals(14 * 1024, bbs[1].limit());
assertEquals(0, pool.getQueueSize());
assertNotNull(pair.getSecond());
pair.getSecond().run(); // CallCleanup#run should put back the BB to pool.
assertEquals(1, pool.getQueueSize());
pool.getBuffer();
pair =
RpcServer.allocateByteBuffToReadInto(
pool, RpcServer.getMinSizeForReservoirUse(pool), 7 * 1024);
buff = pair.getFirst();
assertTrue(buff.hasArray());
assertTrue(buff instanceof SingleByteBuff);
assertEquals(7 * 1024, ((SingleByteBuff) buff).getEnclosingByteBuffer().limit());
assertNull(pair.getSecond());
}
@Override
public void stop() {
server.terminateMainloop();
}
