@Override
public void clientSendRequest(RPCContext context) {
ByteBuffer buf = ByteBuffer.wrap("ap".getBytes());
context.requestCallMeta().put(key, buf);
Assert.assertNotNull(context.getMessage());
Assert.assertNotNull(context.getRequestPayload());
Assert.assertNull(context.getResponsePayload());
}
@Override
public void serverSendResponse(RPCContext context) {
Assert.assertNotNull(context.requestCallMeta());
Assert.assertNotNull(context.responseCallMeta());
Assert.assertNotNull(context.getResponsePayload());
if (!context.requestCallMeta().containsKey(key)) return;
ByteBuffer buf = context.requestCallMeta().get(key);
Assert.assertNotNull(buf);
Assert.assertNotNull(buf.array());
String partialstr = new String(buf.array());
Assert.assertNotNull(partialstr);
Assert.assertEquals("partial string mismatch", "apa", partialstr);
buf = ByteBuffer.wrap((partialstr + "c").getBytes());
Assert.assertTrue(buf.remaining() > 0);
context.responseCallMeta().put(key, buf);
}
@Override
public void serverReceiveRequest(RPCContext context) {
Map<String, ByteBuffer> meta = context.requestCallMeta();
Assert.assertNotNull(meta);
Assert.assertNotNull(context.getMessage());
Assert.assertNull(context.getResponsePayload());
if (!meta.containsKey(key)) return;
ByteBuffer buf = meta.get(key);
Assert.assertNotNull(buf);
Assert.assertNotNull(buf.array());
String partialstr = new String(buf.array());
Assert.assertNotNull(partialstr);
Assert.assertEquals("partial string mismatch", "ap", partialstr);
buf = ByteBuffer.wrap((partialstr + "a").getBytes());
Assert.assertTrue(buf.remaining() > 0);
meta.put(key, buf);
}
@Override
public void serverReceiveRequest(RPCContext context) {
avroHeadersToExchangeHeaders(context.requestCallMeta());
}
@Override
public void clientReceiveResponse(RPCContext context) {
avroHeadersToExchangeHeaders(context.requestCallMeta());
}
@Override
public void clientSendRequest(RPCContext context) {
exchangeHeadersToAvroHeaders(context.requestCallMeta());
}