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

Administrator

2 days ago 4bfe57af40d98331721124837950c4329ad12baf

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

@@ -3,10 +3,18 @@
import cc.mrbird.febs.ai.entity.AiProductPointLink;
import cc.mrbird.febs.ai.mapper.AiProductPointLinkMapper;
import cc.mrbird.febs.ai.service.AiProductPointLinkService;
import cc.mrbird.febs.ai.service.AiProductPointService;
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;

/**
 * @author Administrator
@@ -16,4 +24,31 @@
@Transactional
public class AiProductPointLinkServiceImpl extends ServiceImpl<AiProductPointLinkMapper, AiProductPointLink> implements AiProductPointLinkService {

    private final AiProductPointLinkMapper aiProductPointLinkMapper;
    @Override
    public void deleteByProductPointId(String id) {
        List<AiProductPointLink> aiProductPointLinks = this.selectListByPointId(id);
        if(CollUtil.isNotEmpty(aiProductPointLinks)){
            Set<String> collect = aiProductPointLinks.stream().map(AiProductPointLink::getId).collect(Collectors.toSet());
            aiProductPointLinkMapper.deleteBatchIds(collect);
        }
    }

    @Override
    public List<AiProductPointLink> selectListByPointId(String pointId) {
        LambdaQueryWrapper<AiProductPointLink> query = Wrappers.lambdaQuery(AiProductPointLink.class);
        query.eq(AiProductPointLink::getProductPointId, pointId);
        List<AiProductPointLink> aiProductPointLinks = aiProductPointLinkMapper.selectList(query);
        return aiProductPointLinks;
    }

    @Override
    public List<AiProductPointLink> selectListByQuery(LambdaQueryWrapper<AiProductPointLink> query) {
        return aiProductPointLinkMapper.selectList( query);
    }

    @Override
    public void deleteByQuery(LambdaQueryWrapper<AiProductPointLink> query) {
        aiProductPointLinkMapper.delete(query);
    }
}