Spring Cloud Gateway 源码剖析之Predicate谓词详解

package org.springframework.cloud.gateway.handler.predicate;
 
import org.springframework.cloud.gateway.support.ConfigurationUtils;
import org.springframework.core.style.ExpressionUtils;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.server.ServerWebExchange;
 
import java.util.Arrays;
import java.util.List;
import java.util.function.Predicate;
 
@Validated
public class QueryRoutePredicateFactory extends AbstractRoutePredicateFactory<QueryRoutePredicateFactory.Config> {
    public static final String QUERY_KEY = "key";
 
    public QueryRoutePredicateFactory() {
        super(Config.class);
    }
 
    @Override
    public List<String> shortcutFieldOrder() {
        return Arrays.asList(QUERY_KEY);
    }
 
    @Override
    public Predicate<ServerWebExchange> apply(Config config) {
        return exchange -> {
            // 获取请求参数
            MultiValueMap<String, String> queryParams = exchange.getRequest().getQueryParams();
            // 检查请求参数中是否包含配置的key
            boolean hasKey = queryParams.containsKey(config.key);
            // 根据配置判断是否应该断言为真
            if (config.present) {
                return hasKey;
            } else {
                return !hasKey;
            }
        };
    }
 
    public static class Config {
        @NotEmpty
        private String key;
 
        private boolean present = true;
 
        public String getKey() {
            return key;
        }
 
        public void setKey(String key) {
            this.key = key;
        }
 
        public boolean isPresent() {
            return present;
        }
 
        public void setPresent(boolean present) {
            this.present = present;
        }
    }
}

这个代码示例展示了如何实现一个自定义的路由断言工厂，用于检查HTTP请求中是否存在特定的查询参数。如果参数存在，断言结果为真；如果参数不存在，可以根据配置决定断言为真还是为假。这个例子中的工厂类名为QueryRoutePredicateFactory，它继承自抽象的路由断言工厂基类AbstractRoutePredicateFactory<T>，并使用了Config内部类来定义配置属性。这个例子展示了如何使用Java 8的函数式编程特性来实现路由断言。