From 4bfe57af40d98331721124837950c4329ad12baf Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Mon, 15 Sep 2025 15:07:03 +0800
Subject: [PATCH] feat(ai): 增加公司 ID 筛选功能
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiProductQuestionItemServiceImpl.java | 26 ++++++++++++++++++++++++++
1 files changed, 26 insertions(+), 0 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductQuestionItemServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductQuestionItemServiceImpl.java
index 5f156c3..f4646f0 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductQuestionItemServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductQuestionItemServiceImpl.java
@@ -3,10 +3,17 @@
import cc.mrbird.febs.ai.entity.AiProductQuestionItem;
import cc.mrbird.febs.ai.mapper.AiProductQuestionItemMapper;
import cc.mrbird.febs.ai.service.AiProductQuestionItemService;
+import cn.hutool.core.collection.CollUtil;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
+
+import java.util.List;
+import java.util.Set;
+import java.util.stream.Collectors;
/**
* AI产品题目子表 Service实现类
@@ -19,4 +26,23 @@
@Transactional
public class AiProductQuestionItemServiceImpl extends ServiceImpl<AiProductQuestionItemMapper, AiProductQuestionItem> implements AiProductQuestionItemService {
+ private final AiProductQuestionItemMapper aiProductQuestionItemMapper;
+
+
+ @Override
+ public List<AiProductQuestionItem> getListByQuery(LambdaQueryWrapper<AiProductQuestionItem> query) {
+ return aiProductQuestionItemMapper.selectList( query);
+ }
+
+ @Override
+ public void deleteByQuestionId(String id) {
+ LambdaQueryWrapper<AiProductQuestionItem> query = Wrappers.lambdaQuery(AiProductQuestionItem.class);
+ query.eq(AiProductQuestionItem::getProductQuestionId, id);
+
+ List<AiProductQuestionItem> listByQuery = this.getListByQuery(query);
+ if(CollUtil.isNotEmpty(listByQuery)){
+ Set<String> collect = listByQuery.stream().map(AiProductQuestionItem::getId).collect(Collectors.toSet());
+ this.removeByIds(collect);
+ }
+ }
}
--
Gitblit v1.9.1