From 77e1b14b622a17b2ad824bae24209dfba9ee9e6e Mon Sep 17 00:00:00 2001 From: wzy <wzy19931122ai@163.com> Date: Sun, 07 Aug 2022 14:25:40 +0800 Subject: [PATCH] fix:addBelong --- 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