zoukankan      html  css  js  c++  java
  • dubbo中拦截生产者或消费者服务方法调用

    比如当前有个需求,需要拦截dubbo的服务提供方或者服务消费方的方法,判断参数中是否包含某个关键字进行拦截阻止执行,那么我们可以通过使用dubbo的SPI机制通过实现Filter类来拦截,话不多说直接上代码:

    1、新建一个类实现dubbo的Filter接口

    /**
     * 回滚任务Dubbo拦截正在回滚或备份小区的方法调用,阻止执行
     * @author wl
     * @date 2019-08-16 13:24
     */
    @Activate(group={ Constants.PROVIDER, Constants.CONSUMER })
    public class DubboRollbackTaskFilter implements Filter {
    
        private static final Logger logger = LoggerFactory.getLogger(DubboRollbackTaskFilter.class);
    
        private RedisTemplate redisTemplate;
    
        /**
         * Filter中@Autowired会失效
         * 使用dubbo自动注入redisTemplate类
         * @param redisTemplate
         */
        public void setRedisTemplate(RedisTemplate redisTemplate) {
            this.redisTemplate = redisTemplate;
        }
    
        public static final String ROLLBACK_READIS_KEY = "server_rollback_task_region_ids";
    
        @Override
        public Result invoke(Invoker<?> invoker, Invocation invocation) throws RpcException {
            try {
                String method = invocation.getMethodName();
                String name = invoker.getInterface().getName();
                Object[] args = invocation.getArguments();
                String ageString = JSON.toJSONString(args);
                // redis中获取数据
                Object redisValues = redisTemplate.opsForValue().get(DubboRollbackTaskFilter.ROLLBACK_READIS_KEY);
                logger.debug(">>>>>dubboRollbackTaskFilter-className:{},method:{},args:{},redisValues:{}", name, method, ageString, JSON.toJSONString(redisValues));
                if (null != redisValues && redisValues.toString().trim().length() > 0) {
                    String[] regionIds = redisValues.toString().split(",");
                    for (String regionId : regionIds) {
                        if (regionId.trim().length() <= 0) {
                            continue;
                        }
                        // 只要包含当前有处理的小区,那么直接抛出参数异常
                        if (ageString.contains(regionId)) {
                            logger.info(">>>>>dubboRollbackTaskFilter-拦截:{},method:{},args:{},regionId:{}", name, method, ageString, regionId);
                            return new RpcResult(new IllegalArgumentException("回滚任务拦截提示:当前小区正在备份或回滚数据,一切操作不允许执行!"));
                        }
                    }
                }
            } catch (Exception e) {
                // 拦截异常,不影响正常操作
                logger.error(">>>>>dubboRollbackTaskFilter-异常", e);
            }
            return invoker.invoke(invocation);
        }
    }

    2、META-INF下新建文件

      

      文件内容为:

    dubboRollbackTaskFilter=com.qdingnet.pcloud.helper.utils.DubboRollbackTaskFilter
    

      至此就可以成功的拦截你想拦截的东西

    3、下面解释几个点:

      1、关于@Activate,Activate注解表示一个扩展是否被激活,可以放在类定义和方法上,dubbo用它在spi扩展类定义上,表示这个扩展实现被激活。

        具体如何实现可以百度或看下(这篇文章

        如果不使用此注解可以直接在配置中添加

        1.1 比如配置中添加   dubbo.provider.filter =dubboServiceFilter 

        1.2 或者在dubbo配置xml中添加 <dubbo:provider filter="xxxFilter" /> 或者 <dubbo:consumer filter="xxxFilter" /> 使Filter生效

      2、关于在Filter中使用@Resoure@Autowired

        在dubbo拦截器中以上两个注解是不会生效的,我们可以用以下两种方式注入

        2.1 可以采取通过setter方式来注入其他的bean,就像最上面的代码示例

        2.2 通过Spring的ApplicationContextAware接口

          

        2.3 直接从Spring中取,ServiceBean.getSpringContext().getBean(clazz)

      3、说说拦截的invoke返回方式

        如果想直接返回自定义的拦截,直接使用dubbo提供的RpcResult,然后将异常作为参数传给此类进行返回

  • 相关阅读:
    Linux进程间通信(IPC)
    mq_setattr
    mq_getattr
    mq_unlink
    mq_receive
    mq_send
    mq_close
    POSIX消息队列
    mq_open
    C语言关键字
  • 原文地址:https://www.cnblogs.com/wanglu/p/11378164.html
Copyright © 2011-2022 走看看