diff --git a/src/main/java/com/sqx/modules/invite/dao/InviteDao.java b/src/main/java/com/sqx/modules/invite/dao/InviteDao.java index 41d203da..b94cf694 100644 --- a/src/main/java/com/sqx/modules/invite/dao/InviteDao.java +++ b/src/main/java/com/sqx/modules/invite/dao/InviteDao.java @@ -28,7 +28,8 @@ public interface InviteDao extends BaseMapper { Double selectInviteSum(@Param("state") Integer state, @Param("userId") Long userId); - IPage> selectInviteUser(Page> page, @Param("userId") Long userId,@Param("state") Integer state,@Param("userType") Integer userType); +// IPage> selectInviteUser(Page> page, @Param("userId") Long userId,@Param("state") Integer state,@Param("userType") Integer userType); + IPage> selectInviteUser(Page> page,@Param("code") String code); Invite selectInviteByUser(@Param("userId")Long userId,@Param("inviteeUserId") Long inviteeUserId,@Param("userType") Integer userType); 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 550e3cda..aab59ab9 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 @@ -72,7 +72,8 @@ public class InviteServiceImpl extends ServiceImpl implements @Override public PageUtils selectInviteUser(int page, int limit, Long userId, Integer state, Integer userType) { Page> pages = new Page<>(page, limit); - return new PageUtils(inviteDao.selectInviteUser(pages, userId, state, userType)); + UserEntity userEntity = userService.queryByUserId(userId); + return new PageUtils(inviteDao.selectInviteUser(pages, userEntity.getInvitationCode())); } @Override diff --git a/src/main/resources/mapper/invite/InviteDao.xml b/src/main/resources/mapper/invite/InviteDao.xml index e128f921..6b24f1b9 100644 --- a/src/main/resources/mapper/invite/InviteDao.xml +++ b/src/main/resources/mapper/invite/InviteDao.xml @@ -13,20 +13,37 @@ + + + + + + + + + + + + + + + +