ContractController.java 14.7 KB
Newer Older
manxiaoqiang committed
1 2
package common.controller;

kangxiaoshan committed
3 4 5 6
import common.model.Contract;
import common.model.ContractChange;
import common.model.ContractMoney;
import common.model.User;
manxiaoqiang committed
7
import common.service.ContractService;
manxiaoqiang committed
8
import common.task.ContractSendEmailThread;
kangxiaoshan committed
9
import dic.ContractStatusEnum;
manxiaoqiang committed
10
import dic.OperateObjectTypeEnum;
manxiaoqiang committed
11
import org.apache.commons.io.IOUtils;
2  
manxiaoqiang committed
12 13
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
manxiaoqiang committed
14
import org.springframework.beans.factory.annotation.Autowired;
manxiaoqiang committed
15 16
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
manxiaoqiang committed
17
import org.springframework.http.MediaType;
manxiaoqiang committed
18
import org.springframework.http.ResponseEntity;
manxiaoqiang committed
19 20 21
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.*;
import security.annotation.CurrentAccount;
manxiaoqiang committed
22
import util.NewUserLogThread;
manxiaoqiang committed
23
import util.ResultModel;
manxiaoqiang committed
24
import util.ValidateUtil;
manxiaoqiang committed
25

manxiaoqiang committed
26
import javax.servlet.http.HttpServletRequest;
manxiaoqiang committed
27 28 29 30
import javax.servlet.http.HttpServletResponse;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
manxiaoqiang committed
31
import java.util.HashMap;
manxiaoqiang committed
32
import java.util.List;
manxiaoqiang committed
33
import java.util.Map;
manxiaoqiang committed
34 35 36 37 38 39 40 41

/**
 * Created by mxq on 17/12/21.
 * 优先测试账号查询
 */
@Controller
@RequestMapping("{platform}/contract")
public class ContractController {
2  
manxiaoqiang committed
42
    private static final Logger logger = LoggerFactory.getLogger(ContractController.class);
kangxiaoshan committed
43 44 45 46 47 48 49 50 51 52 53 54 55

    private static final Map<String, String> CONTRACT_STATUS = new HashMap<String, String>() {{
        put(ContractStatusEnum.WAIT.getKey(), ContractStatusEnum.WAIT.getValue());
        put(ContractStatusEnum.EXECUTING.getKey(), ContractStatusEnum.EXECUTING.getValue());
        put(ContractStatusEnum.END.getKey(), ContractStatusEnum.END.getValue());
        put(ContractStatusEnum.CANCEL.getKey(), ContractStatusEnum.CANCEL.getValue());
        put(ContractStatusEnum.MONEY_BACK_ALL.getKey(), ContractStatusEnum.MONEY_BACK_ALL.getValue());
        put(ContractStatusEnum.MONEY_BACK_PART.getKey(), ContractStatusEnum.MONEY_BACK_PART.getValue());
        put(ContractStatusEnum.MONEY_BACK_NONE.getKey(), ContractStatusEnum.MONEY_BACK_NONE.getValue());
        put(ContractStatusEnum.DELETE.getKey(), ContractStatusEnum.DELETE.getValue());

    }};

manxiaoqiang committed
56 57 58 59 60
    @Autowired
    private ContractService service;

    @RequestMapping(value = "find", method = RequestMethod.GET)
    @ResponseBody
manxiaoqiang committed
61
    public ResultModel findAll(@CurrentAccount User loginAccount, @PathVariable String platform,
kangxiaoshan committed
62
                               @RequestParam String startDate, @RequestParam String endDate,String contractId) {
kangxiaoshan committed
63

kangxiaoshan committed
64
        return ResultModel.OK(service.findAll(loginAccount,startDate, endDate, platform,contractId));
manxiaoqiang committed
65 66
    }

manxiaoqiang committed
67
    @RequestMapping(value = "export", method = RequestMethod.GET,produces = MediaType.APPLICATION_OCTET_STREAM_VALUE)
manxiaoqiang committed
68
    @ResponseBody
manxiaoqiang committed
69
    public ResponseEntity<byte[]> export(@CurrentAccount User loginAccount, @PathVariable String platform, @RequestParam String startDate
manxiaoqiang committed
70
                                , @RequestParam String endDate, HttpServletRequest request, HttpServletResponse response) {
kangxiaoshan committed
71
        List<Contract> contractList = service.findAll(loginAccount, startDate, endDate, platform,null);
kangxiaoshan committed
72

manxiaoqiang committed
73

manxiaoqiang committed
74 75
        byte[] content = new byte[0];
        StringBuilder sb = new StringBuilder();
kangxiaoshan committed
76
        sb.append("时间,合同编号,客户账号,签约方,销售,开始日期,结束日期,套餐类型,合同金额,收款金额,合同状态\r\n");
manxiaoqiang committed
77 78
        if(ValidateUtil.isValid(contractList)){
            for (Contract c : contractList){
manxiaoqiang committed
79
                sb.append(c.getDs()).append(",").append(c.getContractCode()).append(",").append(c.getEmail()).append(",").append(c.getCustomerBody()).append(",").append(c.getSaleName()).append(",")
kangxiaoshan committed
80 81
                        .append(c.getStartDate()).append(",").append(c.getEndDate()).append(",").append(c.getPriceLevelName()).append(",").append(c.getMoney()).append(",").append(c.getRebateMoney()).append(",")
                        .append(CONTRACT_STATUS.get(c.getStatus())).append("\r\n");
manxiaoqiang committed
82 83 84 85 86 87 88 89 90
            }
        }
        InputStream in =  new  ByteArrayInputStream(sb.toString().getBytes());
        try {
            content  = IOUtils.toByteArray(in);
        } catch (IOException e) {
            e.printStackTrace();
        }
        HttpHeaders headers = new HttpHeaders();
manxiaoqiang committed
91 92
        response.setHeader("filename", "contract.csv");
        response.setHeader("content-disposition", "attachment; filename=contract.csv");
manxiaoqiang committed
93 94
        response.setHeader("Content-Type","text/csv");
        response.setContentType("APPLICATION/OCTET-STREAM");
manxiaoqiang committed
95

manxiaoqiang committed
96
        NewUserLogThread userlog = new NewUserLogThread(loginAccount.getEmail(), loginAccount.getName(), OperateObjectTypeEnum.CUSTOMER.getKey(), platform, "导出合同", startDate, endDate, request,platform);
manxiaoqiang committed
97
        userlog.start();
manxiaoqiang committed
98 99 100 101 102 103
        //返回文件字符串
        return new ResponseEntity<byte[]>(content, headers, HttpStatus.CREATED);
    }

    @RequestMapping(value = "pay/export", method = RequestMethod.GET,produces = MediaType.APPLICATION_OCTET_STREAM_VALUE)
    @ResponseBody
manxiaoqiang committed
104
    public ResponseEntity<byte[]> exportPay(@CurrentAccount User loginAccount, @RequestParam String startDate, @RequestParam String endDate,
manxiaoqiang committed
105 106
                                             HttpServletRequest request, HttpServletResponse response, @PathVariable String platform) {
        List<ContractMoney> moneyList = service.findPayAll(startDate, endDate, platform);
manxiaoqiang committed
107 108 109
        Map<String, String> map = new HashMap<>();
        map.put("pay", "收款");
        map.put("invoice", "收票");
manxiaoqiang committed
110 111 112 113 114 115

        byte[] content = new byte[0];
        StringBuilder sb = new StringBuilder();
        sb.append("合同编号 ,客户账号 ,签约方 ,事项 ,金额 ,日期 ,修改人 \r\n");
        if(ValidateUtil.isValid(moneyList)){
            for (ContractMoney c : moneyList){
manxiaoqiang committed
116
                sb.append(c.getContractCode()).append(",").append(c.getEmail()).append(",").append(c.getCompany()).append(",").append(map.get(c.getType())).append(",")
manxiaoqiang committed
117
                        .append(c.getMoney()).append(",").append(c.getDs()).append(",").append(c.getCreateName()).append("\r\n");
manxiaoqiang committed
118 119
            }
        }
manxiaoqiang committed
120 121 122 123 124 125
        InputStream in =  new  ByteArrayInputStream(sb.toString().getBytes());
        try {
            content  = IOUtils.toByteArray(in);
        } catch (IOException e) {
            e.printStackTrace();
        }
manxiaoqiang committed
126
        HttpHeaders headers = new HttpHeaders();
manxiaoqiang committed
127 128 129 130 131
        response.setHeader("filename", "contract_pay.csv");
        response.setHeader("content-disposition", "attachment; filename=contract_pay.csv");
       // response.setHeader("Content-Type", "application/octet-stream;charset=UTF-8");
        response.setHeader("Content-Type","text/csv");
        response.setContentType("APPLICATION/OCTET-STREAM");
manxiaoqiang committed
132

manxiaoqiang committed
133
        NewUserLogThread userlog = new NewUserLogThread(loginAccount.getEmail(), loginAccount.getName(), OperateObjectTypeEnum.CUSTOMER.getKey(), platform, "导出收款开票", startDate, endDate, request, platform);
manxiaoqiang committed
134
        userlog.start();
manxiaoqiang committed
135 136 137 138
        //返回文件字符串
        return new ResponseEntity<byte[]>(content, headers, HttpStatus.CREATED);
    }

manxiaoqiang committed
139 140 141 142 143 144
    @RequestMapping(value = "findone", method = RequestMethod.GET)
    @ResponseBody
    public ResultModel findOne(@CurrentAccount User loginAccount, @RequestParam String code) {
        return ResultModel.OK(service.findOne(code));
    }

manxiaoqiang committed
145 146
    @RequestMapping(value = "checkAccount", method = RequestMethod.GET)
    @ResponseBody
manxiaoqiang committed
147 148
    public ResultModel checkAccount(@CurrentAccount User loginAccount, @RequestParam String email, @PathVariable String platform) {
        return ResultModel.OK(service.checkAccount(email, platform));
manxiaoqiang committed
149 150
    }

manxiaoqiang committed
151 152 153 154 155 156 157
    @RequestMapping(value = "checkTime", method = RequestMethod.GET)
    @ResponseBody
    public ResultModel checkTime(@RequestParam String type, @RequestParam String email,
                                 @RequestParam String platform, @RequestParam(required = false) String product) {
        return ResultModel.OK(service.checkTime(email, platform,type,product));
    }

manxiaoqiang committed
158 159 160 161 162 163 164 165
    @RequestMapping(value = "find/body", method = RequestMethod.GET)
    @ResponseBody
    public ResultModel findBody(@CurrentAccount User loginAccount) {
        return ResultModel.OK(service.findBody());
    }

    @RequestMapping(value = "find/code", method = RequestMethod.GET)
    @ResponseBody
kangxiaoshan committed
166 167 168
    public ResultModel findCode(@CurrentAccount User loginAccount, @RequestParam String code,@PathVariable String platform,String company) {

        return ResultModel.OK(service.getContractCode(code,platform,company));
manxiaoqiang committed
169 170 171 172
    }

    @RequestMapping(value = "find/pricelevel", method = RequestMethod.GET)
    @ResponseBody
manxiaoqiang committed
173 174
    public ResultModel findPricelevel(@CurrentAccount User loginAccount, @PathVariable String platform) {
        return ResultModel.OK(service.findSetmeal(platform));
manxiaoqiang committed
175 176
    }

manxiaoqiang committed
177 178 179 180 181 182
    @RequestMapping(value = "find/increment", method = RequestMethod.GET)
    @ResponseBody
    public ResultModel findIncrement(@CurrentAccount User loginAccount) {
        return ResultModel.OK(service.findIncrement());
    }

manxiaoqiang committed
183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198
    @RequestMapping(value = "find/rebat", method = RequestMethod.GET)
    @ResponseBody
    public ResultModel rebat(@CurrentAccount User loginAccount, @RequestParam String product,
                             @RequestParam Integer level, @RequestParam Long money) {
        return ResultModel.OK(service.getRebat(product, level, money));
    }

    @RequestMapping(value = "find/sale", method = RequestMethod.GET)
    @ResponseBody
    public ResultModel rebat(@CurrentAccount User loginAccount) {
        return ResultModel.OK(service.getSales());
    }


    @RequestMapping(value = "create", method = RequestMethod.POST)
    @ResponseBody
manxiaoqiang committed
199
    public ResultModel create(@CurrentAccount User loginAccount, @RequestBody Contract contract, HttpServletRequest request, @PathVariable String platform) {
manxiaoqiang committed
200
        Contract contract1 = service.create(loginAccount, contract);
manxiaoqiang committed
201
        NewUserLogThread userlog = new NewUserLogThread(loginAccount.getEmail(), loginAccount.getName(), OperateObjectTypeEnum.CUSTOMER.getKey(), contract1.getContractCode(), "新建合同","",contract1.toString(), request, platform);
manxiaoqiang committed
202
        userlog.start();
kangxiaoshan committed
203 204 205
//
//        ContractSendEmailThread email = new ContractSendEmailThread(contract);
//        email.start();
manxiaoqiang committed
206
        return ResultModel.OK(contract1);
manxiaoqiang committed
207
    }
manxiaoqiang committed
208 209 210

    @RequestMapping(value = "update", method = RequestMethod.PUT)
    @ResponseBody
manxiaoqiang committed
211
    public ResultModel update(@CurrentAccount User loginAccount, @RequestBody Contract contract, HttpServletRequest request, @PathVariable String platform) {
manxiaoqiang committed
212
        Contract contract1 = service.update(loginAccount, contract);
manxiaoqiang committed
213
        NewUserLogThread userlog = new NewUserLogThread(loginAccount.getEmail(), loginAccount.getName(), OperateObjectTypeEnum.CUSTOMER.getKey(), contract1.getContractCode(), "修改合同","",contract1.toString(), request, platform);
manxiaoqiang committed
214 215
        userlog.start();
        return ResultModel.OK(contract1);
manxiaoqiang committed
216 217 218 219
    }

    @RequestMapping(value = "pay", method = RequestMethod.POST)
    @ResponseBody
manxiaoqiang committed
220
    public ResultModel pay(@CurrentAccount User loginAccount, @RequestBody ContractMoney contract, HttpServletRequest request, @PathVariable String platform) {
manxiaoqiang committed
221
        ContractMoney contract1 = service.pay(loginAccount, contract);
manxiaoqiang committed
222
        NewUserLogThread userlog = new NewUserLogThread(loginAccount.getEmail(), loginAccount.getName(), OperateObjectTypeEnum.CUSTOMER.getKey(), contract1.getContractCode(), "收款开票","",contract1.toString(), request, platform);
manxiaoqiang committed
223 224
        userlog.start();
        return ResultModel.OK(contract1);
manxiaoqiang committed
225 226 227 228
    }

    @RequestMapping(value = "change", method = RequestMethod.POST)
    @ResponseBody
manxiaoqiang committed
229
    public ResultModel change(@CurrentAccount User loginAccount, @RequestBody ContractChange contract, HttpServletRequest request, @PathVariable String platform) {
manxiaoqiang committed
230
        Contract contract1 = service.change(loginAccount, contract);
manxiaoqiang committed
231
        NewUserLogThread userlog = new NewUserLogThread(loginAccount.getEmail(), loginAccount.getName(), OperateObjectTypeEnum.CUSTOMER.getKey(), contract1.getContractCode(), "修改套餐","",contract1.toString(), request, platform);
manxiaoqiang committed
232 233
        userlog.start();
        return ResultModel.OK(contract1);
manxiaoqiang committed
234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251
    }

    @RequestMapping(value = "find/pay", method = RequestMethod.GET)
    @ResponseBody
    public ResultModel findPay(@CurrentAccount User loginAccount, @RequestParam String startDate,
                               @RequestParam String endDate, @RequestParam String code) {
        return ResultModel.OK(service.findPay(startDate, endDate, code));
    }

    @RequestMapping(value = "find/change", method = RequestMethod.GET)
    @ResponseBody
    public ResultModel findChange(@CurrentAccount User loginAccount, @RequestParam String startDate,
                                  @RequestParam String endDate, @RequestParam String code) {
        return ResultModel.OK(service.findChange(startDate, endDate, code));
    }

    @RequestMapping(value = "update/pay", method = RequestMethod.PUT)
    @ResponseBody
manxiaoqiang committed
252
    public ResultModel updatePay(@CurrentAccount User loginAccount, @RequestBody ContractMoney contract, HttpServletRequest request, @PathVariable String platform) {
manxiaoqiang committed
253
        Contract contract1 = service.updatePay(loginAccount, contract);
manxiaoqiang committed
254
        NewUserLogThread userlog = new NewUserLogThread(loginAccount.getEmail(), loginAccount.getName(), OperateObjectTypeEnum.CUSTOMER.getKey(), contract1.getContractCode(), "修改收款开票","", service.updatePay(loginAccount, contract).toString(), request, platform);
manxiaoqiang committed
255 256
        userlog.start();
        return ResultModel.OK(contract1);
manxiaoqiang committed
257
    }
manxiaoqiang committed
258 259 260 261

    @RequestMapping(value = "find/payall", method = RequestMethod.GET)
    @ResponseBody
    public ResultModel findPayAll(@CurrentAccount User loginAccount, @RequestParam String startDate,
manxiaoqiang committed
262 263
                               @RequestParam String endDate, @PathVariable String platform) {
        return ResultModel.OK(service.findPayAll(startDate, endDate, platform));
manxiaoqiang committed
264
    }
manxiaoqiang committed
265

manxiaoqiang committed
266 267 268 269
    @RequestMapping(value = "build", method = RequestMethod.POST)
    @ResponseBody
    public ResultModel build(@RequestBody Contract contract, HttpServletRequest request) {

2  
manxiaoqiang committed
270
        logger.error("tkiio---------------" + contract);
manxiaoqiang committed
271 272
        return ResultModel.OK(service.build(contract));
    }
manxiaoqiang committed
273

kangxiaoshan committed
274
    @RequestMapping(value = "trade", method = RequestMethod.GET)
kangxiaoshan committed
275
    @ResponseBody
kangxiaoshan committed
276
    public ResultModel baseCreate( @PathVariable String platform) {
kangxiaoshan committed
277 278 279 280 281

        return ResultModel.OK(service.getTradeData(platform));
    }


kangxiaoshan committed
282
    @RequestMapping(value = "code/all", method = RequestMethod.GET)
kangxiaoshan committed
283
    @ResponseBody
kangxiaoshan committed
284
    public ResultModel contractCodeAll( @PathVariable String platform,String contractId) {
kangxiaoshan committed
285

kangxiaoshan committed
286
        return ResultModel.OK(service.contractCodeAll(platform,contractId));
kangxiaoshan committed
287 288 289
    }


kangxiaoshan committed
290 291 292 293 294 295 296 297
    @RequestMapping(value = "setstaus", method = RequestMethod.GET)
    @ResponseBody
    public ResultModel contractStatusUpdate( @PathVariable String platform,String contractId,String status) {

        return ResultModel.OK(service.contractStatusUpdate(platform,contractId,status));
    }


kangxiaoshan committed
298 299


manxiaoqiang committed
300
}