From 41f55c835ec9cecea351101c0cc98e1fdd1129ba Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Mon, 11 Jul 2022 10:09:53 +0800
Subject: [PATCH] add exception deal
---
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