@Override public void run() { CLIENT_MANAGER.close(); }}); }
@Override protected <T> T doRefer(Class<T> type, URL url) throws RpcException { FramedClientConnector connector = new FramedClientConnector(HostAndPort.fromParts(url.getHost(), url.getPort())); try { ThriftClientManager clientManager = new ThriftClientManager(); T ref = clientManager.createClient(connector, type).get(); return ref; } catch (Exception e) { throw new RpcException( "Fail to create remoting client for service(" + url + "): " + e.getMessage(), e ); } } }
@Override public PooledObject<T> makeObject() throws Exception { T obj = getThriftClient(interfaceClass).open(getClientManager().createChannel(getConnector()).get()); return new DefaultPooledObject<T>(obj); }
@BeforeMethod(alwaysRun = true) public void setup() throws IllegalAccessException, InstantiationException, TException { codecManager = new ThriftCodecManager(); clientManager = new ThriftClientManager(codecManager); }
private RocksService getRocksClient() throws Exception { if (rocksClient.get() == null) { try { rocksClient.set(clientManager.createClient( new FramedClientConnector(fromParts(host, port)), RocksService.class).get()); logger.info("Opened Rocksdb connection to " + host + ":" + port); } catch (Exception e) { logger.error("Error in open rocksdb to " + host + ":" + port + " " + e); throw e; } } return rocksClient.get(); }
private NiftyClientChannel getChannel(PooledObject<T>p){ return (NiftyClientChannel) getClientManager().getRequestChannel(p.getObject()); } @Override
@Test public void testUnresolvedRemoteAddress() { // Test that getRemoteAddress on a client that connected to '127.0.0.1' does not resolve the IP to 'localhost' // (because doing a reverse lookup causes performance problems - e.g. for logging code) assertTrue(getClientManager().getRemoteAddress(getClient()).toString().startsWith(LOCALHOST_IP_ADDRESS)); } }
@BeforeMethod(alwaysRun = true) public void setup() throws IllegalAccessException, InstantiationException, TException { codecManager = new ThriftCodecManager(); clientManager = new ThriftClientManager(codecManager); syncServer = createSyncServer(); }
private <T> ListenableFuture<T> createUnframedClient( ThriftClientManager clientManager, Class<T> clientType, int servicePort) throws Exception { return clientManager.createClient( new UnframedClientConnector(HostAndPort.fromParts("localhost", servicePort)), clientType); }
private com.facebook.swift.service.ResultCode logSwift(int port, List<com.facebook.swift.service.LogEntry> entries) throws Exception { try ( ThriftClientManager clientManager = new ThriftClientManager(); Scribe scribe = clientManager.createClient( new FramedClientConnector(fromParts("localhost", port)), Scribe.class).get() ) { return scribe.log(entries); } }
@Override public synchronized void cleanup() throws IOException { Closeables.close(clientManager, true); try { clientManager.close(); // CHECKSTYLE: stop IllegalCatch } catch (Exception e) { // CHECKSTYLE: resume IllegalCatch if (LOG.isDebugEnabled()) { LOG.debug( "Exception occurred while trying to close JobProgressTracker", e); } } }
@BeforeMethod public void setupTest() throws Exception { // TODO: move this to setupSuite when TestNG/surefire integration is fixed handler = handlerClass.newInstance(); server = createServer(handler).start(); clientManager = new ThriftClientManager(codecManager); client = createClient(clientManager).get(); }
public DelayedMapAsyncProxyHandler(ThriftClientManager clientManager, ThriftServer targetServer) throws ExecutionException, InterruptedException { FramedClientConnector connector = new FramedClientConnector(HostAndPort.fromParts("localhost", targetServer.getPort())); asyncClient = clientManager.createClient(connector, DelayedMap.AsyncClient.class).get(); }
/** * Try to establish new connection to JobProgressTracker */ private void resetConnection() throws ExecutionException, InterruptedException { clientManager = new ThriftClientManager( new ThriftCodecManager(new ThriftCodec[0]), new NiftyClient( new NettyClientConfigBuilder().setWorkerThreadCount(2).build()), ImmutableSet.<ThriftClientEventHandler>of()); FramedClientConnector connector = new FramedClientConnector(new InetSocketAddress( ClientThriftServer.CLIENT_THRIFT_SERVER_HOST.get(conf), ClientThriftServer.CLIENT_THRIFT_SERVER_PORT.get(conf))); jobProgressTracker = clientManager.createClient(connector, JobProgressTracker.class).get(); }
@Override public void close() { try { if (!isLastThread()) { return; } if (clientManager != null) { clientManager.close(); } } catch (Exception ioex) { logger.error("Error while closing client connection: " + ioex); } }
@BeforeMethod(alwaysRun = true) private void setup() throws Exception { codecManager = new ThriftCodecManager(); clientManager = new ThriftClientManager(codecManager); syncServer = createTargetServer(TARGET_SERVER_THREAD_COUNT); asyncServer = createAsyncServer(PROXY_SERVER_THREAD_COUNT, clientManager, syncServer); }
private ListenableFuture<? extends ClientInterface> createClient(ThriftClientManager clientManager) throws TTransportException, InterruptedException, ExecutionException { HostAndPort address = HostAndPort.fromParts(serverConfig.getBindAddress(), server.getPort()); return clientManager.createClient(new FramedClientConnector(address), clientClass); }
private com.facebook.swift.service.ResultCode logSwift(int port, List<com.facebook.swift.service.LogEntry> entries) throws Exception { NettyClientConfig nettyClientConfig = NettyClientConfig.newBuilder() .setSSLClientConfiguration( new SslClientConfiguration.Builder() .caFile(new File(getClass().getResource("/rsa.crt").getFile())) .build()).build(); NiftyClient niftyClient = new NiftyClient(nettyClientConfig); try ( ThriftClientManager clientManager = new ThriftClientManager(new ThriftCodecManager(), niftyClient, ImmutableSet.of()); Scribe scribe = clientManager.createClient( new FramedClientConnector(fromParts("localhost", port)), Scribe.class).get() ) { return scribe.log(entries); } }
@AfterMethod(alwaysRun = true) public void tearDown() { clientManager.close(); } }
@Test public void testUnframedSyncMethod() throws Exception { TestServerInfo info = startServer(); ThriftClientManager clientManager = new ThriftClientManager(); try (Scribe client = createUnframedClient(clientManager, Scribe.class, info.port).get()) { ResultCode result = client.log(Lists.newArrayList( new LogEntry("testCategory", "testMessage"))); assertEquals(result, ResultCode.OK); } stopServer(info); }