diff --git a/src/main/java/com/sqx/modules/invite/service/impl/InviteServiceImpl.java b/src/main/java/com/sqx/modules/invite/service/impl/InviteServiceImpl.java index b96e6ccf..b2bc0197 100644 --- a/src/main/java/com/sqx/modules/invite/service/impl/InviteServiceImpl.java +++ b/src/main/java/com/sqx/modules/invite/service/impl/InviteServiceImpl.java @@ -87,7 +87,7 @@ public class InviteServiceImpl extends ServiceImpl implements List userIds = list.stream().map(UserEntity::getUserId).collect(Collectors.toList()); LambdaQueryWrapper queryWrapper1 = new LambdaQueryWrapper<>(); - queryWrapper1.gt(InviteAchievement::getCount, 3); + queryWrapper1.ge(InviteAchievement::getCount, 3); queryWrapper1.eq(InviteAchievement::getUserId, userId); queryWrapper1.in(InviteAchievement::getTargetUserId, userIds); List achievementList = inviteAchievementService.list(queryWrapper1); diff --git a/src/test/java/com/sqx/Tets.java b/src/test/java/com/sqx/Tets.java index 168e8281..ba5ce494 100644 --- a/src/test/java/com/sqx/Tets.java +++ b/src/test/java/com/sqx/Tets.java @@ -61,17 +61,16 @@ public class Tets { @Test public void test2() { - inviteAchievementService.remove(new LambdaQueryWrapper<>()); +// inviteAchievementService.remove(new LambdaQueryWrapper<>()); List userinfoList = userService.list(); - for (UserEntity userEntity : userinfoList) { - if (StrUtil.isBlank(userEntity.getInviterCode())) continue; + userinfoList.parallelStream().forEach((userEntity -> { + if (StrUtil.isBlank(userEntity.getInviterCode())) return; UserEntity byUser = userService.queryByInvitationCode(userEntity.getInviterCode()); - if (byUser == null) continue; - + if (byUser == null) return; Integer count = payDetailsDao.selectCount(new LambdaQueryWrapper() .eq(PayDetails::getState, 1) @@ -99,13 +98,11 @@ public class Tets { inviteAchievement.setCreateTime(DateUtil.date()); inviteAchievement.setTargetUserId(userEntity.getUserId()); inviteAchievement.setUserId(byUser.getUserId()); - inviteAchievementService.update(new LambdaQueryWrapper() + inviteAchievementService.update(inviteAchievement, new LambdaQueryWrapper() .eq(InviteAchievement::getUserId, inviteAchievement.getUserId()) .eq(InviteAchievement::getId, inviteAchievement.getId())); } - - } - + })); } @Test