20210423

xiaoyong931011

2021-04-23 d4488a987cccd3ddd51a202a9f8aa71d3cdcbced

 src/main/java/com/xcong/excoin/modules/agent/controller/AgentController.java

@@ -1,7 +1,11 @@
package com.xcong.excoin.modules.agent.controller;

import cn.hutool.core.date.DateTime;
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.toolkit.StringPool;
import com.xcong.excoin.common.annotation.ControllerEndpoint;
import com.xcong.excoin.common.annotation.SubmitRepeat;
import com.xcong.excoin.common.controller.BaseController;
import com.xcong.excoin.common.entity.FebsResponse;
import com.xcong.excoin.common.entity.QueryRequest;
@@ -17,6 +21,7 @@
import javax.validation.Valid;
import javax.validation.constraints.NotBlank;
import javax.validation.constraints.NotNull;
import java.util.Date;
import java.util.Map;

/**
@@ -37,18 +42,29 @@
    public FebsResponse getList(AgentUser agentUser, QueryRequest queryRequest) {
        User user = getCurrentUser();
        agentUser.setRefererId(user.getInviteId());
        String startTime = agentUser.getStartTime();
        if(StrUtil.isNotEmpty(startTime)){
            Date date = DateUtil.parse(startTime);
            agentUser.setStartDate(date);
        }
        String endTime = agentUser.getEndTime();
        if(StrUtil.isNotEmpty(endTime)){
            Date date = DateUtil.parse(endTime);
            agentUser.setEndDate(date);
        }

        Map<String, Object> map = getDataTable(agentService.findAgentList(agentUser, queryRequest));
        return new FebsResponse().success().data(map);
    }

//    @SubmitRepeat
    @PostMapping("add")
    @RequiresPermissions("agent:add")
    @ControllerEndpoint(operation = "新增代理商", exceptionMessage = "新增代理商失败")
    public FebsResponse add(@Valid AgentUser agentUser) {
        User user = getCurrentUser();
        agentService.addAgent(agentUser, user);

        log.info("{}添加代理商{}", user.getInviteId(), agentUser.getInviteId());
        agentService.addAgent(agentUser, user);
        return new FebsResponse().success();
    }