From 34c1ffb1702a7a541b7e1af101f1ee42ad85cd42 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9B=B9=E9=B9=8F=E9=A3=9E?= <rakor2010@gmail.com>
Date: Wed, 21 Jan 2026 14:01:10 +0800
Subject: [PATCH] =?UTF-8?q?=E5=86=85=E9=83=A8=E8=B4=A6=E5=8F=B7=E6=B7=BB?=
 =?UTF-8?q?=E5=8A=A0=E7=9B=B4=E9=94=80=E7=B1=BB=E5=88=AB=E5=AD=97=E6=AE=B5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../common/pojo/request/AccountAddRequest.java  |  5 +++++
 .../common/pojo/vo/AdminUserVO.java             | 10 ++++++++++
 .../repository/entity/AdminUser.java            |  5 +++++
 .../service/impl/AdminUserServiceImpl.java      | 17 +++++++++++++++++
 4 files changed, 37 insertions(+)

diff --git a/nflg-mobilebroken-common/src/main/java/com/nflg/mobilebroken/common/pojo/request/AccountAddRequest.java b/nflg-mobilebroken-common/src/main/java/com/nflg/mobilebroken/common/pojo/request/AccountAddRequest.java
index 3dfd6dce..b18e8c1d 100644
--- a/nflg-mobilebroken-common/src/main/java/com/nflg/mobilebroken/common/pojo/request/AccountAddRequest.java
+++ b/nflg-mobilebroken-common/src/main/java/com/nflg/mobilebroken/common/pojo/request/AccountAddRequest.java
@@ -54,4 +54,9 @@ public class AccountAddRequest {
      */
     @JsonProperty("isGongFuHandler")
     private boolean isGongFuHandler = false;
+
+    /**
+     * 直销类别（国内、国外），字典id
+     */
+    private Long categoryId;
 }
diff --git a/nflg-mobilebroken-common/src/main/java/com/nflg/mobilebroken/common/pojo/vo/AdminUserVO.java b/nflg-mobilebroken-common/src/main/java/com/nflg/mobilebroken/common/pojo/vo/AdminUserVO.java
index 9c153e9c..1dedc6ef 100644
--- a/nflg-mobilebroken-common/src/main/java/com/nflg/mobilebroken/common/pojo/vo/AdminUserVO.java
+++ b/nflg-mobilebroken-common/src/main/java/com/nflg/mobilebroken/common/pojo/vo/AdminUserVO.java
@@ -84,5 +84,15 @@ public class AdminUserVO {
     @JsonProperty("isGongfu")
     private boolean isGongfu;
 
+    /**
+     * 直销类别（国内、国外），字典id
+     */
+    private Long categoryId;
+
+    /**
+     * 直销类别
+     */
+    private String categoryName;
+
     private List<AdminUserVO> children = new ArrayList<>();
 }
diff --git a/nflg-mobilebroken-repository/src/main/java/com/nflg/mobilebroken/repository/entity/AdminUser.java b/nflg-mobilebroken-repository/src/main/java/com/nflg/mobilebroken/repository/entity/AdminUser.java
index d594a7d2..dd17a917 100644
--- a/nflg-mobilebroken-repository/src/main/java/com/nflg/mobilebroken/repository/entity/AdminUser.java
+++ b/nflg-mobilebroken-repository/src/main/java/com/nflg/mobilebroken/repository/entity/AdminUser.java
@@ -132,4 +132,9 @@ public class AdminUser implements Serializable {
      */
     @TableField("is_gongfu")
     private boolean gongfu;
+
+    /**
+     * 直销类别（国内、国外），字典id
+     */
+    private Long categoryId;
 }
diff --git a/nflg-mobilebroken-repository/src/main/java/com/nflg/mobilebroken/repository/service/impl/AdminUserServiceImpl.java b/nflg-mobilebroken-repository/src/main/java/com/nflg/mobilebroken/repository/service/impl/AdminUserServiceImpl.java
index 8aacf2e1..b4a01309 100644
--- a/nflg-mobilebroken-repository/src/main/java/com/nflg/mobilebroken/repository/service/impl/AdminUserServiceImpl.java
+++ b/nflg-mobilebroken-repository/src/main/java/com/nflg/mobilebroken/repository/service/impl/AdminUserServiceImpl.java
@@ -67,6 +67,9 @@ public class AdminUserServiceImpl extends ServiceImpl<AdminUserMapper, AdminUser
     @Resource
     private IDictionaryItemTranslateService dictionaryItemTranslateService;
 
+    @Resource
+    private IDictionaryItemService dictionaryItemService;
+
     @Override
     public AdminUser getUser(String userName, String password) {
         AdminUser user = lambdaQuery()
@@ -99,6 +102,7 @@ public class AdminUserServiceImpl extends ServiceImpl<AdminUserMapper, AdminUser
                     .setProductLine(request.getProductLine())
                     .setGongFuHandler(request.isGongFuHandler())
                     .setGongfu(StrUtil.contains(request.getProductLine(), Constant.MOBILE_BROKEN))
+                    .setCategoryId(request.getCategoryId())
                     .setUpdateBy(AdminUserUtil.getUserId())
                     .setUpdateTime(LocalDateTime.now());
             updateById(user);
@@ -115,6 +119,7 @@ public class AdminUserServiceImpl extends ServiceImpl<AdminUserMapper, AdminUser
                     .setProductLine(request.getProductLine())
                     .setGongFuHandler(request.isGongFuHandler())
                     .setGongfu(StrUtil.contains(request.getProductLine(), Constant.MOBILE_BROKEN))
+                    .setCategoryId(request.getCategoryId())
                     .setTitleId(request.getTitleId())
                     .setCreateBy(AdminUserUtil.getUserId())
                     .setCreateTime(LocalDateTime.now());
@@ -164,6 +169,12 @@ public class AdminUserServiceImpl extends ServiceImpl<AdminUserMapper, AdminUser
         List<AdminUser> users = lambdaQuery().list();
         List<AdminUserRoleMap> roleMaps = adminUserRoleMapService.lambdaQuery().list();
         List<AdminRole> roles = roleService.lambdaQuery().list();
+        List<DictionaryItem> categories = dictionaryItemService.lambdaQuery()
+                .in(DictionaryItem::getId, page.getRecords().stream()
+                        .map(AdminUser::getCategoryId)
+                        .filter(Objects::nonNull)
+                        .collect(Collectors.toList())
+                ).list();
         pageData.setItems(page.getRecords().stream().map(u -> {
             AdminUser userCreate = users.stream()
                     .filter(it -> Objects.equals(it.getId(), u.getCreateBy()))
@@ -188,6 +199,10 @@ public class AdminUserServiceImpl extends ServiceImpl<AdminUserMapper, AdminUser
                     .filter(it -> Objects.equals(it.getUserId(), u.getId()))
                     .map(AdminUserRoleMap::getRoleId)
                     .collect(Collectors.toSet());
+            DictionaryItem category = categories.stream()
+                    .filter(c -> Objects.equals(c.getId(), u.getCategoryId()))
+                    .findFirst()
+                    .orElse(null);
             return new AdminUserVO()
                     .setId(u.getId())
                     .setUserCode(u.getUserCode())
@@ -209,6 +224,8 @@ public class AdminUserServiceImpl extends ServiceImpl<AdminUserMapper, AdminUser
                     .setGongFuHandler(u.isGongFuHandler())
                     .setProductLine(u.getProductLine())
                     .setGongfu(u.isGongfu())
+                    .setCategoryId(u.getCategoryId())
+                    .setCategoryName(Objects.isNull(category) ? "" : category.getName())
                     .setRoles(
                             roles.stream()
                                     .filter(r -> rmaps.contains(r.getId()))