From 577bef7f8ece263c4bc272bab23fcc8aa4b6083b Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Tue, 16 Sep 2025 11:56:01 +0800
Subject: [PATCH] feat(ai): 产品知识点增加分类功能
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiProductPointServiceImpl.java | 23 +++++++++++++++++++++++
1 files changed, 23 insertions(+), 0 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductPointServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductPointServiceImpl.java
index 9d0c44e..c496f82 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductPointServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductPointServiceImpl.java
@@ -1,13 +1,16 @@
package cc.mrbird.febs.ai.service.impl;
import cc.mrbird.febs.ai.entity.AiProduct;
+import cc.mrbird.febs.ai.entity.AiProductCategory;
import cc.mrbird.febs.ai.entity.AiProductPoint;
import cc.mrbird.febs.ai.mapper.AiProductPointMapper;
+import cc.mrbird.febs.ai.service.AiProductCategoryService;
import cc.mrbird.febs.ai.service.AiProductPointLinkService;
import cc.mrbird.febs.ai.service.AiProductPointService;
import cc.mrbird.febs.ai.util.UUID;
import cc.mrbird.febs.common.entity.FebsResponse;
import cc.mrbird.febs.common.entity.QueryRequest;
+import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
@@ -21,6 +24,9 @@
import java.util.Date;
import java.util.List;
+import java.util.Map;
+import java.util.Set;
+import java.util.stream.Collectors;
/**
* AI产品知识点 Service实现类
@@ -35,6 +41,7 @@
private final AiProductPointMapper aiProductPointMapper;
private final AiProductPointLinkService aiProductPointLinkService;
+ private final AiProductCategoryService aiProductCategoryService;
@Override
public AiProductPoint getById(String id) {
@@ -49,6 +56,20 @@
query.eq(AiProductPoint::getCompanyId, dto.getCompanyId());
}
Page<AiProductPoint> pages = aiProductPointMapper.selectPage(page, query);
+ List<AiProductPoint> records = pages.getRecords();
+ if (CollUtil.isNotEmpty( records)){
+ Set<String> collect = records.stream().map(AiProductPoint::getProductCategoryId).collect(Collectors.toSet());
+
+ if(CollUtil.isNotEmpty( collect)){
+ Map<String, AiProductCategory> map = aiProductCategoryService.selectMapByIds(collect);
+ for (AiProductPoint record : records){
+ AiProductCategory orDefault = map.getOrDefault(record.getProductCategoryId(), null);
+ if(ObjectUtil.isNotNull(orDefault)){
+ record.setProductCategoryName(orDefault.getName());
+ }
+ }
+ }
+ }
return pages;
}
@@ -56,6 +77,7 @@
public FebsResponse add(AiProductPoint dto) {
AiProductPoint entity = new AiProductPoint();
entity.setId(UUID.getSimpleUUIDString());
+ entity.setProductCategoryId(dto.getProductCategoryId());
entity.setCompanyId(dto.getCompanyId());
entity.setIsNormal(dto.getIsNormal() );
entity.setFinderUserName(dto.getFinderUserName());
@@ -76,6 +98,7 @@
this.update(null,
Wrappers.lambdaUpdate(AiProductPoint.class)
.set(AiProductPoint::getIsNormal, dto.getIsNormal())
+ .set(AiProductPoint::getProductCategoryId, dto.getProductCategoryId())
.set(AiProductPoint::getFinderUserName, dto.getFinderUserName())
.set(AiProductPoint::getFeedId, dto.getFeedId())
.set(AiProductPoint::getTitle, dto.getTitle())
--
Gitblit v1.9.1