com.quancheng.saluki.core.grpc.client.internal.GrpcStubClient.java Source code

Introduction

Here is the source code for com.quancheng.saluki.core.grpc.client.internal.GrpcStubClient.java

Source

/*
 * Copyright (c) 2016, Quancheng-ec.com All right reserved. This software is the
 * confidential and proprietary information of Quancheng-ec.com ("Confidential
 * Information"). You shall not disclose such Confidential Information and shall
 * use it only in accordance with the terms of the license agreement you entered
 * into with Quancheng-ec.com.
 */
package com.quancheng.saluki.core.grpc.client.internal;

import java.lang.reflect.Method;

import org.apache.commons.lang3.StringUtils;

import com.quancheng.saluki.core.common.Constants;
import com.quancheng.saluki.core.grpc.client.GrpcProtocolClient;
import com.quancheng.saluki.core.utils.ReflectUtils;

import io.grpc.Channel;

/**
 * @author shimingliu 20161214 ?9:50:40
 * @version GrpcStubClient.java, v 0.0.1 20161214 ?9:50:40 shimingliu
 */
public class GrpcStubClient<AbstractStub> implements GrpcProtocolClient<AbstractStub> {

    private final Class<? extends AbstractStub> stubClass;

    public GrpcStubClient(Class<? extends AbstractStub> stubClass) {
        this.stubClass = stubClass;
    }

    public String getStubClassName() {
        return this.stubClass.getName();
    }

    @SuppressWarnings("unchecked")
    @Override
    public AbstractStub getGrpcClient(ChannelPool channelPool, int callType, int callTimeout) {
        String stubClassName = GrpcStubClient.this.getStubClassName();
        Channel channel = null;
        if (StringUtils.contains(stubClassName, "$")) {
            try {
                String parentName = StringUtils.substringBefore(stubClassName, "$");
                Class<?> clzz = ReflectUtils.name2class(parentName);
                Method method;
                switch (callType) {
                case Constants.RPCTYPE_ASYNC:
                    method = clzz.getMethod("newFutureStub", io.grpc.Channel.class);
                    break;
                case Constants.RPCTYPE_BLOCKING:
                    method = clzz.getMethod("newBlockingStub", io.grpc.Channel.class);
                    break;
                default:
                    method = clzz.getMethod("newFutureStub", io.grpc.Channel.class);
                    break;
                }
                channel = channelPool.borrowChannel(null);
                AbstractStub stubInstance = (AbstractStub) method.invoke(null, channel);
                return stubInstance;
            } catch (Exception e) {
                throw new IllegalArgumentException("stub definition not correctdo not edit proto generat file",
                        e);
            } finally {
                if (channel != null) {
                    channelPool.returnChannel(null, channel);
                }
            }
        } else {
            throw new IllegalArgumentException("stub definition not correctdo not edit proto generat file");
        }
    }
}