From edd5bf1c50327c7da1c18f82b46f8cee64d5bab1 Mon Sep 17 00:00:00 2001
From: KKSU <15274802129@163.com>
Date: Tue, 09 Jan 2024 14:37:59 +0800
Subject: [PATCH] 20220812

---
 src/main/java/com/xcong/farmer/cms/modules/system/util/LoginUserUtil.java |    4 ++++
 1 files changed, 4 insertions(+), 0 deletions(-)

diff --git a/src/main/java/com/xcong/farmer/cms/modules/system/util/LoginUserUtil.java b/src/main/java/com/xcong/farmer/cms/modules/system/util/LoginUserUtil.java
index ac1f9e0..29c76ce 100644
--- a/src/main/java/com/xcong/farmer/cms/modules/system/util/LoginUserUtil.java
+++ b/src/main/java/com/xcong/farmer/cms/modules/system/util/LoginUserUtil.java
@@ -16,4 +16,8 @@
         HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.currentRequestAttributes()).getRequest();
         return (UserEntity) request.getSession().getAttribute("user");
     }
+
+    public static Long getCompanyId() {
+        return getLoginUser().getCompanyId() == null ? 0L : getLoginUser().getCompanyId();
+    }
 }

--
Gitblit v1.9.1