EmploymentSite/src/main/java/com/hz/employmentsite/controller/jobUserManager/SigninMgtController.java

package com.hz.employmentsite.controller.jobUserManager;

import com.github.pagehelper.PageInfo;
import com.hz.employmentsite.filter.exception.BaseErrorEnum;
import com.hz.employmentsite.filter.exception.BaseException;
import com.hz.employmentsite.filter.exception.BaseResponse;
import com.hz.employmentsite.filter.exception.RespGenerstor;
import com.hz.employmentsite.model.PcRecommendMgt;
import com.hz.employmentsite.services.service.AccountService;
import com.hz.employmentsite.services.service.jobUserManager.RecommendMgtService;
import com.hz.employmentsite.services.service.jobUserManager.SigninMgtService;
import com.hz.employmentsite.util.DateUtils;
import com.hz.employmentsite.util.ExcelHelper;
import com.hz.employmentsite.vo.jobUserManager.*;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Date;
import java.util.List;

@RestController
@RequestMapping(value = "/api/jobusermgr/signinMgt")
public class SigninMgtController {
    @Autowired
    private SigninMgtService signinMgtService;
    @Autowired
    private AccountService accountService;
    @Autowired
    private DateUtils dateUtils;

    @ResponseBody
    @GetMapping("/getList")
    public BaseResponse getList(@RequestParam("pageIndex") int pageIndex, @RequestParam("pageSize") int pageSize,
                                @RequestParam(required = false) List<String> signinIDList,
                                @RequestParam(required = false) String name, @RequestParam(required = false) String siteID,
                                @RequestParam(required = false) String regionCode, @RequestParam(required = false) String signinType,
                                @RequestParam(required = false) Date startDate, @RequestParam(required = false) Date endDate) {

        PageInfo<SigninMgtVo> result = signinMgtService.getList(pageIndex, pageSize,signinIDList, name, siteID ,regionCode, signinType, startDate, endDate);
        return RespGenerstor.success(result);
    }

    @ResponseBody
    @GetMapping("/getListById")
    public BaseResponse getListById(@RequestParam(required = false) String id) {
        return RespGenerstor.success(signinMgtService.getDataById(id));
    }

    @GetMapping("/export")
    public BaseResponse export(HttpServletResponse response, @RequestParam(required = false) Boolean isExport,
                               @RequestParam("pageIndex") int pageIndex, @RequestParam("pageSize") int pageSize,
                               @RequestParam(required = false) List<String> signinIDList,
                               @RequestParam(required = false) String name, @RequestParam(required = false) String siteID,
                               @RequestParam(required = false) String regionCode, @RequestParam(required = false) String signinType,
                               @RequestParam(required = false) Date startDate, @RequestParam(required = false) Date endDate) throws Exception {
        PageInfo<SigninMgtVo> result = signinMgtService.getList(pageIndex, pageSize, signinIDList, name, siteID, regionCode, signinType, startDate, endDate);
        if (isExport == null || !isExport) {
            return RespGenerstor.success(result);
        } else {
            ExcelHelper excelHelper = new ExcelHelper();
            ExcelHelper.ExcelData data = excelHelper.new ExcelData();
            data.setTitles(Arrays.asList(new String[]{"序号", "外出人员", "外出时间", "所属县区", "所属驿站", "外出类型", "走访企业", "走访人员", "走访内容"}));
            int i = 0;
            List<List<Object>> rowDatas = new ArrayList();
            for (SigninMgtVo item : result.getList()) {
                List<Object> row = new ArrayList();
                ++i;
                row.add(i);
                row.add(item.getSigninerName());
                row.add(dateUtils.dateToStrFormat(item.getSigninTime(),"yyyy-MM-dd"));
                row.add(item.getRegionName());
                row.add(item.getSiteName());
                row.add(item.getSigninTypeName());
                row.add(item.getCompanyName());
                row.add(item.getJobUserName());
                row.add(item.getContent());
                rowDatas.add(row);
            }
            data.setRows(rowDatas);
            excelHelper.exportExcel(response, data);
            return null;
        }

    }

    @ResponseBody
    @PostMapping("/delete")
    public BaseResponse<Integer> deleteSigninMgt(HttpServletRequest request, @RequestBody List<String> ids) {
        Integer result = signinMgtService.delete(request, ids);
        if (result <= 0)
            throw new BaseException(BaseErrorEnum.DELETE_NOT_DATA_ERROR);
        return RespGenerstor.success(1);
    }

    @ResponseBody
    @PostMapping("/save")
    public BaseResponse<Integer> save(@RequestBody SigninMgtVo data) {
        var result = signinMgtService.save(data, accountService.getLoginUserID());
        return RespGenerstor.success(result);
    }

}