public Future<DLSN> write(VALUE data) { return service.apply(data); }
public Future<BoxedUnit> close() { return client.close(); } }
public void run() throws Exception { PinLaterClient client = new PinLaterClient(cmdLine); final PinLater.ServiceIface iface = client.getIface(); if (mode.equals("create")) { createQueue(iface); } else if (mode.equals("check_dequeue")) { checkDequeue(iface); } else if (mode.equals("lookup")) { lookupJobs(iface); } else if (mode.equals("get_job_count")) { getJobCount(iface); } else if (mode.equals("get_queue_names")) { getQueueNames(iface); } else if (mode.equals("enqueue")) { issueEnqueueRequests(iface); } else { Preconditions.checkArgument(mode.equals("dequeue")); issueDequeueAckRequests(iface); } client.getService().release(); }
@Override public Status status() { return delegate.status(); }
client.release(); // Close down the clinet System.out.println("Everybody is done, let's see what they got: "); for (BlockingCallResponse blockingCallResponse : responses) {
@Override public Future<Rep> apply(Req req, Service<Req, Rep> service) { Future<Rep> result = null; outstandingAsync.inc(); final Stopwatch stopwatch = Stopwatch.createStarted(); try { result = service.apply(req); serviceExec.registerSuccessfulEvent(stopwatch.stop().elapsed(TimeUnit.MICROSECONDS)); } finally { outstandingAsync.dec(); if (null == result) { serviceExec.registerFailedEvent(stopwatch.stop().elapsed(TimeUnit.MICROSECONDS)); } } return result; }
public Future<BoxedUnit> closeFuture() { if (closeCalled) return null; closeCalled = true; return client.close(); } }
public Future<ServerInfo> handshake() { try { // TODO: size TMemoryBuffer __memoryTransport__ = new TMemoryBuffer(512); TProtocol __prot__ = this.protocolFactory.getProtocol(__memoryTransport__); __prot__.writeMessageBegin(new TMessage("handshake", TMessageType.CALL, 0)); handshake_args __args__ = new handshake_args(); __args__.write(__prot__); __prot__.writeMessageEnd(); byte[] __buffer__ = Arrays.copyOfRange(__memoryTransport__.getArray(), 0, __memoryTransport__.length()); ThriftClientRequest __request__ = new ThriftClientRequest(__buffer__, false); Future<byte[]> __done__ = this.service.apply(__request__); return __done__.flatMap(new Function<byte[], Future<ServerInfo>>() { public Future<ServerInfo> apply(byte[] __buffer__) { TMemoryInputTransport __memoryTransport__ = new TMemoryInputTransport(__buffer__); TProtocol __prot__ = ServiceToClient.this.protocolFactory.getProtocol(__memoryTransport__); try { return Future.value((new Client(__prot__)).recv_handshake()); } catch (Exception e) { return Future.exception(e); } } }); } catch (TException e) { return Future.exception(e); } } public Future<ServerInfo> handshakeWithClientInfo(ClientInfo clientInfo) {
public Future<BoxedUnit> closeFuture() { if (closeCalled) return null; closeCalled = true; return client.close(); }
public Future<Void> setAcceptNewStream(boolean enabled) { try { // TODO: size TMemoryBuffer __memoryTransport__ = new TMemoryBuffer(512); TProtocol __prot__ = this.protocolFactory.getProtocol(__memoryTransport__); __prot__.writeMessageBegin(new TMessage("setAcceptNewStream", TMessageType.CALL, 0)); setAcceptNewStream_args __args__ = new setAcceptNewStream_args(); __args__.setEnabled(enabled); __args__.write(__prot__); __prot__.writeMessageEnd(); byte[] __buffer__ = Arrays.copyOfRange(__memoryTransport__.getArray(), 0, __memoryTransport__.length()); ThriftClientRequest __request__ = new ThriftClientRequest(__buffer__, false); Future<byte[]> __done__ = this.service.apply(__request__); return __done__.flatMap(new Function<byte[], Future<Void>>() { public Future<Void> apply(byte[] __buffer__) { TMemoryInputTransport __memoryTransport__ = new TMemoryInputTransport(__buffer__); TProtocol __prot__ = ServiceToClient.this.protocolFactory.getProtocol(__memoryTransport__); try { (new Client(__prot__)).recv_setAcceptNewStream(); return Future.value(null); } catch (Exception e) { return Future.exception(e); } } }); } catch (TException e) { return Future.exception(e); } } }
public Future<BoxedUnit> closeFuture() { if (closeCalled) return null; closeCalled = true; return client.close(); }
public Future<ServerInfo> handshakeWithClientInfo(ClientInfo clientInfo) { try { // TODO: size TMemoryBuffer __memoryTransport__ = new TMemoryBuffer(512); TProtocol __prot__ = this.protocolFactory.getProtocol(__memoryTransport__); __prot__.writeMessageBegin(new TMessage("handshakeWithClientInfo", TMessageType.CALL, 0)); handshakeWithClientInfo_args __args__ = new handshakeWithClientInfo_args(); __args__.setClientInfo(clientInfo); __args__.write(__prot__); __prot__.writeMessageEnd(); byte[] __buffer__ = Arrays.copyOfRange(__memoryTransport__.getArray(), 0, __memoryTransport__.length()); ThriftClientRequest __request__ = new ThriftClientRequest(__buffer__, false); Future<byte[]> __done__ = this.service.apply(__request__); return __done__.flatMap(new Function<byte[], Future<ServerInfo>>() { public Future<ServerInfo> apply(byte[] __buffer__) { TMemoryInputTransport __memoryTransport__ = new TMemoryInputTransport(__buffer__); TProtocol __prot__ = ServiceToClient.this.protocolFactory.getProtocol(__memoryTransport__); try { return Future.value((new Client(__prot__)).recv_handshakeWithClientInfo()); } catch (Exception e) { return Future.exception(e); } } }); } catch (TException e) { return Future.exception(e); } } public Future<WriteResponse> heartbeat(String stream, WriteContext ctx) {
public Future<BoxedUnit> close() { return client.close(); } }
public Future<WriteResponse> heartbeat(String stream, WriteContext ctx) { try { // TODO: size TMemoryBuffer __memoryTransport__ = new TMemoryBuffer(512); TProtocol __prot__ = this.protocolFactory.getProtocol(__memoryTransport__); __prot__.writeMessageBegin(new TMessage("heartbeat", TMessageType.CALL, 0)); heartbeat_args __args__ = new heartbeat_args(); __args__.setStream(stream); __args__.setCtx(ctx); __args__.write(__prot__); __prot__.writeMessageEnd(); byte[] __buffer__ = Arrays.copyOfRange(__memoryTransport__.getArray(), 0, __memoryTransport__.length()); ThriftClientRequest __request__ = new ThriftClientRequest(__buffer__, false); Future<byte[]> __done__ = this.service.apply(__request__); return __done__.flatMap(new Function<byte[], Future<WriteResponse>>() { public Future<WriteResponse> apply(byte[] __buffer__) { TMemoryInputTransport __memoryTransport__ = new TMemoryInputTransport(__buffer__); TProtocol __prot__ = ServiceToClient.this.protocolFactory.getProtocol(__memoryTransport__); try { return Future.value((new Client(__prot__)).recv_heartbeat()); } catch (Exception e) { return Future.exception(e); } } }); } catch (TException e) { return Future.exception(e); } } public Future<WriteResponse> heartbeatWithOptions(String stream, WriteContext ctx, HeartbeatOptions options) {
@Override public Future<BoxedUnit> close(Time deadline) { return client.close(deadline); }
public Future<WriteResponse> delete(String stream, WriteContext ctx) { try { // TODO: size TMemoryBuffer __memoryTransport__ = new TMemoryBuffer(512); TProtocol __prot__ = this.protocolFactory.getProtocol(__memoryTransport__); __prot__.writeMessageBegin(new TMessage("delete", TMessageType.CALL, 0)); delete_args __args__ = new delete_args(); __args__.setStream(stream); __args__.setCtx(ctx); __args__.write(__prot__); __prot__.writeMessageEnd(); byte[] __buffer__ = Arrays.copyOfRange(__memoryTransport__.getArray(), 0, __memoryTransport__.length()); ThriftClientRequest __request__ = new ThriftClientRequest(__buffer__, false); Future<byte[]> __done__ = this.service.apply(__request__); return __done__.flatMap(new Function<byte[], Future<WriteResponse>>() { public Future<WriteResponse> apply(byte[] __buffer__) { TMemoryInputTransport __memoryTransport__ = new TMemoryInputTransport(__buffer__); TProtocol __prot__ = ServiceToClient.this.protocolFactory.getProtocol(__memoryTransport__); try { return Future.value((new Client(__prot__)).recv_delete()); } catch (Exception e) { return Future.exception(e); } } }); } catch (TException e) { return Future.exception(e); } } public Future<Void> setAcceptNewStream(boolean enabled) {
public Future<BoxedUnit> close() { return client.close(); } }
public Future<WriteResponse> release(String stream, WriteContext ctx) { try { // TODO: size TMemoryBuffer __memoryTransport__ = new TMemoryBuffer(512); TProtocol __prot__ = this.protocolFactory.getProtocol(__memoryTransport__); __prot__.writeMessageBegin(new TMessage("release", TMessageType.CALL, 0)); release_args __args__ = new release_args(); __args__.setStream(stream); __args__.setCtx(ctx); __args__.write(__prot__); __prot__.writeMessageEnd(); byte[] __buffer__ = Arrays.copyOfRange(__memoryTransport__.getArray(), 0, __memoryTransport__.length()); ThriftClientRequest __request__ = new ThriftClientRequest(__buffer__, false); Future<byte[]> __done__ = this.service.apply(__request__); return __done__.flatMap(new Function<byte[], Future<WriteResponse>>() { public Future<WriteResponse> apply(byte[] __buffer__) { TMemoryInputTransport __memoryTransport__ = new TMemoryInputTransport(__buffer__); TProtocol __prot__ = ServiceToClient.this.protocolFactory.getProtocol(__memoryTransport__); try { return Future.value((new Client(__prot__)).recv_release()); } catch (Exception e) { return Future.exception(e); } } }); } catch (TException e) { return Future.exception(e); } } public Future<WriteResponse> create(String stream, WriteContext ctx) {
public Future<BoxedUnit> close() { return client.close(); } }
public Future<WriteResponse> write(String stream, ByteBuffer data) { try { // TODO: size TMemoryBuffer __memoryTransport__ = new TMemoryBuffer(512); TProtocol __prot__ = this.protocolFactory.getProtocol(__memoryTransport__); __prot__.writeMessageBegin(new TMessage("write", TMessageType.CALL, 0)); write_args __args__ = new write_args(); __args__.setStream(stream); __args__.setData(data); __args__.write(__prot__); __prot__.writeMessageEnd(); byte[] __buffer__ = Arrays.copyOfRange(__memoryTransport__.getArray(), 0, __memoryTransport__.length()); ThriftClientRequest __request__ = new ThriftClientRequest(__buffer__, false); Future<byte[]> __done__ = this.service.apply(__request__); return __done__.flatMap(new Function<byte[], Future<WriteResponse>>() { public Future<WriteResponse> apply(byte[] __buffer__) { TMemoryInputTransport __memoryTransport__ = new TMemoryInputTransport(__buffer__); TProtocol __prot__ = ServiceToClient.this.protocolFactory.getProtocol(__memoryTransport__); try { return Future.value((new Client(__prot__)).recv_write()); } catch (Exception e) { return Future.exception(e); } } }); } catch (TException e) { return Future.exception(e); } } public Future<WriteResponse> writeWithContext(String stream, ByteBuffer data, WriteContext ctx) {