浏览代码

新增:获取登录用户绑定供应商。

lym 4 年之前
父节点
当前提交
f4dbf23ca0
共有 1 个文件被更改,包括 24 次插入0 次删除
  1. 24 0
      forest-admin/admin-server/src/main/java/com/hwrj/cloud/admin/util/CurrUserUtil.java

+ 24 - 0
forest-admin/admin-server/src/main/java/com/hwrj/cloud/admin/util/CurrUserUtil.java

@@ -52,8 +52,32 @@ public class CurrUserUtil {
      */
     public static int getUserId() {
         UmsAdmin umsAdmin = getUmsAdmin();
+        if (StringUtils.isEmpty(umsAdmin.getId())){
+            throw new GlobalException(1,"请登录!");
+        }
         long userId = umsAdmin.getId();
         return (int) userId;
     }
+
+    /**
+     * 获取用户绑定供应商id
+     */
+    public static long getCompId() {
+        UmsAdmin umsAdmin = getUmsAdmin();
+        if (StringUtils.isEmpty(umsAdmin.getCompId())){
+            throw new GlobalException(1,"用户未绑定供应商!");
+        }
+        long compId = umsAdmin.getCompId();
+        return compId;
+    }
+
+    /**
+     * 获取用户绑定供应商名称
+     */
+    public static String getCompName() {
+        UmsAdmin umsAdmin = getUmsAdmin();
+        String compName = umsAdmin.getCompName();
+        return compName;
+    }
 }