用户新增伙伴

ruoyi-api/src/main/java/com/ruoyi/api/controller/user/ApiUserController.java

@@ -6,7 +6,9 @@ import com.ruoyi.agent.service.IAgentService;
 import com.ruoyi.api.controller.common.AbstractApiController;
 import com.ruoyi.business.service.IBusinessService;
 import com.ruoyi.common.annotation.RepeatSubmit;
+import com.ruoyi.common.core.domain.PageQuery;
 import com.ruoyi.common.core.domain.R;
+import com.ruoyi.common.core.page.TableDataInfo;
 import com.ruoyi.common.core.validate.AddGroup;
 import com.ruoyi.common.core.validate.EditGroup;
 import com.ruoyi.common.enums.FilePathSplicingType;
@@ -53,6 +55,17 @@ public class ApiUserController extends AbstractApiController {
     private final IAgentService agentService;
     private final IBusinessService businessService;
 
+    /**
+     * 查直级下级分页
+     */
+    @ApiOperation("查询直级下级分页")
+    @FilePathSplicing(type = FilePathSplicingType.RESPONSE)
+    @GetMapping("/partnerPage")
+    public TableDataInfo<UserVo> partnerPage(@Validated UserBo bo, PageQuery pageQuery) {
+        bo.setParentUserId(getUserId(true));
+        return iUserService.partnerPage(bo, pageQuery);
+    }
+
     /**
      * 查用户余额
      */

ruoyi-user/src/main/java/com/ruoyi/user/domain/vo/UserVo.java

@@ -211,4 +211,10 @@ public class UserVo {
      */
     @ApiModelProperty("已提现佣金")
     private BigDecimal commissionTakeOut;
+
+    /**
+     * 伙伴数
+     */
+    @ApiModelProperty("伙伴数")
+    private Integer partnerNum;
 }

ruoyi-user/src/main/java/com/ruoyi/user/mapper/UserMapper.java

@@ -37,4 +37,6 @@ public interface UserMapper extends BaseMapperPlus<UserMapper, User, UserVo> {
     Page<UserVo> selectBusinessPage(@Param("page") Page<Object> build, @Param("bo") UserBo bo);
 
     List<UserVo> getAllParentUser(@Param("ids")List<Long> parentUserIdList);
+
+    Integer selectPartnerNum(@Param("userId") Long userId);
 }

ruoyi-user/src/main/java/com/ruoyi/user/service/IUserService.java

@@ -203,4 +203,6 @@ public interface IUserService {
     List<UserVo> getAllParentUser(Long userId);
 
     User updateCommissionAble(Long userId, BigDecimal ableGains, boolean isAdd);
+
+    TableDataInfo<UserVo> partnerPage(UserBo bo, PageQuery pageQuery);
 }

ruoyi-user/src/main/java/com/ruoyi/user/service/impl/UserServiceImpl.java

@@ -319,6 +319,7 @@ public class UserServiceImpl implements IUserService {
         user.setMobile(wxUserDto.getMobile());
         user.setCreateDay(new Date());
         user.setLastLoginTime(new Date());
+        user.setInvitationCode(wxUserDto.getInvitationCode());
         //保存用户信息
         UserBo userBo = BeanUtil.toBean(user, UserBo.class);
         insertByBo(userBo);
@@ -361,6 +362,8 @@ public class UserServiceImpl implements IUserService {
                 throw new ServiceException("用户不存在");
             }
         }
+        //查伙伴数
+        obj.setPartnerNum(this.baseMapper.selectPartnerNum(userId));
         return obj;
     }
 
@@ -875,6 +878,13 @@ public class UserServiceImpl implements IUserService {
         return user;
     }
 
+    @Override
+    public TableDataInfo<UserVo> partnerPage(UserBo bo, PageQuery pageQuery) {
+        LambdaQueryWrapper<User> lqw = buildQueryWrapper(bo);
+        Page<UserVo> result = baseMapper.selectVoPage(pageQuery.build(), lqw);
+        return TableDataInfo.build(result);
+    }
+
     private String generateUniqueMemberCode(Long userId) {
         //生成唯一码 用户不会重复 6-8位
         String userIdPart = String.valueOf(userId);

ruoyi-user/src/main/resources/mapper/user/UserMapper.xml

@@ -191,4 +191,8 @@ PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
             #{item}
         </foreach>
     </select>
+
+    <select id="selectPartnerNum" resultType="java.lang.Integer">
+        select count(id) from tb_user where parent_user_id = #{userId}
+    </select>
 </mapper>

ruoyi-weixin/ruoyi-weixin-miniapp/src/main/java/com/ruoyi/weixin/domain/WxUserDto.java

@@ -77,4 +77,9 @@ public class WxUserDto implements Serializable {
      * 4-app
      */
     private Integer clientType;
+
+    /**
+     * 邀请码
+     */
+    private String invitationCode;
 }