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
 */
@Slf4j
@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);
    }
}