Spring Cloud - 手写 Gateway 源码，实现自定义局部 FilterFactory

在Spring Cloud Gateway中，创建一个自定义的GatewayFilterFactory可以通过实现GatewayFilterFactory接口并注册到Spring容器中来完成。以下是一个简单的自定义GatewayFilterFactory的示例，它会在请求头中添加一个自定义的头信息。

import org.springframework.cloud.gateway.filter.GatewayFilter;
import org.springframework.cloud.gateway.filter.GatewayFilterChain;
import org.springframework.cloud.gateway.filter.factory.GatewayFilterFactory;
import org.springframework.core.io.buffer.DataBuffer;
import org.springframework.core.io.buffer.DataBufferFactory;
import org.springframework.http.server.reactive.ServerHttpResponse;
import org.springframework.stereotype.Component;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Mono;
 
@Component
public class CustomGatewayFilterFactory implements GatewayFilterFactory {
 
    @Override
    public GatewayFilter apply(Object config) {
        return (exchange, chain) -> chain.filter(exchange)
                .then(Mono.fromRunnable(() -> {
                    ServerHttpResponse response = exchange.getResponse();
                    response.getHeaders().set("Custom-Header", "CustomValue");
                }));
    }
}

在上述代码中，我们创建了一个名为CustomGatewayFilterFactory的类，并实现了GatewayFilterFactory接口。apply方法返回一个GatewayFilter，在请求响应链中，它会设置一个自定义的响应头。

要使这个自定义的GatewayFilterFactory生效，你需要将其注册为Spring容器中的一个组件，即通过@Component注解标记。

在application.yml或application.properties配置文件中，你可以这样配置使用这个自定义的GatewayFilterFactory：

spring:
  cloud:
    gateway:
      routes:
        - id: custom_filter_route
          uri: https://example.org
          filters:
            - CustomGatewayFilterFactory

在上面的配置中，我们为一个路由添加了我们自定义的GatewayFilterFactory。当请求经过这个路由时，它将触发我们自定义的过滤器，在响应中添加一个自定义的头信息。