package cc.mrbird.febs.ai.service.impl;
|
|
import cc.mrbird.febs.ai.entity.AiMember;
|
import cc.mrbird.febs.ai.entity.AiMemberTalk;
|
import cc.mrbird.febs.ai.mapper.AiMemberMapper;
|
import cc.mrbird.febs.ai.req.member.ApiMemberPageDto;
|
import cc.mrbird.febs.ai.res.member.ApiMemberPageVo;
|
import cc.mrbird.febs.ai.service.AiMemberService;
|
import cc.mrbird.febs.ai.service.AiMemberTalkService;
|
import cc.mrbird.febs.common.entity.FebsResponse;
|
import cc.mrbird.febs.common.utils.LoginUserUtil;
|
import cc.mrbird.febs.mall.entity.MallMember;
|
import cc.mrbird.febs.mall.mapper.MallMemberMapper;
|
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.plugins.pagination.Page;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import lombok.RequiredArgsConstructor;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.stereotype.Service;
|
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.Set;
|
import java.util.stream.Collectors;
|
|
/**
|
* @author Administrator
|
*/
|
@Slf4j
|
@Service
|
@RequiredArgsConstructor
|
public class AiMemberServiceImpl extends ServiceImpl<AiMemberMapper, AiMember> implements AiMemberService {
|
|
private final AiMemberMapper aiMemberMapper;
|
private final MallMemberMapper mallMemberMapper;
|
private final AiMemberTalkService aiMemberTalkService;
|
@Override
|
public AiMember getById(String id) {
|
return aiMemberMapper.selectById( id);
|
}
|
|
@Override
|
public FebsResponse myTeam(ApiMemberPageDto dto) {
|
List<ApiMemberPageVo> objects = new ArrayList<>();
|
|
String companyId = LoginUserUtil.getLoginUser().getCompanyId();
|
|
// 创建分页对象，传入当前页和每页大小
|
Page<AiMember> page = new Page<>(dto.getPageNow(), dto.getPageSize());
|
|
LambdaQueryWrapper<AiMember> queryWrapper = Wrappers.lambdaQuery(AiMember.class);
|
queryWrapper.eq(AiMember::getCompanyId, companyId);
|
Page<AiMember> pageListByQuery = aiMemberMapper.selectPage(page, queryWrapper);
|
List<AiMember> records = pageListByQuery.getRecords();
|
if (CollUtil.isNotEmpty( records)){
|
objects = buildMemberPages(companyId,records,objects);
|
}
|
return new FebsResponse().success().data(objects);
|
}
|
|
private List<ApiMemberPageVo> buildMemberPages(String companyId,List<AiMember> records,List<ApiMemberPageVo> objects) {
|
/**
|
* 获取用户信息
|
*/
|
//stream流操作records，获取全部的id
|
Set<String> memberIds = records.stream().map(AiMember::getId).collect(Collectors.toSet());
|
|
List<MallMember> mallMembers = mallMemberMapper.selectList(
|
Wrappers.lambdaQuery(MallMember.class)
|
.select(MallMember::getMemberUuid, MallMember::getName, MallMember::getRealName)
|
.in(MallMember::getMemberUuid, memberIds)
|
.eq(MallMember::getCompanyId, companyId)
|
);
|
//Stream流操作mallMembers，获取一个map<memberUuid,mallMember>的对象
|
Map<String, MallMember> mallMemberMap = mallMembers.stream().collect(Collectors.toMap(MallMember::getMemberUuid, mallMember -> mallMember));
|
|
/**
|
* 获取用户联系次数
|
* ai陪练
|
* ai答题
|
*/
|
List<AiMemberTalk> aiMemberTalks = aiMemberTalkService.getListByCompanyId(companyId);
|
//Stream流操作aiMemberTalks，获取一个map<memberUuid,aiMemberTalk>的对象
|
|
|
for (AiMember aiMember : records){
|
ApiMemberPageVo apiMemberPageVo = new ApiMemberPageVo();
|
apiMemberPageVo.setMemberUuid(aiMember.getId());
|
|
//判断mallMemberMap中是否存在该会员
|
apiMemberPageVo.setMemberName(mallMemberMap.containsKey(aiMember.getId()) ? mallMemberMap.get(aiMember.getId()).getName() : "");
|
apiMemberPageVo.setNickName(mallMemberMap.containsKey(aiMember.getId()) ? mallMemberMap.get(aiMember.getId()).getName() : "");
|
|
objects.add(apiMemberPageVo);
|
}
|
return objects;
|
}
|
}
|
