SpringCloud Gateway 里给请求添加header信息

原创
09/02 11:45
阅读数 2.3K


import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;

import org.springframework.cloud.gateway.filter.GatewayFilterChain;
import org.springframework.cloud.gateway.filter.GlobalFilter;
import org.springframework.http.server.reactive.ServerHttpRequest;
import org.springframework.security.core.context.ReactiveSecurityContextHolder;
import org.springframework.security.core.context.SecurityContext;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;
import org.springframework.web.server.ServerWebExchange;


import reactor.core.publisher.Mono;


@Component
public class WebFluxUserRequestInfoFilter implements GlobalFilter {
    private static final String ORG_CODE = "11000001";
    private static final String CHANNEL_CODE = "WEBQHZX001";
    private static final String HEADER_USER_INFO_ENCODE = "X-User-Info-Encode";

    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {

        return ReactiveSecurityContextHolder.getContext().map(SecurityContext::getAuthentication).map(authentication -> {
            UserRequestInfo userInfo = new UserRequestInfo();
            Object details = authentication.getDetails();
            if(details instanceof JWTPlayload){
                JWTPlayload jwtPlayload = (JWTPlayload) details;
                String userId = jwtPlayload.getSub();
                userInfo.setUserId(userId);
                userInfo.setUserName(jwtPlayload.getCname());
            }
            return userInfo;
        }).defaultIfEmpty(new UserRequestInfo()).flatMap(userInfo -> {
            UserRequestInfoHolder.setInstance(userInfo);
            String userInfoJson;
            try {
                String userInfoEncode = exchange.getRequest().getHeaders().getFirst(HEADER_USER_INFO_ENCODE);
                if(!StringUtils.isEmpty(userInfoEncode) && "false".equalsIgnoreCase(userInfoEncode)){
                    userInfoJson = JsonUtils.object2Json(userInfo);
                }else{
                    userInfoJson = URLEncoder.encode(JsonUtils.object2Json(userInfo), GlobalConstant.CHARSET);
                }
            } catch (UnsupportedEncodingException e) {
                throw new RuntimeException("URLEncoder.encode UserRequestInfo 失败");
            }
            ServerHttpRequest newRequest = exchange.getRequest().mutate()
                    .header(HeaderDefinition.USER_INFO, userInfoJson)
                    .build();
            return chain.filter(exchange.mutate().request(newRequest).build());
        });

    }

}

 

展开阅读全文
打赏
0
0 收藏
分享
加载中
更多评论
打赏
0 评论
0 收藏
0
分享
返回顶部
顶部