Example usage for org.apache.hadoop.ipc RPC setProtocolEngine

Introduction

In this page you can find the example usage for org.apache.hadoop.ipc RPC setProtocolEngine.

Prototype

public static void setProtocolEngine(Configuration conf, Class<?> protocol, Class<?> engine) 

Source Link

Document

Set a protocol to use a non-default RpcEngine.

Usage

From source file:org.apache.slider.server.appmaster.rpc.RpcBinder.java

License:Apache License

/**
 * Add the protobuf engine to the configuration. Harmless and inexpensive
 * if repeated/* ww  w  . java 2 s .  c  o  m*/
 * @param conf configuration to patch
 * @return the protocol class
 */
public static Class<SliderClusterProtocolPB> registerSliderAPI(Configuration conf) {
    Class<SliderClusterProtocolPB> sliderClusterAPIClass = SliderClusterProtocolPB.class;
    RPC.setProtocolEngine(conf, sliderClusterAPIClass, ProtobufRpcEngine.class);

    //quick sanity check here
    assert verifyBondedToProtobuf(conf, sliderClusterAPIClass);

    return sliderClusterAPIClass;
}

---

From source file:org.apache.tez.client.TezClientUtils.java

License:Apache License

@Private
public static DAGClientAMProtocolBlockingPB getAMProxy(final Configuration conf, String amHost, int amRpcPort,
        org.apache.hadoop.yarn.api.records.Token clientToAMToken) throws IOException {

    final InetSocketAddress serviceAddr = NetUtils.createSocketAddrForHost(amHost, amRpcPort);
    UserGroupInformation userUgi = UserGroupInformation
            .createRemoteUser(UserGroupInformation.getCurrentUser().getUserName());
    if (clientToAMToken != null) {
        Token<ClientToAMTokenIdentifier> token = ConverterUtils.convertFromYarn(clientToAMToken, serviceAddr);
        userUgi.addToken(token);//from   ww w  .  j a  v a  2  s.c o m
    }
    if (LOG.isDebugEnabled()) {
        LOG.debug("Connecting to Tez AM at " + serviceAddr);
    }
    DAGClientAMProtocolBlockingPB proxy = null;
    try {
        proxy = userUgi.doAs(new PrivilegedExceptionAction<DAGClientAMProtocolBlockingPB>() {
            @Override
            public DAGClientAMProtocolBlockingPB run() throws IOException {
                RPC.setProtocolEngine(conf, DAGClientAMProtocolBlockingPB.class, ProtobufRpcEngine.class);
                return (DAGClientAMProtocolBlockingPB) RPC.getProxy(DAGClientAMProtocolBlockingPB.class, 0,
                        serviceAddr, conf);
            }
        });
    } catch (InterruptedException e) {
        throw new IOException("Failed to connect to AM", e);
    }
    return proxy;
}

---

From source file:org.apache.tez.dag.api.client.DAGClientServer.java

License:Apache License

private Server createServer(Class<?> pbProtocol, InetSocketAddress addr, Configuration conf, int numHandlers,
        BlockingService blockingService, String portRangeConfig) throws IOException {
    RPC.setProtocolEngine(conf, pbProtocol, ProtobufRpcEngine.class);
    RPC.Server server = new RPC.Builder(conf).setProtocol(pbProtocol).setInstance(blockingService)
            .setBindAddress(addr.getHostName()).setPort(addr.getPort()).setNumHandlers(numHandlers)
            .setVerbose(false).setPortRangeConfig(portRangeConfig).setSecretManager(secretManager).build();
    server.addProtocol(RPC.RpcKind.RPC_PROTOCOL_BUFFER, pbProtocol, blockingService);
    return server;
}

---

From source file:org.apache.tez.service.impl.TezTestServiceProtocolClientImpl.java

License:Apache License

public TezTestServiceProtocolBlockingPB createProxy() throws IOException {
    TezTestServiceProtocolBlockingPB p;/*from  www  .  ja  va  2 s . c om*/
    // TODO Fix security
    RPC.setProtocolEngine(conf, TezTestServiceProtocolBlockingPB.class, ProtobufRpcEngine.class);
    p = (TezTestServiceProtocolBlockingPB) RPC.getProxy(TezTestServiceProtocolBlockingPB.class, 0, serverAddr,
            conf);
    return p;
}

---

From source file:org.apache.tez.service.impl.TezTestServiceProtocolServerImpl.java

License:Apache License

private RPC.Server createServer(Class<?> pbProtocol, InetSocketAddress addr, Configuration conf,
        int numHandlers, BlockingService blockingService) throws IOException {
    RPC.setProtocolEngine(conf, pbProtocol, ProtobufRpcEngine.class);
    RPC.Server server = new RPC.Builder(conf).setProtocol(pbProtocol).setInstance(blockingService)
            .setBindAddress(addr.getHostName()).setPort(0).setNumHandlers(numHandlers).build();
    // TODO Add security.
    return server;
}

---

From source file:org.hdl.caffe.yarn.app.api.impl.pb.client.CaffeClusterPBClientImpl.java

License:Apache License

public CaffeClusterPBClientImpl(long clientVersion, InetSocketAddress addr, Configuration conf)
        throws IOException {
    RPC.setProtocolEngine(conf, CaffeClusterPB.class, ProtobufRpcEngine.class);
    proxy = (CaffeClusterPB) RPC.getProxy(CaffeClusterPB.class, clientVersion, addr, conf);
}

---

• «
• 1
• 2
• »