经验分享 – 自研网关:特殊URL功能的开发

自研网关系统已开源，求star
项目地址:

现在网关已经具备转发的功能，但有些场景还是不满足

  1. 如果业务系统临时上了某个功能，出问题，需要不重启限制使用，如何处理。
  2. 如果业务系统正在升级，不想让用户继续访问，并提示友好，如何处理
  3. 如果业务系统想将某个页面或某个功能转发到某个地址，如何处理。
    为了满足上述场景，开发了一个特殊URL的功能,主要针对相关URL做临时的处理.

特殊URL处理

特殊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;
    }

}

主要是通过过滤器对特殊的URL进行拦截，根据不同的情况来做判断，做跳转或异常抛出
具体代码大家可以去我的开源项目中看，欢迎提ISSUE及需求。
因为只有一个人，也欢迎大家参与进来.

  • 网关地址:https://gitee.com/starmark947618/starmark-gateway
    下期计划：
    1。权限模块兼容多模块，可以不使用网关系统，按传统的方式做权限拦截
    2。业务系统的工程projectCode传参的调整，当前每次请求都传projectCode,对前端负担太重了.
正文完
2021-05-26