本文共 2519 字,大约阅读时间需要 8 分钟。
package com.poly.gateway.filter.auth;import com.fasterxml.jackson.databind.ObjectMapper;import com.poly.gateway.common.BusinessException;import lombok.extern.slf4j.Slf4j;import org.apache.commons.lang3.StringUtils;import org.springframework.cloud.gateway.filter.GatewayFilterChain;import org.springframework.cloud.gateway.filter.GlobalFilter;import org.springframework.core.Ordered;import org.springframework.core.io.buffer.DataBuffer;import org.springframework.http.MediaType;import org.springframework.http.server.reactive.ServerHttpResponse;import org.springframework.stereotype.Component;import org.springframework.web.server.ServerWebExchange;import reactor.core.publisher.Flux;import reactor.core.publisher.Mono;/** * 认证过滤器 * * @author 向振华 * @date 2020/09/16 10:30 */@Slf4j@Componentpublic class AuthFilter implements GlobalFilter, Ordered { @Override public int getOrder() { return 0; } @Override public Monofilter(ServerWebExchange exchange, GatewayFilterChain chain) { if (authenticate(exchange)) { return chain.filter(exchange); } else { return errorInfo(exchange, "认证不通过", 1); } } private boolean authenticate(ServerWebExchange exchange) { // 进行认证逻辑,返回true表示认证通过 return true; } public static Mono errorInfo(ServerWebExchange exchange, String message, Integer status) { Map resultMap = new HashMap<>(); resultMap.put("status", status == null ? 2 : status); resultMap.put("message", StringUtils.isBlank(message) ? "服务异常!" : message); resultMap.put("info", null); resultMap.put("data", null); return Mono.defer(() -> { byte[] bytes; try { bytes = new ObjectMapper().writeValueAsBytes(resultMap); } catch (JsonProcessingException e) { log.error("网关响应异常:", e); throw new BusinessException("信息序列化异常"); } catch (Exception e) { log.error("网关响应异常:", e); throw new BusinessException("写入响应异常"); } ServerHttpResponse response = exchange.getResponse(); response.getHeaders().add("Content-Type", MediaType.APPLICATION_JSON_UTF8.toString()); DataBuffer buffer = response.bufferFactory().wrap(bytes); return response.writeWith(Flux.just(buffer)); }); }}
以上代码经过优化,主要做了以下修改:
转载地址:http://eaxiz.baihongyu.com/