1

src/main/java/com/aida/seller/common/context/UserContext.java

@@ -0,0 +1,24 @@
+package com.aida.seller.common.context;
+
+import com.aida.seller.model.vo.AuthPrincipalVo;
+
+public class UserContext {
+    private static final ThreadLocal<AuthPrincipalVo> userHolder = new ThreadLocal<>();
+
+    public static AuthPrincipalVo getUserHolder() {
+        return userHolder.get();
+    }
+
+    public static void delete() {
+        userHolder.remove();
+    }
+
+    public static void setUserHolder(AuthPrincipalVo authPrincipalVo) {
+        userHolder.set(authPrincipalVo);
+    }
+
+    public static Long getUserId() {
+        AuthPrincipalVo holder = userHolder.get();
+        return holder != null ? holder.getId() : null;
+    }
+}