From 6601d3ecbdcb94a014eaebe275bf824e2c25ef1a Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Fri, 08 Jul 2022 19:01:35 +0800
Subject: [PATCH] fix
---
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