dubbo源码阅读ContextFilter&ConsumerContextFilter的消费者和提供者简单传递Attachment

例如当前我们的消费者Customer为Controller,需要通过登录者主键去访问后台信息,但是我们的登陆者信息单点登录现在是存在消费者这一端,那我们可以使用Dubbo的setAttachMents去存储我们所需要的信息,提供者就通过getAttachMents后获取我们所需要的附件信息,调用我们在后面会提供方法

实例:

消费者Customer

public class DubboConsumerParamsFilter implements Filter {

    @Override
    public Result invoke(Invoker<?> invoker, Invocation invocation) throws RpcException {
        Map<String, String> context = new HashMap<>();
        String userId;
        try {
             //登录用户信息我们存在了UserManager中
             userId = UserManager.getLoginUser().getUserId();
        }catch (Exception e){
            userId = "SYSTEM" ;
        }
        // 获取userID  UserController  获取session
        //key我们定义的常量
        context.put(Constans.DUBBO_USERID_KEY, userId);
        RpcContext.getContext().setAttachments(context);
        return invoker.invoke(invocation);
    }
}

提供者Provider

public class DubboProviderParamsFilter implements Filter {
    @Override
    public Result invoke(Invoker<?> invoker, Invocation invocation) throws RpcException {

        String userId = RpcContext.getContext().getAttachment(Constans.DUBBO_USERID_KEY);
        //获取当前线程id
           //我们设定了一个DubboUserMananger去操作当前线程
        DubboUserManager.setDubboUserId(userId);
        return invoker.invoke(invocation);
    }
}

操作当前线程并且是安全的,我们这里选择的是分段Map

public class DubboUserManager {
    private static ConcurrentHashMap<String, String> dubboUsersMap = new ConcurrentHashMap<String, String>() ;

    public DubboUserManager() {
    }

    public static void setDubboUserId(String userId){
        dubboUsersMap.put(String.valueOf(Thread.currentThread().getId()),userId);
    }
    public static String getDubboUserId(){
        return dubboUsersMap.get(String.valueOf(Thread.currentThread().getId()));
    }
}

调用我们直接在消费者端使用

DubboUserManager.getDubboUserId()就可以获取我们需要的userId
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值