io.nuls.base.api.provider.BaseRpcService Maven / Gradle / Ivy

Go to download
package io.nuls.base.api.provider;

import io.nuls.core.rpc.model.message.Response;
import io.nuls.core.rpc.netty.processor.ResponseMessageProcessor;
import io.nuls.core.constant.ErrorCode;
import io.nuls.core.log.Log;
import io.nuls.core.model.StringUtils;
import io.nuls.core.parse.MapUtils;

import java.util.HashMap;
import java.util.Map;
import java.util.function.Function;

/**
 * @Author: zhoulijun
 * @Time: 2019-03-07 19:46
 * @Description: rpc provider 基础类
 */
public abstract class BaseRpcService extends BaseService {

    public static final ErrorCode RPC_ERROR_CODE = ErrorCode.init("10016");

    /**
     * 调用其他模块rpc接口
     * @param module 模块名称
     * @param method rpc接口名称
     * @param req 业务参数对象
     * @param callback 回调函数
     * @param  返回值泛型类型
     * @return
     */
    protected  Result callRpc(String module,String method,Object req,Function callback) {
        Map params = MapUtils.beanToLinkedMap(req);
        Log.debug("call {} rpc , method : {},param : {}",module,method,params);
        Response cmdResp = null;
        try {
            cmdResp = ResponseMessageProcessor.requestAndResponse(module, method, params);
            Log.debug("result : {}",cmdResp);
        } catch (Exception e) {
            Log.warn("Calling remote interface failed. module:{} - interface:{} - message:{}", module, method, e.getMessage());
            return fail(RPC_ERROR_CODE,e.getMessage());
        }
        if (!cmdResp.isSuccess()) {
            String comment = cmdResp.getResponseComment();
            if(StringUtils.isBlank(comment)) {
                comment = "";
            }
            Log.warn("Calling remote interface failed. module:{} - interface:{} - ResponseComment:{}", module, method, cmdResp.getResponseComment());
            if(cmdResp.getResponseStatus() == Response.FAIL){
                //business error
                String errorCode = cmdResp.getResponseErrorCode();
                if(StringUtils.isBlank(errorCode)){
                    return fail(RPC_ERROR_CODE, StringUtils.isBlank(comment) ? "unknown error" : comment);
                }
                return fail(ErrorCode.init(errorCode), comment);
            }else{
                if(StringUtils.isNotBlank(comment)) {
                    return fail(RPC_ERROR_CODE, comment);
                }
                return fail(RPC_ERROR_CODE, "unknown error");
            }
        }
        return callback.apply((R) ((HashMap) cmdResp.getResponseData()).get(method));
    }

    protected abstract   Result call(String method, Object req, Function callback);

    protected Result callReturnString(String method,Object req, String fieldName){
        Function fun = res->{
            String data = (String) res.get(fieldName);
            return success(data);
        };
        return call(method,req,fun);
    }

    protected Result callResutlMap(String method,Object req){
        Function fun = res->{
            return success(res);
        };
        return call(method,req,fun);
    }


}