From 5fd478fcee23bbdac0fe4fbe9becdd2a28d14c56 Mon Sep 17 00:00:00 2001
From: Junjie <xjj@123>
Date: 星期六, 13 七月 2024 14:27:04 +0800
Subject: [PATCH] #

---
 zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/entity/User.java |   36 ++++++++++++++++++++++++++----------
 1 files changed, 26 insertions(+), 10 deletions(-)

diff --git a/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/entity/User.java b/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/entity/User.java
index 7fe7403..648abb0 100644
--- a/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/entity/User.java
+++ b/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/entity/User.java
@@ -1,9 +1,12 @@
 package com.zy.asrs.wcs.system.entity;
 
 import com.baomidou.mybatisplus.annotation.*;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.zy.asrs.framework.common.Cools;
 import com.zy.asrs.framework.common.SpringUtils;
+import com.zy.asrs.wcs.system.service.DeptService;
 import com.zy.asrs.wcs.system.service.HostService;
+import com.zy.asrs.wcs.system.service.UserRoleService;
 import com.zy.asrs.wcs.system.service.UserService;
 import io.swagger.annotations.ApiModelProperty;
 import lombok.Data;
@@ -164,6 +167,9 @@
     private String memo;
 
     @TableField(exist = false)
+    private Long[] roleIds;
+
+    @TableField(exist = false)
     private List<Role> roles;
 
     @TableField(exist = false)
@@ -247,14 +253,14 @@
         }
     }
 
-//    public String getDeptId$(){
-//        DeptService service = SpringUtils.getBean(DeptService.class);
-//        Dept dept = service.getById(this.deptId);
-//        if (!Cools.isEmpty(dept)){
-//            return String.valueOf(dept.getName());
-//        }
-//        return null;
-//    }
+    public String getDeptId$(){
+        DeptService service = SpringUtils.getBean(DeptService.class);
+        Dept dept = service.getById(this.deptId);
+        if (!Cools.isEmpty(dept)){
+            return String.valueOf(dept.getName());
+        }
+        return null;
+    }
 
     public String getHostId$(){
         HostService service = SpringUtils.getBean(HostService.class);
@@ -300,7 +306,7 @@
         UserService service = SpringUtils.getBean(UserService.class);
         User user = service.getById(this.createBy);
         if (!Cools.isEmpty(user)){
-            return String.valueOf(user.getId());
+            return String.valueOf(user.getNickname());
         }
         return null;
     }
@@ -316,11 +322,21 @@
         UserService service = SpringUtils.getBean(UserService.class);
         User user = service.getById(this.updateBy);
         if (!Cools.isEmpty(user)){
-            return String.valueOf(user.getId());
+            return String.valueOf(user.getNickname());
         }
         return null;
     }
 
+    public Long[] getRoleIds() {
+        return this.roleIds;
+    }
+
+    public Long[] getUserRoleIds() {
+        UserRoleService service = SpringUtils.getBean(UserRoleService.class);
+        List<UserRole> list = service.list(new LambdaQueryWrapper<UserRole>().eq(UserRole::getUserId, this.id));
+        return list.stream().map(UserRole::getRoleId).toArray(Long[]::new);
+    }
+
     @Override
     public boolean isAccountNonExpired() {
         return true;

--
Gitblit v1.9.1