package cc.mrbird.febs.ai.service.impl;
|
|
import cc.mrbird.febs.ai.entity.*;
|
import cc.mrbird.febs.ai.mapper.AiMemberMapper;
|
import cc.mrbird.febs.ai.req.member.ApiMemberTeamPageDto;
|
import cc.mrbird.febs.ai.req.member.ApiMemberTeamPracticeDto;
|
import cc.mrbird.febs.ai.req.member.ApiMemberTeamStudyDto;
|
import cc.mrbird.febs.ai.res.member.ApiMemberTeamPageVo;
|
import cc.mrbird.febs.ai.res.member.ApiMemberTeamPracticeVo;
|
import cc.mrbird.febs.ai.res.member.ApiMemberTeamStudyVo;
|
import cc.mrbird.febs.ai.res.productPoint.ApiProductPointListVo;
|
import cc.mrbird.febs.ai.service.*;
|
import cc.mrbird.febs.common.entity.FebsResponse;
|
import cc.mrbird.febs.common.exception.FebsException;
|
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 cn.hutool.core.date.DateUtil;
|
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.*;
|
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;
|
private final AiMemberAnswerService aiMemberAnswerService;
|
private final AiMemberPointService aiMemberPointService;
|
private final AiProductService aiProductService;
|
private final AiProductPointService aiProductPointService;
|
@Override
|
public AiMember getById(String id) {
|
return aiMemberMapper.selectById( id);
|
}
|
|
@Override
|
public FebsResponse myTeam(ApiMemberTeamPageDto dto) {
|
List<ApiMemberTeamPageVo> objects = new ArrayList<>();
|
|
String companyId = LoginUserUtil.getLoginUser().getCompanyId();
|
String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid();
|
Integer checkOrder = LoginUserUtil.getLoginUser().getCheckOrder();
|
if (1 != checkOrder){
|
throw new FebsException("非管理员，无权限查看");
|
}
|
|
// 创建分页对象，传入当前页和每页大小
|
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(memberUuid,companyId,records,objects);
|
}
|
return new FebsResponse().success().data(objects);
|
}
|
|
private List<ApiMemberTeamPageVo> buildMemberPages(String memberUuid, String companyId, List<AiMember> records, List<ApiMemberTeamPageVo> 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<memberId,Integer>的对象,integer代表按照memberId分组的数量
|
Map<String, Integer> memberIdToCntMapTalk = aiMemberTalks.stream()
|
.collect(Collectors.groupingBy(
|
AiMemberTalk::getMemberId,
|
Collectors.summingInt(t -> 1)
|
));
|
|
List<AiMemberAnswer> aiMemberAnswers = aiMemberAnswerService.getListByCompanyId(companyId);
|
//stream流操作aiMemberAnswers，获取一个Map<memberId,Integer>的对象,integer代表按照memberId分组的数量
|
Map<String, Integer> memberIdToCntMapAnswer = aiMemberAnswers.stream()
|
.collect(Collectors.groupingBy(
|
AiMemberAnswer::getMemberId,
|
Collectors.summingInt(a -> 1)
|
));
|
/**
|
* 用户学习总时长
|
*/
|
List<AiMemberPoint> aiMemberPoints = aiMemberPointService.getListByCompanyId(companyId);
|
Map<String, Integer> memberStudyTimeMap = aiMemberPoints.stream().collect(Collectors.groupingBy(
|
AiMemberPoint::getMemberId,
|
Collectors.summingInt(AiMemberPoint::getTotalTime)
|
));
|
|
|
for (AiMember aiMember : records){
|
String aiMemberId = aiMember.getId();
|
ApiMemberTeamPageVo apiMemberTeamPageVo = new ApiMemberTeamPageVo();
|
apiMemberTeamPageVo.setType(0);
|
apiMemberTeamPageVo.setMemberUuid(aiMemberId);
|
if (memberUuid.equals(aiMemberId)){
|
apiMemberTeamPageVo.setType(1);
|
}
|
|
//判断mallMemberMap中是否存在该会员
|
apiMemberTeamPageVo.setNickName(mallMemberMap.containsKey(aiMemberId) ? mallMemberMap.get(aiMemberId).getName() : "");
|
apiMemberTeamPageVo.setMemberName(mallMemberMap.containsKey(aiMemberId) ? mallMemberMap.get(aiMemberId).getName() : "");
|
Integer practiceCnt = 0;
|
if (memberIdToCntMapTalk.containsKey(aiMemberId)){
|
practiceCnt = memberIdToCntMapTalk.get(aiMemberId);
|
}
|
if (memberIdToCntMapAnswer.containsKey(aiMemberId)){
|
practiceCnt = memberIdToCntMapAnswer.get(aiMemberId) + practiceCnt;
|
}
|
apiMemberTeamPageVo.setPracticeCnt(practiceCnt);
|
|
Integer studyTime = 0;
|
if (memberStudyTimeMap.containsKey(aiMemberId)){
|
studyTime = memberStudyTimeMap.get(aiMemberId);
|
}
|
apiMemberTeamPageVo.setStudyTime(DateUtil.secondToTime(studyTime));
|
objects.add(apiMemberTeamPageVo);
|
}
|
return objects;
|
}
|
|
|
|
@Override
|
public FebsResponse practice(ApiMemberTeamPracticeDto dto) {
|
List<ApiMemberTeamPracticeVo> vos = new ArrayList<>();
|
|
String companyId = LoginUserUtil.getLoginUser().getCompanyId();
|
String memberUuid = dto.getMemberUuid();
|
|
List<AiMemberTalk> aiMemberTalks = aiMemberTalkService.getListByCompanyIdAndMemberUuid(companyId, memberUuid);
|
Map<String, List<AiMemberTalk>> aiMemberTalkMap = new HashMap<>();
|
if (CollUtil.isNotEmpty(aiMemberTalks)){
|
//stream操作aiMemberTalks，返回一个根据productId分组的集合
|
aiMemberTalkMap = aiMemberTalks.stream().collect(Collectors.groupingBy(AiMemberTalk::getProductId));
|
}
|
|
List<AiMemberAnswer> aiMemberAnswers = aiMemberAnswerService.getListByCompanyIdAndMemberUuid(companyId, memberUuid);
|
Map<String, List<AiMemberAnswer>> aiMemberAnswerMap = new HashMap<>();
|
if (CollUtil.isNotEmpty(aiMemberAnswers)){
|
aiMemberAnswerMap = aiMemberAnswers.stream().collect(Collectors.groupingBy(AiMemberAnswer::getProductId));
|
}
|
|
// 获取aiMemberTalkMap和aiMemberAnswerMap的全部的key，并且去重
|
Set<String> productIds = new HashSet<>();
|
if (aiMemberTalkMap != null) {
|
productIds.addAll(aiMemberTalkMap.keySet());
|
}
|
if (aiMemberAnswerMap != null) {
|
productIds.addAll(aiMemberAnswerMap.keySet());
|
}
|
if (CollUtil.isEmpty(productIds)){
|
return new FebsResponse().success().data(vos);
|
}
|
|
List<AiProduct> aiProducts = aiProductService.getProductListByQuery(
|
Wrappers.lambdaQuery(AiProduct.class)
|
.select(AiProduct::getId,AiProduct::getName, AiProduct::getTarget)
|
.in(AiProduct::getId, productIds)
|
);
|
if (CollUtil.isEmpty(aiProducts)){
|
return new FebsResponse().success().data(vos);
|
}
|
|
for (AiProduct aiProduct : aiProducts){
|
|
String productId = aiProduct.getId();
|
ApiMemberTeamPracticeVo vo = new ApiMemberTeamPracticeVo();
|
vo.setName(aiProduct.getName());
|
vo.setTarget(aiProduct.getTarget());
|
|
Integer answerCnt = 0;
|
if (aiMemberAnswerMap.containsKey(productId)){
|
answerCnt = aiMemberAnswerMap.get(productId).size();
|
}
|
vo.setAnswerCnt(answerCnt);
|
|
Integer talkCnt = 0;
|
if (aiMemberTalkMap.containsKey(productId)){
|
talkCnt = aiMemberTalkMap.get(productId).size();
|
}
|
vo.setTalkCnt(talkCnt);
|
|
vos.add( vo);
|
}
|
|
|
return new FebsResponse().success().data(vos);
|
}
|
|
@Override
|
public FebsResponse study(ApiMemberTeamStudyDto dto) {
|
|
List<ApiMemberTeamStudyVo> vos = new ArrayList<>();
|
|
String companyId = LoginUserUtil.getLoginUser().getCompanyId();
|
String memberUuid = dto.getMemberUuid();
|
|
List<AiMemberPoint> aiMemberPoints = aiMemberPointService.getListByCompanyIdAndMemberUuid(companyId, memberUuid);
|
if (CollUtil.isEmpty(aiMemberPoints)){
|
return new FebsResponse().success().data(vos);
|
}
|
Map<String, AiMemberPoint> aiMemberPointMap = new HashMap<>();
|
if (CollUtil.isNotEmpty(aiMemberPoints)){
|
aiMemberPoints.forEach(aiMemberPoint -> aiMemberPointMap.put(aiMemberPoint.getProductPointId(),aiMemberPoint));
|
}
|
|
//stream流操作aiMemberPoints，返回一个productPointId的set集合
|
Set<String> productPointIdSet = aiMemberPoints.stream().map(AiMemberPoint::getProductPointId).collect(Collectors.toSet());
|
|
List<AiProductPoint> aiProductPoints = aiProductPointService.getBaseMapper().selectList(
|
Wrappers.lambdaQuery(AiProductPoint.class)
|
.select(AiProductPoint::getId,AiProductPoint::getTitle)
|
.in(AiProductPoint::getId, productPointIdSet)
|
);
|
if (CollUtil.isEmpty(aiProductPoints)){
|
return new FebsResponse().success().data(vos);
|
}
|
|
for (AiProductPoint aiProductPoint : aiProductPoints){
|
ApiMemberTeamStudyVo vo = new ApiMemberTeamStudyVo();
|
vo.setTitle(aiProductPoint.getTitle());
|
Integer totalTime = 0;
|
if (aiMemberPointMap.containsKey(aiProductPoint.getId())){
|
totalTime = aiMemberPointMap.get(aiProductPoint.getId()).getTotalTime();
|
}
|
vo.setTotalTime(DateUtil.secondToTime(totalTime));
|
|
vos.add(vo);
|
}
|
|
return new FebsResponse().success().data(vos);
|
}
|
}
|
