|
|
@@ -0,0 +1,44 @@
|
|
|
+package com.inspur.smsb.gateway.filter;
|
|
|
+
|
|
|
+import com.google.common.base.Strings;
|
|
|
+import com.nimbusds.jose.JWSObject;
|
|
|
+import lombok.extern.slf4j.Slf4j;
|
|
|
+import org.springframework.cloud.gateway.filter.GatewayFilterChain;
|
|
|
+import org.springframework.cloud.gateway.filter.GlobalFilter;
|
|
|
+import org.springframework.http.server.reactive.ServerHttpRequest;
|
|
|
+import org.springframework.stereotype.Component;
|
|
|
+import org.springframework.web.server.ServerWebExchange;
|
|
|
+import reactor.core.publisher.Mono;
|
|
|
+
|
|
|
+import java.text.ParseException;
|
|
|
+
|
|
|
+/**
|
|
|
+ * 过滤器,在请求头中解析 JWT 字段,并解析出 userId 回填
|
|
|
+ *
|
|
|
+ * @author liangke
|
|
|
+ */
|
|
|
+@Slf4j
|
|
|
+@Component
|
|
|
+public class WebFluxUserRequestInfoFilter implements GlobalFilter {
|
|
|
+ @Override
|
|
|
+ public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
|
|
|
+ try {
|
|
|
+ String token = exchange.getRequest().getHeaders().getFirst("Authorization");
|
|
|
+ if (Strings.isNullOrEmpty(token)) {
|
|
|
+ return chain.filter(exchange);
|
|
|
+ }
|
|
|
+ String realToken = token.replace("Bearer ", "");
|
|
|
+ JWSObject jwsObject = JWSObject.parse(realToken);
|
|
|
+
|
|
|
+ ServerHttpRequest request = exchange.getRequest()
|
|
|
+ .mutate()
|
|
|
+ .header("userId", String.valueOf(jwsObject.getPayload().toJSONObject().get("sub")))
|
|
|
+ .build();
|
|
|
+ // 把新的 exchange 放回到过滤链
|
|
|
+ return chain.filter(exchange.mutate().request(request).build());
|
|
|
+ } catch (ParseException e) {
|
|
|
+ log.error(e.getMessage(), e);
|
|
|
+ return chain.filter(exchange);
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|