feat(ai): 增加公司 ID 筛选功能

Administrator

2 days ago 4bfe57af40d98331721124837950c4329ad12baf

 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);
        }
    }
}