From 8dfe6a5054c6222dde9e23f3fec20a8ed760ed02 Mon Sep 17 00:00:00 2001 From: Helius <wangdoubleone@gmail.com> Date: Mon, 22 Mar 2021 18:30:03 +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