openfeign 高版本对参数限制处理.

Author: 1991wangliang@gmail.com

txlcn-tracing/src/main/java/com/codingapi/txlcn/tracing/TracingContext.java

@@ -20,8 +20,10 @@
 import com.codingapi.txlcn.common.util.Maps;
 import com.codingapi.txlcn.common.util.id.RandomUtils;
 import lombok.extern.slf4j.Slf4j;
+import org.springframework.util.Base64Utils;
 import org.springframework.util.StringUtils;
 
+import java.nio.charset.Charset;
 import java.util.*;
 
 /**
@@ -62,6 +64,13 @@ public void init(Map<String, String> initFields) {
         if (Objects.isNull(fields)) {
             this.fields = new HashMap<>();
         }
+        //APP_MAP base64 解码
+        if(initFields.containsKey(TracingConstants.APP_MAP)){
+            String appMapVal = initFields.get(TracingConstants.APP_MAP);
+            if(!appMapVal.startsWith("{")||!appMapVal.contains("{")){
+                initFields.put(TracingConstants.APP_MAP,baseString2appMap(appMapVal));
+            }
+        }
         this.fields.putAll(initFields);
     }
 
@@ -91,14 +100,22 @@ public void addApp(String serviceId, String address) {
         raiseNonGroupException();
     }
 
-    public String appMapString() {
+    public String appMapBase64String() {
         if (hasGroup()) {
-            return this.fields.get(TracingConstants.APP_MAP);
+            return Base64Utils.encodeToString(this.fields.get(TracingConstants.APP_MAP).getBytes(Charset.forName("utf8")));
         }
         raiseNonGroupException();
         return "";
     }
 
+    private String baseString2appMap(String base64Str) {
+        //解码
+        if(!"".equals(base64Str)){
+            base64Str = Base64Utils.encodeToString(base64Str.getBytes(Charset.forName("utf8")));
+        }
+        return base64Str;
+    }
+
     public JSONObject appMap() {
         if (hasGroup()) {
             String appMap = this.fields.get(TracingConstants.APP_MAP);

txlcn-tracing/src/main/java/com/codingapi/txlcn/tracing/dubbo/DubboRequestInterceptor.java

@@ -35,7 +35,7 @@ public class DubboRequestInterceptor implements Filter {
     public Result invoke(Invoker<?> invoker, Invocation invocation) throws RpcException {
         if (TracingContext.tracing().hasGroup()) {
             RpcContext.getContext().setAttachment(TracingConstants.HEADER_KEY_GROUP_ID, TracingContext.tracing().groupId());
-            RpcContext.getContext().setAttachment(TracingConstants.HEADER_KEY_APP_MAP, TracingContext.tracing().appMapString());
+            RpcContext.getContext().setAttachment(TracingConstants.HEADER_KEY_APP_MAP, TracingContext.tracing().appMapBase64String());
         }
         return invoker.invoke(invocation);
     }

txlcn-tracing/src/main/java/com/codingapi/txlcn/tracing/http/FeignRequestInterceptor.java

@@ -38,6 +38,6 @@ public class FeignRequestInterceptor implements RequestInterceptor {
     @Override
     public void apply(RequestTemplate requestTemplate) {
         requestTemplate.header(TracingConstants.HEADER_KEY_GROUP_ID, TracingContext.tracing().groupId());
-        requestTemplate.header(TracingConstants.HEADER_KEY_APP_MAP, TracingContext.tracing().appMapString());
+        requestTemplate.header(TracingConstants.HEADER_KEY_APP_MAP, TracingContext.tracing().appMapBase64String());
     }
 }

txlcn-tracing/src/main/java/com/codingapi/txlcn/tracing/http/RestTemplateRequestInterceptor.java

@@ -46,7 +46,7 @@ public ClientHttpResponse intercept(
             @NonNull HttpRequest httpRequest, @NonNull byte[] bytes,
             @NonNull ClientHttpRequestExecution clientHttpRequestExecution) throws IOException {
         httpRequest.getHeaders().add(TracingConstants.HEADER_KEY_GROUP_ID, TracingContext.tracing().groupId());
-        httpRequest.getHeaders().add(TracingConstants.HEADER_KEY_APP_MAP, TracingContext.tracing().appMapString());
+        httpRequest.getHeaders().add(TracingConstants.HEADER_KEY_APP_MAP, TracingContext.tracing().appMapBase64String());
         return clientHttpRequestExecution.execute(httpRequest, bytes);
     }
 }