Przeglądaj źródła

rpc 增加调用方信息

panh 5 lat temu
rodzic
commit
9dfbde8df3

+ 12 - 4
framework-boot/src/main/java/com/mrxu/framework/boot/entity/RpcRequestInfo.java

@@ -5,6 +5,18 @@ import lombok.Data;
 @Data
 public class RpcRequestInfo {
 
+    private RpcRequestInfo() {}
+
+    private static ThreadLocal<RpcRequestInfo> rpcRequestInfo = new ThreadLocal<RpcRequestInfo>();
+
+    public static RpcRequestInfo get() {
+        return rpcRequestInfo.get();
+    }
+
+    public static void remove() {
+        rpcRequestInfo.remove();
+    }
+
     private String trackId;
 
     private String merchantId;
@@ -13,8 +25,4 @@ public class RpcRequestInfo {
 
     private String ip;
 
-    public String toString() {
-        return "{trackId:'"+trackId+"',merchantId:'"+merchantId+"',username:'"+username+"',ip:'"+ip+"'}";
-    }
-
 }

+ 0 - 3
framework-boot/src/main/java/com/mrxu/framework/boot/web/ServletUtils.java

@@ -1,6 +1,5 @@
 package com.mrxu.framework.boot.web;
 
-import com.mrxu.framework.boot.entity.RpcRequestInfo;
 import com.mrxu.framework.common.util.StrFunc;
 import org.springframework.web.context.request.RequestAttributes;
 import org.springframework.web.context.request.RequestContextHolder;
@@ -20,8 +19,6 @@ import java.util.*;
  */
 public class ServletUtils {
 
-    public static ThreadLocal<RpcRequestInfo> rpcRequestInfo = new ThreadLocal<RpcRequestInfo>();
-
     /**
      * 获取String参数
      */