View Javadoc

1   /*
2    * Licensed to the Apache Software Foundation (ASF) under one
3    * or more contributor license agreements.  See the NOTICE file
4    * distributed with this work for additional information
5    * regarding copyright ownership.  The ASF licenses this file
6    * to you under the Apache License, Version 2.0 (the
7    * "License"); you may not use this file except in compliance
8    * with the License.  You may obtain a copy of the License at
9    *
10   *   http://www.apache.org/licenses/LICENSE-2.0
11   *
12   * Unless required by applicable law or agreed to in writing, software
13   * distributed under the License is distributed on an "AS IS" BASIS,
14   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15   * See the License for the specific language governing permissions and
16   * limitations under the License.
17   */
18  package org.apache.omid.tso.client;
19  
20  import org.apache.omid.proto.TSOProto;
21  import org.apache.omid.proto.TSOProto.Response;
22  import org.slf4j.Logger;
23  import org.slf4j.LoggerFactory;
24  
25  import java.util.concurrent.ExecutionException;
26  
27  /**
28   * Communication endpoint for TSO clients.
29   */
30  public class TSOClientOneShot {
31  
32      private static final Logger LOG = LoggerFactory.getLogger(TSOClientOneShot.class);
33  
34      private final String host;
35      private final int port;
36  
37      public TSOClientOneShot(String host, int port) {
38  
39          this.host = host;
40          this.port = port;
41  
42      }
43  
44      public TSOProto.Response makeRequest(TSOProto.Request request)
45              throws InterruptedException, ExecutionException {
46          TSOClientRaw raw = new TSOClientRaw(host, port);
47  
48          // do handshake
49          TSOProto.HandshakeRequest.Builder handshake = TSOProto.HandshakeRequest.newBuilder();
50          handshake.setClientCapabilities(TSOProto.Capabilities.newBuilder().build());
51          raw.write(TSOProto.Request.newBuilder()
52                  .setHandshakeRequest(handshake.build()).build());
53          Response response = raw.getResponse().get();
54          assert (response.getHandshakeResponse().getClientCompatible());
55  
56          raw.write(request);
57          response = raw.getResponse().get();
58  
59          raw.close();
60          return response;
61      }
62  
63  }