001/* 002 * Licensed to the Apache Software Foundation (ASF) under one 003 * or more contributor license agreements. See the NOTICE file 004 * distributed with this work for additional information 005 * regarding copyright ownership. The ASF licenses this file 006 * to you under the Apache License, Version 2.0 (the 007 * "License"); you may not use this file except in compliance 008 * with the License. You may obtain a copy of the License at 009 * 010 * http://www.apache.org/licenses/LICENSE-2.0 011 * 012 * Unless required by applicable law or agreed to in writing, software 013 * distributed under the License is distributed on an "AS IS" BASIS, 014 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 015 * See the License for the specific language governing permissions and 016 * limitations under the License. 017 */ 018package org.apache.hadoop.hbase.client; 019 020import java.io.IOException; 021import org.apache.hadoop.hbase.ServerName; 022import org.apache.hadoop.hbase.TableName; 023import org.apache.yetus.audience.InterfaceAudience; 024 025import org.apache.hbase.thirdparty.com.google.protobuf.RpcController; 026 027import org.apache.hadoop.hbase.shaded.protobuf.generated.ClientProtos; 028 029/** 030 * A RegionServerCallable set to use the Client protocol. Also includes some utility methods so can 031 * hide protobuf references here rather than have them sprinkled about the code base. 032 */ 033@InterfaceAudience.Private 034public abstract class ClientServiceCallable<T> 035 extends RegionServerCallable<T, ClientProtos.ClientService.BlockingInterface> { 036 037 public ClientServiceCallable(Connection connection, TableName tableName, byte[] row, 038 RpcController rpcController, int priority) { 039 super(connection, tableName, row, rpcController, priority); 040 } 041 042 @Override 043 protected void setStubByServiceName(ServerName serviceName) throws IOException { 044 setStub(getConnection().getClient(serviceName)); 045 } 046 047 // Below here are simple methods that contain the stub and the rpcController. 048 protected ClientProtos.GetResponse doGet(ClientProtos.GetRequest request) 049 throws org.apache.hbase.thirdparty.com.google.protobuf.ServiceException { 050 return getStub().get(getRpcController(), request); 051 } 052 053 protected ClientProtos.MutateResponse doMutate(ClientProtos.MutateRequest request) 054 throws org.apache.hbase.thirdparty.com.google.protobuf.ServiceException { 055 return getStub().mutate(getRpcController(), request); 056 } 057}