Administrator
2 days ago 3b05389a5bcf93ead612761835952699a4b2e302
src/main/java/cc/mrbird/febs/ai/service/impl/AiProductRoleLinkServiceImpl.java
@@ -1,12 +1,20 @@
package cc.mrbird.febs.ai.service.impl;
import cc.mrbird.febs.ai.entity.AiProductPointLink;
import cc.mrbird.febs.ai.entity.AiProductRoleLink;
import cc.mrbird.febs.ai.mapper.AiProductRoleLinkMapper;
import cc.mrbird.febs.ai.service.AiProductRoleLinkService;
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 lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;
/**
 * @author Administrator
@@ -15,4 +23,33 @@
@Service
@RequiredArgsConstructor
public class AiProductRoleLinkServiceImpl extends ServiceImpl<AiProductRoleLinkMapper, AiProductRoleLink> implements AiProductRoleLinkService {
    private final AiProductRoleLinkMapper aiProductRoleLinkMapper;
    @Override
    public void deleteByProductRoleId(String id) {
        List<AiProductRoleLink> list = this.selectListByRoleId(id);
        if(CollUtil.isNotEmpty(list)){
            Set<String> collect = list.stream().map(AiProductRoleLink::getId).collect(Collectors.toSet());
            aiProductRoleLinkMapper.deleteBatchIds(collect);
        }
    }
    @Override
    public List<AiProductRoleLink> selectListByRoleId(String id) {
        LambdaQueryWrapper<AiProductRoleLink> query = Wrappers.lambdaQuery(AiProductRoleLink.class);
        query.eq(AiProductRoleLink::getProductRoleId, id);
        List<AiProductRoleLink> list = aiProductRoleLinkMapper.selectList(query);
        return list;
    }
    @Override
    public List<AiProductRoleLink> selectListByQuery(LambdaQueryWrapper<AiProductRoleLink> query) {
        return aiProductRoleLinkMapper.selectList( query);
    }
    @Override
    public void deleteByQuery(LambdaQueryWrapper<AiProductRoleLink> query) {
        aiProductRoleLinkMapper.delete(query);
    }
}