soul网关学习19-网关全局处理

2021-02-05  本文已影响0人  niuxin

今天准备讲一点简单的东西。我们知道,不管是我们去做框架还是应用程序本身,都或多或少会有些全局处理的逻辑,比如API用户鉴权,再比如统一异常处理。那我们就来看下soul网关中对于请求的全局处理部分又包括那些。
soul网关代码中大致可以找出来如下几个全局处理的逻辑:

  1. GlobalPlugin-全局插件
  2. GlobalErrorHandler-全局异常处理
  3. SoulWebHandler-soul网关web请求处理
  4. CrossFilter-跨域支持过滤器
  5. FileSizeFilter-文件大小限制过滤器
  6. ExcludeFilter-URL排除过滤器
  7. WebSocketParamFilter-websocket请求参数过滤器

GlobalPlugin-全局插件

    public Mono<Void> execute(final ServerWebExchange exchange, final SoulPluginChain chain) {
        final ServerHttpRequest request = exchange.getRequest();
        final HttpHeaders headers = request.getHeaders();
        final String upgrade = headers.getFirst("Upgrade");
        // 构造soul上下文
        SoulContext soulContext;
        // 非websocket的处理
        if (StringUtils.isBlank(upgrade) || !"websocket".equals(upgrade)) {
            soulContext = builder.build(exchange);
        } else {
            // websocket的处理
            final MultiValueMap<String, String> queryParams = request.getQueryParams();
            soulContext = transformMap(queryParams);
        }
        exchange.getAttributes().put(Constants.CONTEXT, soulContext);
        return chain.execute(exchange);
    }
    public SoulContext build(final ServerWebExchange exchange) {
        final ServerHttpRequest request = exchange.getRequest();
        String path = request.getURI().getPath();
        // 根据当前的请求路径从缓存中获取到metaData
        MetaData metaData = MetaDataCache.getInstance().obtain(path);
        // 原始metaData放到exchange中
        if (Objects.nonNull(metaData) && metaData.getEnabled()) {
            exchange.getAttributes().put(Constants.META_DATA, metaData);
        }
        // 将metaData转换为SoulContext
        return transform(request, metaData);
    }

小结

GlobalErrorHandler-全局异常处理

SoulWebHandler-soul网关web请求处理

CrossFilter-跨域支持过滤器

    public Mono<Void> filter(final ServerWebExchange exchange, final WebFilterChain chain) {
        ServerHttpRequest request = exchange.getRequest();
        // 判断是否为跨域请求,判断的逻辑:是否有origin参数,若有,再判断当前请求的全路径是否与origin一致
        // 如果判断为跨域请求,则在reponse的header头中添加一些跨域支持的参数
        if (CorsUtils.isCorsRequest(request)) {
            ServerHttpResponse response = exchange.getResponse();
            HttpHeaders headers = response.getHeaders();
            headers.add("Access-Control-Allow-Origin", ALLOWED_ORIGIN);
            headers.add("Access-Control-Allow-Methods", ALLOWED_METHODS);
            headers.add("Access-Control-Max-Age", MAX_AGE);
            headers.add("Access-Control-Allow-Headers", ALLOWED_HEADERS);
            headers.add("Access-Control-Expose-Headers", ALLOWED_EXPOSE);
            headers.add("Access-Control-Allow-Credentials", "true");
            // 判断是否为optins请求,若是直接返回请求成功;否则会继续执行过滤器链
            if (request.getMethod() == HttpMethod.OPTIONS) {
                response.setStatusCode(HttpStatus.OK);
                return Mono.empty();
            }
        }
        return chain.filter(exchange);
    }

FileSizeFilter-文件大小限制过滤器

    public Mono<Void> filter(@NonNull final ServerWebExchange exchange, @NonNull final WebFilterChain chain) {
        MediaType mediaType = exchange.getRequest().getHeaders().getContentType();
        // 只针对带文件的表单数据的请求,会有文件大小限制
        if (MediaType.MULTIPART_FORM_DATA.isCompatibleWith(mediaType)) {
            ServerRequest serverRequest = ServerRequest.create(exchange,
                    messageReaders);
            return  //先将请求中的body转换为DataBuffer
                    serverRequest.bodyToMono(DataBuffer.class)
                    .flatMap(size -> {
                        // dataBuffer的容量是否大于最大的文件大小,若大则直接返回异常
                        if (size.capacity() > BYTES_PER_MB * fileMaxSize) {
                            // 。。。。
                        }
                        // 因上面已经读取了body数据,在io流中body体是不能读取多次的
                        // 需要重新生成一个新的ServerHttpRequest对象包装原有的body数据,并传递到程序后面的请求中
                        BodyInserter<Mono<DataBuffer>, ReactiveHttpOutputMessage> bodyInsert = BodyInserters.fromPublisher(Mono.just(size), DataBuffer.class);
                        HttpHeaders headers = new HttpHeaders();
                        headers.putAll(exchange.getRequest().getHeaders());
                        headers.remove(HttpHeaders.CONTENT_LENGTH);
                        CachedBodyOutputMessage outputMessage = new CachedBodyOutputMessage(
                                exchange, headers);
                        return bodyInsert.insert(outputMessage, new BodyInserterContext())
                                .then(Mono.defer(() -> {
                                    // ServerHttpRequest的装饰,用传入的CachedBodyOutputMessage替换原有Request的body
                                    ServerHttpRequest decorator = decorate(exchange, outputMessage);
                                    // 在原来exchange的基础上,重新构造了新的exchange对象
                                    return chain.filter(exchange.mutate().request(decorator).build());

                                }));
                    });
        }
        return chain.filter(exchange);
    }

ExcludeFilter-url排除过滤器

WebSocketParamFilter-websocket请求参数过滤器

上一篇下一篇

猜你喜欢

热点阅读