SpringCloud Gateway里怎么給請求添加header信息

這篇文章主要講解了“SpringCloud Gateway里怎么給請求添加header信息”，文中的講解內(nèi)容簡單清晰，易于學(xué)習(xí)與理解，下面請大家跟著小編的思路慢慢深入，一起來研究和學(xué)習(xí)“SpringCloud Gateway里怎么給請求添加header信息”吧！

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());
        });

    }

}

感謝各位的閱讀，以上就是“SpringCloud Gateway里怎么給請求添加header信息”的內(nèi)容了，經(jīng)過本文的學(xué)習(xí)后，相信大家對SpringCloud Gateway里怎么給請求添加header信息這一問題有了更深刻的體會，具體使用情況還需要大家實踐驗證。這里是億速云，小編將為大家推送更多相關(guān)知識點的文章，歡迎關(guān)注！