/*
    * Licensed to the Apache Software Foundation (ASF) under one
    * or more contributor license agreements.  See the NOTICE file
    * distributed with this work for additional information
    * regarding copyright ownership.  The ASF licenses this file
    * to you under the Apache License, Version 2.0 (the
    * "License"); you may not use this file except in compliance
    * with the License.  You may obtain a copy of the License at
    *
   * http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  
  package org.apache.hadoop.hbase.coprocessor;
  
  import com.google.protobuf.RpcCallback;
  import com.google.protobuf.RpcController;
  import com.google.protobuf.Service;
  import org.apache.hadoop.hbase.Coprocessor;
  import org.apache.hadoop.hbase.CoprocessorEnvironment;
  import org.apache.hadoop.hbase.ipc.RpcServer;
  import org.apache.hadoop.hbase.ipc.protobuf.generated.TestProtos;
  import org.apache.hadoop.hbase.ipc.protobuf.generated.TestProtos.AddrResponseProto;
  import org.apache.hadoop.hbase.ipc.protobuf.generated.TestProtos.EmptyRequestProto;
  import org.apache.hadoop.hbase.ipc.protobuf.generated.TestProtos.EmptyResponseProto;
  import org.apache.hadoop.hbase.ipc.protobuf.generated.TestProtos.PauseRequestProto;
  import org.apache.hadoop.hbase.ipc.protobuf.generated.TestRpcServiceProtos;
  import org.apache.hadoop.hbase.protobuf.ResponseConverter;
  import org.apache.hadoop.hbase.util.Threads;
  
  import java.io.IOException;
  
  /**
   * Test implementation of a coprocessor endpoint exposing the
   * {@link TestRpcServiceProtos.TestProtobufRpcProto} service methods. For internal use by unit tests
   * only.
   */
  public class ProtobufCoprocessorService extends TestRpcServiceProtos.TestProtobufRpcProto
      implements CoprocessorService, Coprocessor {
    public ProtobufCoprocessorService() {
    }
  
    @Override
    public Service getService() {
      return this;
    }
  
    @Override
    public void ping(RpcController controller, TestProtos.EmptyRequestProto request,
        RpcCallback<TestProtos.EmptyResponseProto> done) {
      done.run(TestProtos.EmptyResponseProto.getDefaultInstance());
    }
  
    @Override
    public void echo(RpcController controller, TestProtos.EchoRequestProto request,
        RpcCallback<TestProtos.EchoResponseProto> done) {
      String message = request.getMessage();
      done.run(TestProtos.EchoResponseProto.newBuilder().setMessage(message).build());
    }
  
    @Override
    public void error(RpcController controller, TestProtos.EmptyRequestProto request,
        RpcCallback<TestProtos.EmptyResponseProto> done) {
      ResponseConverter.setControllerException(controller, new IOException("Test exception"));
      done.run(null);
    }
  
    @Override
    public void pause(RpcController controller, PauseRequestProto request,
        RpcCallback<EmptyResponseProto> done) {
      Threads.sleep(request.getMs());
      done.run(EmptyResponseProto.getDefaultInstance());
    }
  
    @Override
    public void addr(RpcController controller, EmptyRequestProto request,
        RpcCallback<AddrResponseProto> done) {
      done.run(AddrResponseProto.newBuilder().setAddr(RpcServer.getRemoteAddress().getHostAddress())
          .build());
    }
  
    @Override
    public void start(CoprocessorEnvironment env) throws IOException {
      // To change body of implemented methods use File | Settings | File Templates.
    }
  
    @Override
    public void stop(CoprocessorEnvironment env) throws IOException {
      // To change body of implemented methods use File | Settings | File Templates.
    }
  
  }