在Dubbo 里如何手动实现网关聚合功能?

核心是使用RpcContext内置对象,RpcContext 是一个 ThreadLocal 的临时状态记录器,当接收到 RPC 请求,或发起 RPC 请求时,RpcContext 的状态都会变化。比如:A 调 B,B 再调 C,则 B 机器上,在 B 调 C 之前,RpcContext 记录的是 A 调 B 的信息,在 B 调 C 之后,RpcContext 记录的是 B 调 C 的信息。

// 远程调用
xxxService.xxx();
// 本端是否为消费端,这里会返回true
boolean isConsumerSide = RpcContext.getContext().isConsumerSide();
// 获取最后一次调用的提供方IP地址
String serverIP = RpcContext.getContext().getRemoteHost();
// 获取当前服务配置信息,所有配置信息都将转换为URL的参数
String application = RpcContext.getContext().getUrl().getParameter("application");
// 注意:每发起RPC调用,上下文状态会变化
yyyService.yyy();

思路是把聚合服务作为PRC调用方,调用一次服务后,记录服务的ip地址,然后配合配置文件里定义的端口,进行跳转;这种方案的缺点是ip是可以随机的,但是端口需要在配置文件里写死

聚合服务核心代码:

@Controller
@RequestMapping(value = "router")
public class RouterController {

    @Value("${services.ports.user}")
    private String userPort;

    @Value("${services.ports.content}")
    private String contentPort;

    @Reference(version = "${services.versions.user.v1}")
    private UserConsumerService userConsumerService;

    @Reference(version = "${services.versions.content.v1}")
    private ContentConsumerService contentConsumerService;

    @RequestMapping(value = "user", method = RequestMethod.GET)
    public String user(String path) {
        // 远程调用
        userConsumerService.info();
        return getRequest(userPort,path);
    }

    @RequestMapping(value = "content", method = RequestMethod.GET)
    public String content(String path) {
        // 远程调用
        contentConsumerService.info();
        return getRequest(contentPort,path);
    }

    /**
     * 获取请求地址
     * @param port
     * @param path
     * @return
     */
    public String getRequest(String port,String path) {

        // 本端是否为消费端,这里会返回true
        boolean isConsumerSide = RpcContext.getContext().isConsumerSide();

        if (isConsumerSide) {
            // 获取最后一次调用的提供方IP地址
            String serverIP = RpcContext.getContext().getRemoteHost();
            String format = String.format("redirect:http://%s:%s%s", serverIP, port, path);
            return format;
        }
        return null;
    }
}
posted @ 2020-09-24 10:44  镇魂帆-张  阅读(495)  评论(0编辑  收藏  举报