From ca27c0a260a8cd728d278af16fb234ae089dc1c6 Mon Sep 17 00:00:00 2001 From: wzy <wzy19931122ai@163.com> Date: Fri, 17 Jul 2020 20:24:34 +0800 Subject: [PATCH] modify --- src/main/java/com/xcong/excoin/modules/agent/controller/MemberController.java | 20 ++++++++++++++++++++ 1 files changed, 20 insertions(+), 0 deletions(-) diff --git a/src/main/java/com/xcong/excoin/modules/agent/controller/MemberController.java b/src/main/java/com/xcong/excoin/modules/agent/controller/MemberController.java index 70a3d5f..8cdabf9 100644 --- a/src/main/java/com/xcong/excoin/modules/agent/controller/MemberController.java +++ b/src/main/java/com/xcong/excoin/modules/agent/controller/MemberController.java @@ -1,10 +1,12 @@ package com.xcong.excoin.modules.agent.controller; import com.xcong.excoin.common.controller.BaseController; +import com.xcong.excoin.common.entity.FebsConstant; import com.xcong.excoin.common.entity.FebsResponse; import com.xcong.excoin.common.entity.QueryRequest; import com.xcong.excoin.modules.agent.entity.MemberEntity; import com.xcong.excoin.modules.agent.service.IMemberService; +import com.xcong.excoin.system.entity.User; import lombok.RequiredArgsConstructor; import lombok.extern.slf4j.Slf4j; import org.springframework.validation.annotation.Validated; @@ -29,7 +31,25 @@ @GetMapping("getList") public FebsResponse getList(MemberEntity member, QueryRequest request) { + User user = getCurrentUser(); + member.setInviteId(FebsConstant.USER_TYPE_ADMIN.equals(user.getType()) ? FebsConstant.DEFAULT_REFERER_ID : user.getInviteId()); Map<String, Object> data = getDataTable(memberService.findMemberListInPage(member, request)); return new FebsResponse().success().data(data); } + + @GetMapping("getFeeAmountList") + public FebsResponse getFeeAmountList(MemberEntity memberEntity, QueryRequest request) { + User user = getCurrentUser(); + memberEntity.setRefererIds(FebsConstant.USER_TYPE_ADMIN.equals(user.getType()) ? FebsConstant.DEFAULT_REFERER_ID : user.getInviteId()); + Map<String, Object> data = getDataTable(memberService.findFeeAmountList(memberEntity, request)); + return new FebsResponse().success().data(data); + } + + @GetMapping("getMemberAssets") + public FebsResponse getMemberAssets(MemberEntity memberEntity, QueryRequest request) { + User user = getCurrentUser(); + memberEntity.setRefererIds(FebsConstant.USER_TYPE_ADMIN.equals(user.getType()) ? FebsConstant.DEFAULT_REFERER_ID : user.getInviteId()); + Map<String, Object> data = getDataTable(memberService.findMemberAssetList(memberEntity, request)); + return new FebsResponse().success().data(data); + } } -- Gitblit v1.9.1