SpringCloud Gateway网关 -- 开发自定义局部过滤器

实现需求:希望打印出方法运行的时间

 filters:
     - StripPrefix=1
     - Time=true/false

1 注意代码中注释的地方

package cn.haiwang.gatewayservice.filter;

import lombok.Data;
import lombok.extern.slf4j.Slf4j;
import org.springframework.cloud.gateway.filter.GatewayFilter;
import org.springframework.cloud.gateway.filter.GatewayFilterChain;
import org.springframework.cloud.gateway.filter.factory.AbstractGatewayFilterFactory;
import org.springframework.stereotype.Component;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Mono;
import java.net.URI;
import java.util.Arrays;
import java.util.List;

/**
 * 开发自定义局部过滤器,显示方法运行的时间
 */
/* 1 添加到Spring Bean中 */
@Component
@Slf4j
/* 2 名称必须以GatewayFilterFactory结尾 */
public class TimeGatewayFilterFactory extends AbstractGatewayFilterFactory<TimeGatewayFilterFactory.PeizhiInfo> {

    /* 3 使用无参构造 */
    public TimeGatewayFilterFactory() {
        super(TimeGatewayFilterFactory.PeizhiInfo.class);
    }

    @Override
    public GatewayFilter apply(PeizhiInfo config) {

        return new GatewayFilter() {
            @Override
            public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
                Boolean showTime = config.showTime;
                if(showTime!=null) {
                    if (showTime.equals(false)) {
                        return chain.filter(exchange);
                    }
                }

                URI uri = exchange.getRequest().getURI();
                exchange.getAttributes().put("BEGIN_TIME",System.currentTimeMillis());

                return chain.filter(exchange).then(Mono.fromRunnable(()->{
                    Long beginTime = exchange.getAttribute("BEGIN_TIME");
                    if(beginTime!=null) {
                        log.warn(uri+" 请示耗时 {} ms",System.currentTimeMillis() - beginTime);
                    }
                }));
            }
        };
    }

    /*
    60行 和 65行 名称要对应上
     */
    @Override
    public List<String> shortcutFieldOrder() {
        return Arrays.asList("showTime");
    }

    @Data
    protected static class PeizhiInfo {
        private Boolean showTime;
    }
}

2 使用上面的过滤器

spring:
  cloud:
    gateway:
      routes:
        - id: product_route
          uri: lb://product-service # lb指的是从nacos中按照名称获取微服务,并遵循负载均 衡策略
          predicates:
            - Path=/product-serv/**
          filters:
            - StripPrefix=1
            - Time=false # 使用自定义局部过滤器 ,不打印运行时间
        - id: order_route
          uri: lb://order-service # lb指的是从nacos中按照名称获取微服务,并遵循负载均 衡策略
          predicates:
            - Path=/order-serv/**
          filters:
            - StripPrefix=1
            - Time=true # 使用自定义局部过滤器 ,打印运行时间

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值