synchronized void close() { if (null != client) { client.close(); } }
@Override public void close() { client.close(); } }
@Override public void run() { LOG.info("Started writer {}.", idx); while (running) { rateLimiter.getLimiter().acquire(batchSize); String streamName = streamNames.get(random.nextInt(numStreams)); final long requestMillis = System.currentTimeMillis(); final List<ByteBuffer> data = buildBufferList(batchSize, requestMillis, messageSizeBytes); if (null == data) { break; } List<Future<DLSN>> results = dlc.writeBulk(streamName, data); for (Future<DLSN> result : results) { result.addEventListener(new TimedRequestHandler(streamName, requestMillis)); } } dlc.close(); } }
protected void balanceFromSource(DistributedLogClientBuilder clientBuilder, ClusterBalancer balancer, String source, Optional<RateLimiter> rateLimiter) throws Exception { InetSocketAddress sourceAddr = DLSocketAddress.parseSocketAddress(source); DistributedLogClientBuilder sourceClientBuilder = DistributedLogClientBuilder.newBuilder(clientBuilder) .host(sourceAddr); Pair<DistributedLogClient, MonitorServiceClient> clientPair = ClientUtils.buildClient(sourceClientBuilder); try { Await.result(clientPair.getRight().setAcceptNewStream(false)); logger.info("Disable accepting new stream on proxy {}.", source); balancer.balanceAll(source, rebalanceConcurrency, rateLimiter); } finally { clientPair.getLeft().close(); } } }
client.close();
@Override public void run() { LOG.info("Started writer {}.", idx); while (running) { this.limiter.getLimiter().acquire(); final String streamName = streamNames.get(random.nextInt(numStreams)); final long requestMillis = System.currentTimeMillis(); final ByteBuffer data = buildBuffer(requestMillis, messageSizeBytes); if (null == data) { break; } if (null != writer) { writer.write(data).addEventListener( new TimedRequestHandler(streamName, requestMillis)); } else { dlc.write(streamName, data).addEventListener( new TimedRequestHandler(streamName, requestMillis)); } } if (null != writer) { writer.close(); } dlc.close(); } }
client.close();
client.close();
); FutureUtils.result(Future.collect(writeFutures)); client.close();
@Override public void close() throws IOException { this.running = false; for (AsyncLogReader reader : logReaders) { if (null != reader) { FutureUtils.result(reader.asyncClose()); } } for (DistributedLogManager dlm : dlms) { if (null != dlm) { dlm.close(); } } namespace.close(); SchedulerUtils.shutdownScheduler(executorService, 2, TimeUnit.MINUTES); SchedulerUtils.shutdownScheduler(callbackExecutor, 2, TimeUnit.MINUTES); if (this.dlc != null) { this.dlc.close(); } for (DLZkServerSet serverSet: serverSets) { serverSet.close(); } }
client.close();
client.close();
@Override protected int runCmd(CommandLine commandLine) throws Exception { try { parseCommandLine(commandLine); } catch (ParseException pe) { System.err.println("ERROR: failed to parse commandline : '" + pe.getMessage() + "'"); printUsage(); return -1; } DistributedLogClientBuilder clientBuilder = DistributedLogClientBuilder.newBuilder() .name("proxy_tool") .clientId(ClientId$.MODULE$.apply("proxy_tool")) .maxRedirects(2) .host(address) .clientBuilder(ClientBuilder.get() .connectionTimeout(Duration.fromSeconds(2)) .tcpConnectTimeout(Duration.fromSeconds(2)) .requestTimeout(Duration.fromSeconds(10)) .hostConnectionLimit(1) .hostConnectionCoresize(1) .keepAlive(true) .failFast(false)); Pair<DistributedLogClient, MonitorServiceClient> clientPair = ClientUtils.buildClient(clientBuilder); try { return runCmd(clientPair); } finally { clientPair.getLeft().close(); } }
return runCmd(client); } finally { client.close();