自研网关:特殊URL功能的开发

@Slf4j
public class GatewaySpecUrlWebFilter implements WebFilter {


    @Autowired
    private IProjectSpecUrlService projectSpecUrlService;

    @Override
    public Mono<Void> filter(ServerWebExchange exchange, WebFilterChain chain) {
        String projectCode = RequestHelper.getRequestProjectCode(exchange.getRequest());
        Assert.hasText(projectCode, "工程编码不存在");

        //特殊URL处理;
        if (this.handleSpecUrls(exchange, projectCode)) {

            return chain.filter(exchange.mutate().request(exchange.getRequest().mutate().path("/redirect").build()).build());

        }

        return chain.filter(exchange);
    }

    @SneakyThrows
    private boolean handleSpecUrls(ServerWebExchange exchange, String projectCode) {

        List<ProjectSpecUrlDto> specUrlDtos = projectSpecUrlService.findSpecUrlListByProjectCode(projectCode);
        for (ProjectSpecUrlDto projectSpecUrlDto : specUrlDtos) {
            if (this.handleSpecUrl(projectSpecUrlDto, exchange, projectCode)) {
                return true;
            }
        }
        return false;
    }

    @SneakyThrows
    private boolean handleSpecUrl(ProjectSpecUrlDto projectSpecUrlDto, ServerWebExchange exchange, String projectCode) {
        boolean isMatch = RequestHelper.isMatchUrl(projectSpecUrlDto.getAccessUrl(), exchange.getRequest().getURI().getPath(), projectCode);
        if (isMatch) {
            //URL匹配
            if (projectSpecUrlDto.getOperation() == 0) {
                //   chain.filter(exchange.mutate().response(ServerResponse.temporaryRedirect(URI.create(projectSpecUrlDto.getRedirectUrl())).build()).request(exchange.getRequest()).build());
                //  chain.filter(exchange.mutate().request(exchange.getRequest().mutate().uri(new URI(projectSpecUrlDto.getRedirectUrl())).build()).build());
                exchange.getAttributes().put("gateway_spec_url_redirect_url",projectSpecUrlDto.getRedirectUrl());
                return true;
            } else if (projectSpecUrlDto.getOperation() == 1) {
                throw new CustomException(projectSpecUrlDto.getErrorMsg());

            } else if (projectSpecUrlDto.getOperation() == 2) {
                throw new BaseException(SystemErrorType.SPEC_URL_ERROR_ACCESS);

            }
        }
        return false;
    }

}