소스 검색

rpc 增加调用方信息

panh 5 년 전
부모
커밋
2c4bcd38f2
1개의 변경된 파일5개의 추가작업 그리고 1개의 파일을 삭제
  1. 5 1
      framework-boot/src/main/java/com/mrxu/framework/boot/entity/RpcRequestInfo.java

+ 5 - 1
framework-boot/src/main/java/com/mrxu/framework/boot/entity/RpcRequestInfo.java

@@ -10,7 +10,11 @@ public class RpcRequestInfo {
     private static ThreadLocal<RpcRequestInfo> rpcRequestInfo = new ThreadLocal<RpcRequestInfo>();
 
     public static RpcRequestInfo get() {
-        return rpcRequestInfo.get();
+        RpcRequestInfo rs = rpcRequestInfo.get();
+        if(rs == null) {
+            rs = new RpcRequestInfo();
+        }
+        return rs;
     }
 
     public static void remove() {