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 19 package org.apache.hadoop.yarn.server.api.impl.pb.client; 20 21 import java.io.Closeable; 22 import java.io.IOException; 23 import java.net.InetSocketAddress; 24 25 import org.apache.hadoop.conf.Configuration; 26 import org.apache.hadoop.ipc.ProtobufRpcEngine; 27 import org.apache.hadoop.ipc.RPC; 28 import org.apache.hadoop.yarn.server.api.SCMAdminProtocol; 29 import org.apache.hadoop.yarn.server.api.SCMAdminProtocolPB; 30 import org.apache.hadoop.yarn.server.api.protocolrecords.impl.pb.RunSharedCacheCleanerTaskRequestPBImpl; 31 import org.apache.hadoop.yarn.server.api.protocolrecords.impl.pb.RunSharedCacheCleanerTaskResponsePBImpl; 32 import org.apache.hadoop.yarn.server.api.protocolrecords.RunSharedCacheCleanerTaskRequest; 33 import org.apache.hadoop.yarn.server.api.protocolrecords.RunSharedCacheCleanerTaskResponse; 34 import org.apache.hadoop.yarn.exceptions.YarnException; 35 import org.apache.hadoop.yarn.ipc.RPCUtil; 36 import org.apache.hadoop.yarn.proto.YarnServiceProtos; 37 38 import com.google.protobuf.ServiceException; 39 40 public class SCMAdminProtocolPBClientImpl implements SCMAdminProtocol, 41 Closeable { 42 43 private SCMAdminProtocolPB proxy; 44 SCMAdminProtocolPBClientImpl(long clientVersion, InetSocketAddress addr, Configuration conf)45 public SCMAdminProtocolPBClientImpl(long clientVersion, 46 InetSocketAddress addr, Configuration conf) throws IOException { 47 RPC.setProtocolEngine(conf, SCMAdminProtocolPB.class, 48 ProtobufRpcEngine.class); 49 proxy = RPC.getProxy(SCMAdminProtocolPB.class, clientVersion, addr, conf); 50 } 51 52 @Override close()53 public void close() { 54 if (this.proxy != null) { 55 RPC.stopProxy(this.proxy); 56 } 57 } 58 59 @Override runCleanerTask( RunSharedCacheCleanerTaskRequest request)60 public RunSharedCacheCleanerTaskResponse runCleanerTask( 61 RunSharedCacheCleanerTaskRequest request) throws YarnException, 62 IOException { 63 YarnServiceProtos.RunSharedCacheCleanerTaskRequestProto requestProto = 64 ((RunSharedCacheCleanerTaskRequestPBImpl) request).getProto(); 65 try { 66 return new RunSharedCacheCleanerTaskResponsePBImpl(proxy.runCleanerTask(null, 67 requestProto)); 68 } catch (ServiceException e) { 69 RPCUtil.unwrapAndThrowException(e); 70 return null; 71 } 72 } 73 } 74