From a3ee255827cd44b92c697e5f3e313359e4407547 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Thu, 18 Jun 2020 18:16:02 +0800
Subject: [PATCH] add buy list

---
 src/main/java/com/xcong/excoin/modules/agent/controller/MemberController.java |    4 ++++
 1 files changed, 4 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..978b2d1 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,6 +31,8 @@
 
     @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);
     }

--
Gitblit v1.9.1