diff options
Diffstat (limited to 'exec/java-exec/src/main/java/org/apache/drill/exec/rpc/BasicServer.java')
-rw-r--r-- | exec/java-exec/src/main/java/org/apache/drill/exec/rpc/BasicServer.java | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/exec/java-exec/src/main/java/org/apache/drill/exec/rpc/BasicServer.java b/exec/java-exec/src/main/java/org/apache/drill/exec/rpc/BasicServer.java index f5e77f10c..8f533e3cf 100644 --- a/exec/java-exec/src/main/java/org/apache/drill/exec/rpc/BasicServer.java +++ b/exec/java-exec/src/main/java/org/apache/drill/exec/rpc/BasicServer.java @@ -73,7 +73,7 @@ public abstract class BasicServer<T extends EnumLite, C extends RemoteConnection ch.closeFuture().addListener(getCloseHandler(connection)); ch.pipeline().addLast( // - getDecoder(connection.getAllocator()), // + getDecoder(connection.getAllocator(), getOutOfMemoryHandler()), // new RpcDecoder("s-" + rpcConfig.getName()), // new RpcEncoder("s-" + rpcConfig.getName()), // getHandshakeHandler(connection), new InboundHandler(connection), // @@ -85,7 +85,11 @@ public abstract class BasicServer<T extends EnumLite, C extends RemoteConnection }); } - public abstract ProtobufLengthDecoder getDecoder(BufferAllocator allocator); + public OutOfMemoryHandler getOutOfMemoryHandler() { + return OutOfMemoryHandler.DEFAULT_INSTANCE; + } + + public abstract ProtobufLengthDecoder getDecoder(BufferAllocator allocator, OutOfMemoryHandler outOfMemoryHandler); @Override public boolean isClient() { |