com.github.jingshouyan.jrpc.client.pool.TransportFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of j-rpc-client Show documentation
There is a newer version: 0.8.3
package com.github.jingshouyan.jrpc.client.pool;

import com.github.jingshouyan.jrpc.base.bean.ServerInfo;
import com.github.jingshouyan.jrpc.base.thrift.Jrpc;
import com.github.jingshouyan.jrpc.client.transport.Transport;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.pool2.BasePooledObjectFactory;
import org.apache.commons.pool2.PooledObject;
import org.apache.commons.pool2.PooledObjectFactory;
import org.apache.commons.pool2.impl.DefaultPooledObject;
import org.apache.thrift.async.TAsyncClientManager;
import org.apache.thrift.protocol.TBinaryProtocol;
import org.apache.thrift.protocol.TProtocolFactory;
import org.apache.thrift.transport.TNonblockingSocket;


/**
 * @author jingshouyan
 * #date 2018/4/17 20:41
 */
@Slf4j
public class TransportFactory extends BasePooledObjectFactory implements PooledObjectFactory {

    private ServerInfo serverInfo;
    private TAsyncClientManager clientManager;
    private static TProtocolFactory protocolFactory = new TBinaryProtocol.Factory();

    public TransportFactory(ServerInfo serverInfo, TAsyncClientManager clientManager) {
        this.serverInfo = serverInfo;
        this.clientManager = clientManager;
    }

    @Override
    public Transport create() {
        try {
            Transport transport = new Transport();
            transport.setKey(serverInfo.getInstance());

            TNonblockingSocket nonblockingSocket = new TNonblockingSocket(serverInfo.getHost(), serverInfo.getPort(), serverInfo.getTimeout());
            Jrpc.AsyncClient asyncClient = new Jrpc.AsyncClient(protocolFactory, clientManager, nonblockingSocket);
            asyncClient.setTimeout(serverInfo.getTimeout());
            transport.setNonblockingSocket(nonblockingSocket);
            transport.setAsyncClient(asyncClient);
            log.debug("client pool make object success. {}==>{},{}:{}",
                    serverInfo.getName(), serverInfo.getInstance(), serverInfo.getHost(), serverInfo.getPort());
            return transport;
        } catch (Exception e) {
            log.warn("client pool make object error.", e);
            throw new RuntimeException(e);
        }
    }

    @Override
    public PooledObject wrap(Transport transport) {
        return new DefaultPooledObject<>(transport);
    }

    @Override
    public void destroyObject(PooledObject p) {
        Transport transport = p.getObject();
        if (transport != null) {
            transport.close();
        }
    }

    @Override
    public boolean validateObject(PooledObject p) {
        Transport transport = p.getObject();
        return transport != null && transport.isOpen();
    }
}