From 9b3f0e0c496871e9bc633bafe2512cace82dd5e2 Mon Sep 17 00:00:00 2001 From: Administrator <15274802129@163.com> Date: Mon, 22 Sep 2025 16:36:55 +0800 Subject: [PATCH] feat(ai):为成员团队练习和学习接口添加分页功能 --- src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberServiceImpl.java | 25 +++++++++++++++++++++++-- 1 files changed, 23 insertions(+), 2 deletions(-) diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberServiceImpl.java index ac90ba3..f7c232f 100644 --- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberServiceImpl.java +++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberServiceImpl.java @@ -1,6 +1,7 @@ package cc.mrbird.febs.ai.service.impl; import cc.mrbird.febs.ai.entity.*; +import cc.mrbird.febs.ai.enumerates.AiTypeEnum; import cc.mrbird.febs.ai.mapper.AiMemberMapper; import cc.mrbird.febs.ai.req.member.ApiMemberTeamPageDto; import cc.mrbird.febs.ai.req.member.ApiMemberTeamPracticeDto; @@ -158,7 +159,17 @@ String companyId = LoginUserUtil.getLoginUser().getCompanyId(); String memberUuid = dto.getMemberUuid(); - List<AiMemberTalk> aiMemberTalks = aiMemberTalkService.getListByCompanyIdAndMemberUuid(companyId, memberUuid); + // 创建分页对象,传入当前页和每页大小 + Page<AiMemberTalk> page = new Page<>(dto.getPageNow(), dto.getPageSize()); + LambdaQueryWrapper<AiMemberTalk> queryWrapper = Wrappers.lambdaQuery(AiMemberTalk.class); + queryWrapper.eq(AiMemberTalk::getCompanyId, companyId); + queryWrapper.eq(AiMemberTalk::getMemberId, memberUuid); + queryWrapper.eq(AiMemberTalk::getState, AiTypeEnum.AI_MEMBER_TALK_STATE_DONE.getCode()); + Page<AiMemberTalk> pageListByQuery = aiMemberTalkService.getBaseMapper().selectPage(page, queryWrapper); + List<AiMemberTalk> aiMemberTalks = pageListByQuery.getRecords(); + +// List<AiMemberTalk> aiMemberTalks = aiMemberTalkService.getListByCompanyIdAndMemberUuid(companyId, memberUuid); + Map<String, List<AiMemberTalk>> aiMemberTalkMap = new HashMap<>(); if (CollUtil.isNotEmpty(aiMemberTalks)){ //stream操作aiMemberTalks,返回一个根据productId分组的集合 @@ -226,7 +237,17 @@ String companyId = LoginUserUtil.getLoginUser().getCompanyId(); String memberUuid = dto.getMemberUuid(); - List<AiMemberPoint> aiMemberPoints = aiMemberPointService.getListByCompanyIdAndMemberUuid(companyId, memberUuid); + + + // 创建分页对象,传入当前页和每页大小 + Page<AiMemberPoint> page = new Page<>(dto.getPageNow(), dto.getPageSize()); + LambdaQueryWrapper<AiMemberPoint> queryWrapper = Wrappers.lambdaQuery(AiMemberPoint.class); + queryWrapper.eq(AiMemberPoint::getCompanyId, companyId); + queryWrapper.eq(AiMemberPoint::getMemberId, memberUuid); + Page<AiMemberPoint> pageListByQuery = aiMemberPointService.getBaseMapper().selectPage(page, queryWrapper); + List<AiMemberPoint> aiMemberPoints = pageListByQuery.getRecords(); + +// List<AiMemberPoint> aiMemberPoints = aiMemberPointService.getListByCompanyIdAndMemberUuid(companyId, memberUuid); if (CollUtil.isEmpty(aiMemberPoints)){ return new FebsResponse().success().data(vos); } -- Gitblit v1.9.1