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

Administrator

2 days ago 4bfe57af40d98331721124837950c4329ad12baf

 src/main/java/cc/mrbird/febs/ai/service/impl/AiProductServiceImpl.java

@@ -3,15 +3,13 @@
import cc.mrbird.febs.ai.entity.*;
import cc.mrbird.febs.ai.mapper.AiProductMapper;
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.service.*;
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;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
@@ -42,6 +40,7 @@
    private final AiProductCategoryService aiProductCategoryService;
    private final AiProductPointLinkService aiProductPointLinkService;
    private final AiProductRoleLinkService aiProductRoleLinkService;
    private final AiProductQuestionLinkService aiProductQuestionLinkService;

    @Override
    public AiProduct getById(String id) {
@@ -53,7 +52,7 @@

        Page<AiProduct> page = new Page<>(request.getPageNum(), request.getPageSize());
        LambdaQueryWrapper<AiProduct> query = Wrappers.lambdaQuery(AiProduct.class);
        if (ObjectUtil.isNotNull(dto.getProductCategoryId())){
        if (StrUtil.isNotEmpty(dto.getProductCategoryId())){
            query.eq(AiProduct::getProductCategoryId, dto.getProductCategoryId());
        }
        query.ne(AiProduct::getState, 2);
@@ -159,9 +158,10 @@
    }

    @Override
    public List<AiProduct> selectList() {
    public List<AiProduct> selectList(String companyId) {
        return aiProductMapper.selectList(
                Wrappers.lambdaQuery(AiProduct.class)
                        .eq(AiProduct::getCompanyId, companyId)
                .ne(AiProduct::getState, 2)
        );
    }
@@ -217,4 +217,29 @@
        }
        return new FebsResponse().success().message("操作成功");
    }

    @Override
    public FebsResponse productQuestionSet(AdminMoveChooseInfoDto dto) {
        String chooseId = dto.getChooseId();
        List<String> chooseIds = dto.getChooseIds();
        AiProduct aiProduct = this.getById(chooseId);
        if (ObjectUtil.isNotNull(aiProduct)) {
            aiProductQuestionLinkService.deleteByQuery(
                    Wrappers.lambdaQuery(AiProductQuestionLink.class)
                            .eq(AiProductQuestionLink::getProductId,chooseId)
            );
            if(CollUtil.isNotEmpty(chooseIds)){
                Date createdTime = new Date();
                for (String item : chooseIds){
                    AiProductQuestionLink entity = new AiProductQuestionLink();
                    entity.setId(UUID.getSimpleUUIDString());
                    entity.setProductId(chooseId);
                    entity.setProductQuestionId(item);
                    entity.setCreatedTime(createdTime);
                    aiProductQuestionLinkService.getBaseMapper().insert(entity);
                }
            }
        }
        return new FebsResponse().success().message("操作成功");
    }
}