Gateway网关过滤器GlobalFilter返回前端JSON数据

 

废话不多说,直接上干货

文章主旨：   将GlobalFilter拦截下来的请求,进行校验,并封装错误信息然后返回前端JSON

正文
 

先看该篇文章的示例接口:

package cn.itcast.gateway;

import com.alibaba.fastjson.JSON;
import org.springframework.cloud.gateway.filter.GatewayFilterChain;
import org.springframework.cloud.gateway.filter.GlobalFilter;
import org.springframework.core.annotation.Order;
import org.springframework.core.io.buffer.DataBuffer;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.server.reactive.ServerHttpRequest;
import org.springframework.http.server.reactive.ServerHttpResponse;
import org.springframework.stereotype.Component;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Mono;
import java.nio.charset.StandardCharsets;
import java.util.HashMap;
import java.util.Map;

@Order(-1)
@Component
public class AuthorizeFilter implements GlobalFilter {
    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
        
        //1.从请求头中获取token
        ServerHttpRequest request = exchange.getRequest();
        HttpHeaders requestHeaders = request.getHeaders();
        String token = requestHeaders.getFirst("token");
        ServerHttpResponse response = exchange.getResponse();

        //2.判断token是否为空
        if (token != null && token != "") {
            //3.如果不为空,对token进行校验
            if ("admin".equals(token)) {
                return chain.filter(exchange);
            } else {

                //3.1设置状态码(黑马出品,必出精品)
                response.setStatusCode(HttpStatus.UNAUTHORIZED);

                //3.2封装返回数据
                Map<String, Object> map = new HashMap<>();
                map.put("msg", "未登录");
                map.put("code", "NOTLOGING");

                //3.3作JSON转换
                byte[] bytes = JSON.toJSONString(map).getBytes(StandardCharsets.UTF_8);

                //3.4调用bufferFactory方法,生成DataBuffer对象
                DataBuffer buffer = response.bufferFactory().wrap(bytes);

                //4.调用Mono中的just方法,返回要写给前端的JSON数据
                return response.writeWith(Mono.just(buffer));
            }
        } else {
            return exchange.getResponse().setComplete();
        }
    }
}

 文章用到的JSON是阿里提供的fastjson工具包

        <dependency>
            <groupId>com.alibaba</groupId>
            <artifactId>fastjson</artifactId>
            <version>1.2.76</version>
        </dependency>

如果不想导包,也可直接用Jackson

Jackson库是JSON Java库，用于在Java程序中解析JSON数据。

byte[] bytes = JacksonUtils.toJsonBytes(map);