Spring Cloud Gateway -- cookie添加修改

我就是我 2022-04-25 10:30 1356阅读 0赞

### Spring Cloud Gateway -- cookie添加修改 ###

*  前言
 *  cookie的修改

# 前言 #

由于接口需要有个token认证，需要放在cookie中，但是这个token中含有= ，+， / 等符号，存储在前端导致了乱码。所以想了个办法在网关这层将cookie 在解码一下再放进去。

# cookie的修改 #

原本我是采用这种方式 new一个新的MultiValueMap 将修改好的cookie放入。  
之后在通过public MultiValueMap<String, HttpCookie> getCookies() 这个方法  
但是没有效果不知道为什么。(有大佬知道吗)

ServerHttpRequest oldRequest= exchange.getRequest();
            MultiValueMap<String, HttpCookie> cookieMultiValueMap =  oldRequest.getCookies();
            HttpCookie cookie = cookieMultiValueMap.getFirst("LtpaToken");
    
            MultiValueMap<String, HttpCookie> NewCookieMultiValueMap = new LinkedMultiValueMap<>();
            NewCookieMultiValueMap.add("LtpaToken",cookie);
    		
    		
    		newRequest = new ServerHttpRequestDecorator(newRequest) { 
                  public MultiValueMap<String, HttpCookie> getCookies() { 
    		        return NewCookieMultiValueMap;
    		    }
              };

没办法采用修改header方法

@Override
        public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) { 
            ServerHttpRequest oldRequest= exchange.getRequest();
            MultiValueMap<String, HttpCookie> cookieMultiValueMap =  oldRequest.getCookies();
            HttpCookie cookie = cookieMultiValueMap.getFirst("LtpaToken");
            if(cookie != null){ 
                String cookieValue = URLDecoder.decode(cookie.getValue());
    
                //把request重新包装,继续传递
                URI uri = oldRequest.getURI();
                ServerHttpRequest newRequest = oldRequest.mutate().uri(uri).build();
    
                // 定义新的消息头
                HttpHeaders headers = new HttpHeaders();
                headers.putAll(exchange.getRequest().getHeaders());
    
    
                headers.remove(HttpHeaders.COOKIE);
                headers.set(HttpHeaders.COOKIE, "LtpaToken="+cookieValue);
    
                newRequest = new ServerHttpRequestDecorator(newRequest) { 
                    @Override
                    public HttpHeaders getHeaders() { 
                        return headers;
                    }
                };
                return chain.filter(exchange.mutate().request(newRequest).build());
            }else{ 
                return chain.filter(exchange);
            }
        }