@Inject public DruidAvaticaHandler( final DruidMeta druidMeta, @Self final DruidNode druidNode, final AvaticaMonitor avaticaMonitor ) { super(new LocalService(druidMeta), avaticaMonitor); setServerRpcMetadata(new Service.RpcMetadataResponse(druidNode.getHostAndPortToUse())); }
private RpcMetadataResponse createRpcServerMetadata(ServerConnector connector) throws UnknownHostException { String host = connector.getHost(); if (null == host) { // "null" means binding to all interfaces, we need to pick one so the client gets a real // address and not "0.0.0.0" or similar. host = InetAddress.getLocalHost().getHostName(); } final int port = connector.getLocalPort(); return new RpcMetadataResponse( String.format(Locale.ROOT, "%s:%d", host, port)); }
private RpcMetadataResponse createRpcServerMetadata(ServerConnector connector) throws UnknownHostException { String host = connector.getHost(); if (null == host) { // "null" means binding to all interfaces, we need to pick one so the client gets a real // address and not "0.0.0.0" or similar. host = InetAddress.getLocalHost().getHostName(); } final int port = connector.getLocalPort(); return new RpcMetadataResponse( String.format(Locale.ROOT, "%s:%d", host, port)); }
final RpcMetadataResponse rpcMetadata = new RpcMetadataResponse("localhost:8765"); LinkedList<Response> responses = new LinkedList<>();
RpcMetadataResponse metadata = new RpcMetadataResponse("localhost:8765"); FetchResponse response = new FetchResponse(frame, false, false, metadata);
RpcMetadataResponse metadata = new RpcMetadataResponse("localhost:8765"); FetchResponse response = new FetchResponse(frame, false, false, metadata);
RpcMetadataResponse metadata = new RpcMetadataResponse("localhost:8765"); FetchResponse response = new FetchResponse(frame, false, false, metadata);
@Test public void testToClientRTE() { final String message = "There was an error"; final int code = 23; final String state = "a1b2c"; final AvaticaSeverity severity = AvaticaSeverity.ERROR; final List<String> exceptions = Arrays.asList("Server Stacktrace 1", "Server Stacktace 2"); final RpcMetadataResponse metadata = new RpcMetadataResponse("localhost:8765"); final ErrorResponse resp = new ErrorResponse(message, code, state, severity, exceptions, metadata); AvaticaClientRuntimeException exception = resp.toException(); assertTrue("Expected error message to end with '" + resp.errorMessage + "', but was '" + exception.getMessage() + "'", exception.getMessage().endsWith(resp.errorMessage)); assertEquals(resp.errorCode, exception.getErrorCode()); assertEquals(resp.severity, exception.getSeverity()); assertEquals(resp.sqlState, exception.getSqlState()); assertEquals(resp.exceptions, exception.getServerExceptions()); assertEquals(resp.rpcMetadata, exception.getRpcMetadata()); } }
@Test public void testToClientRTE() { final String message = "There was an error"; final int code = 23; final String state = "a1b2c"; final AvaticaSeverity severity = AvaticaSeverity.ERROR; final List<String> exceptions = Arrays.asList("Server Stacktrace 1", "Server Stacktace 2"); final RpcMetadataResponse metadata = new RpcMetadataResponse("localhost:8765"); final ErrorResponse resp = new ErrorResponse(message, code, state, severity, exceptions, metadata); AvaticaClientRuntimeException exception = resp.toException(); assertTrue("Expected error message to end with '" + resp.errorMessage + "', but was '" + exception.getMessage() + "'", exception.getMessage().endsWith(resp.errorMessage)); assertEquals(resp.errorCode, exception.getErrorCode()); assertEquals(resp.severity, exception.getSeverity()); assertEquals(resp.sqlState, exception.getSqlState()); assertEquals(resp.exceptions, exception.getServerExceptions()); assertEquals(resp.rpcMetadata, exception.getRpcMetadata()); } }
@Test public void testToClientRTE() { final String message = "There was an error"; final int code = 23; final String state = "a1b2c"; final AvaticaSeverity severity = AvaticaSeverity.ERROR; final List<String> exceptions = Arrays.asList("Server Stacktrace 1", "Server Stacktace 2"); final RpcMetadataResponse metadata = new RpcMetadataResponse("localhost:8765"); final ErrorResponse resp = new ErrorResponse(message, code, state, severity, exceptions, metadata); AvaticaClientRuntimeException exception = resp.toException(); assertTrue("Expected error message to end with '" + resp.errorMessage + "', but was '" + exception.getMessage() + "'", exception.getMessage().endsWith(resp.errorMessage)); assertEquals(resp.errorCode, exception.getErrorCode()); assertEquals(resp.severity, exception.getSeverity()); assertEquals(resp.sqlState, exception.getSqlState()); assertEquals(resp.exceptions, exception.getServerExceptions()); assertEquals(resp.rpcMetadata, exception.getRpcMetadata()); } }
@Test public void testGetters() { final String errorMsg = "My error message"; final int errorCode = 10; final String sqlState = "abc12"; final AvaticaSeverity severity = AvaticaSeverity.ERROR; final List<String> stacktraces = Arrays.asList("my stack trace"); final RpcMetadataResponse metadata = new RpcMetadataResponse("localhost:8765"); AvaticaClientRuntimeException e = new AvaticaClientRuntimeException(errorMsg, errorCode, sqlState, severity, stacktraces, metadata); assertEquals(errorMsg, e.getMessage()); assertEquals(errorCode, e.getErrorCode()); assertEquals(severity, e.getSeverity()); assertEquals(stacktraces, e.getServerExceptions()); assertEquals(metadata, e.getRpcMetadata()); }
@Test public void testGetters() { final String errorMsg = "My error message"; final int errorCode = 10; final String sqlState = "abc12"; final AvaticaSeverity severity = AvaticaSeverity.ERROR; final List<String> stacktraces = Arrays.asList("my stack trace"); final RpcMetadataResponse metadata = new RpcMetadataResponse("localhost:8765"); AvaticaClientRuntimeException e = new AvaticaClientRuntimeException(errorMsg, errorCode, sqlState, severity, stacktraces, metadata); assertEquals(errorMsg, e.getMessage()); assertEquals(errorCode, e.getErrorCode()); assertEquals(severity, e.getSeverity()); assertEquals(stacktraces, e.getServerExceptions()); assertEquals(metadata, e.getRpcMetadata()); }
@Test public void testGetters() { final String errorMsg = "My error message"; final int errorCode = 10; final String sqlState = "abc12"; final AvaticaSeverity severity = AvaticaSeverity.ERROR; final List<String> stacktraces = Arrays.asList("my stack trace"); final RpcMetadataResponse metadata = new RpcMetadataResponse("localhost:8765"); AvaticaClientRuntimeException e = new AvaticaClientRuntimeException(errorMsg, errorCode, sqlState, severity, stacktraces, metadata); assertEquals(errorMsg, e.getMessage()); assertEquals(errorCode, e.getErrorCode()); assertEquals(severity, e.getSeverity()); assertEquals(stacktraces, e.getServerExceptions()); assertEquals(metadata, e.getRpcMetadata()); }
static RpcMetadataResponse fromProto(Responses.RpcMetadata msg) { String serverAddress = null; if (msg.hasField(SERVER_ADDRESS_DESCRIPTOR)) { serverAddress = msg.getServerAddress(); } return new RpcMetadataResponse(serverAddress); }
@Test public void testEquality() { final String message = "There was an error"; final int code = 23; final String state = "a1b2c"; final AvaticaSeverity severity = AvaticaSeverity.ERROR; final List<String> exceptions = Arrays.asList("Server Stacktrace 1", "Server Stacktace 2"); final RpcMetadataResponse metadata = new RpcMetadataResponse("localhost:8765"); assertEquals(new ErrorResponse(message, code, state, severity, exceptions, metadata), new ErrorResponse(message, code, state, severity, exceptions, metadata)); }
@Test public void testEquality() { final String message = "There was an error"; final int code = 23; final String state = "a1b2c"; final AvaticaSeverity severity = AvaticaSeverity.ERROR; final List<String> exceptions = Arrays.asList("Server Stacktrace 1", "Server Stacktace 2"); final RpcMetadataResponse metadata = new RpcMetadataResponse("localhost:8765"); assertEquals(new ErrorResponse(message, code, state, severity, exceptions, metadata), new ErrorResponse(message, code, state, severity, exceptions, metadata)); }
@Inject public DruidAvaticaHandler( final DruidMeta druidMeta, @Self final DruidNode druidNode, final AvaticaMonitor avaticaMonitor ) { super(new LocalService(druidMeta), avaticaMonitor); setServerRpcMetadata(new Service.RpcMetadataResponse(druidNode.getHostAndPortToUse())); }
@Test public void testEquality() { final String message = "There was an error"; final int code = 23; final String state = "a1b2c"; final AvaticaSeverity severity = AvaticaSeverity.ERROR; final List<String> exceptions = Arrays.asList("Server Stacktrace 1", "Server Stacktace 2"); final RpcMetadataResponse metadata = new RpcMetadataResponse("localhost:8765"); assertEquals(new ErrorResponse(message, code, state, severity, exceptions, metadata), new ErrorResponse(message, code, state, severity, exceptions, metadata)); }
static RpcMetadataResponse fromProto(Responses.RpcMetadata msg) { final Descriptor desc = msg.getDescriptorForType(); String serverAddress = null; if (ProtobufService.hasField(msg, desc, Responses.RpcMetadata.SERVER_ADDRESS_FIELD_NUMBER)) { serverAddress = msg.getServerAddress(); } return new RpcMetadataResponse(serverAddress); }
static RpcMetadataResponse fromProto(Responses.RpcMetadata msg) { String serverAddress = null; if (msg.hasField(SERVER_ADDRESS_DESCRIPTOR)) { serverAddress = msg.getServerAddress(); } return new RpcMetadataResponse(serverAddress); }