From 49dbcdadd860b7613063ee618fbe38a0593d5bd9 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Mon, 11 Jul 2022 15:45:29 +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