From 5e54f37b9bfd01874c22c017c768adb2f8fee24d Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Tue, 12 Jul 2022 11:47:39 +0800
Subject: [PATCH] fix column update

---
 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