From 85e05f4a7c93fb4715e26b1b9ee04fab944abf15 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Fri, 01 Aug 2025 16:31:41 +0800
Subject: [PATCH] feat(ai): 新增 AI 陪练配置功能

---
 src/main/java/cc/mrbird/febs/ai/service/impl/AiProductServiceImpl.java |   28 ++++++++++++++++++++++++++++
 1 files changed, 28 insertions(+), 0 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductServiceImpl.java
index a67e99d..6feb107 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductServiceImpl.java
@@ -5,6 +5,7 @@
 import cc.mrbird.febs.ai.req.AdminMoveChooseInfoDto;
 import cc.mrbird.febs.ai.service.AiProductCategoryService;
 import cc.mrbird.febs.ai.service.AiProductPointLinkService;
+import cc.mrbird.febs.ai.service.AiProductRoleLinkService;
 import cc.mrbird.febs.ai.service.AiProductService;
 import cc.mrbird.febs.ai.util.UUID;
 import cc.mrbird.febs.common.entity.FebsResponse;
@@ -40,6 +41,7 @@
     private final AiProductMapper aiProductMapper;
     private final AiProductCategoryService aiProductCategoryService;
     private final AiProductPointLinkService aiProductPointLinkService;
+    private final AiProductRoleLinkService aiProductRoleLinkService;
 
     @Override
     public AiProduct getById(String id) {
@@ -189,4 +191,30 @@
         }
         return new FebsResponse().success().message("操作成功");
     }
+
+    @Override
+    public FebsResponse productRoleSet(AdminMoveChooseInfoDto dto) {
+
+        String chooseId = dto.getChooseId();
+        List<String> chooseIds = dto.getChooseIds();
+        AiProduct aiProduct = this.getById(chooseId);
+        if (ObjectUtil.isNotNull(aiProduct)) {
+            aiProductRoleLinkService.deleteByQuery(
+                    Wrappers.lambdaQuery(AiProductRoleLink.class)
+                            .eq(AiProductRoleLink::getProductId,chooseId)
+            );
+            if(CollUtil.isNotEmpty(chooseIds)){
+                Date createdTime = new Date();
+                for (String item : chooseIds){
+                    AiProductRoleLink entity = new AiProductRoleLink();
+                    entity.setId(UUID.getSimpleUUIDString());
+                    entity.setProductId(chooseId);
+                    entity.setProductRoleId(item);
+                    entity.setCreatedTime(createdTime);
+                    aiProductRoleLinkService.getBaseMapper().insert(entity);
+                }
+            }
+        }
+        return new FebsResponse().success().message("操作成功");
+    }
 }

--
Gitblit v1.9.1