EmploymentSite/src/main/java/com/hz/employmentsite/controller/taskAndLog/DoWorkController.java

package com.hz.employmentsite.controller.taskAndLog;

import com.github.pagehelper.PageInfo;
import com.hz.employmentsite.filter.exception.BaseErrorEnum;
import com.hz.employmentsite.filter.exception.BaseResponse;
import com.hz.employmentsite.filter.exception.RespGenerstor;
import com.hz.employmentsite.mapper.PcSiteUserMapper;
import com.hz.employmentsite.model.PcSiteUserExample;
import com.hz.employmentsite.services.service.AccountService;
import com.hz.employmentsite.services.service.taskAndLog.DoWorkService;
import com.hz.employmentsite.util.ExcelHelper;
import com.hz.employmentsite.vo.taskAndLog.DoWorkVo;
import com.hz.employmentsite.vo.taskAndLog.RecordsCountVo;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import javax.servlet.http.HttpServletResponse;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.UUID;

@RestController
@RequestMapping(value = "/api/taskAndLog/doWork")
public class DoWorkController {
    @Autowired
    private AccountService accountService;

    @Autowired
    private DoWorkService doWorkService;

    @Autowired
    private PcSiteUserMapper pcSiteUserMapper;

    @Autowired
    private ExcelHelper excelHelper;

    @ResponseBody
    @GetMapping("/getList")
    public BaseResponse getList(@RequestParam("pageIndex") int pageIndex, @RequestParam("pageSize") int pageSize, @RequestParam(required = false) List<String> doWorkIDList,
                                @RequestParam(required = false) String userName,@RequestParam(required = false) String workStartDate,  @RequestParam(required = false) String workEndDate,
                                @RequestParam(required = false) Integer doTypeID, @RequestParam(required = false) String regionCode,@RequestParam(required = false) String streetCode){
        PageInfo<DoWorkVo> result = doWorkService.getList(pageIndex, pageSize,doWorkIDList, userName, workStartDate,workEndDate,doTypeID,regionCode,streetCode);
        return RespGenerstor.success(result);
    }


    @GetMapping("get")
    public BaseResponse<DoWorkVo> get(@RequestParam(required = false) String doWorkID) {
        var data = doWorkService.get(doWorkID);
        if (data == null) {
            data = new DoWorkVo();
            data.doWorkID = UUID.randomUUID().toString();
            PcSiteUserExample siteUserExp = new PcSiteUserExample();
            PcSiteUserExample.Criteria siteUserCro = siteUserExp.createCriteria();
            siteUserCro.andUserIDEqualTo(accountService.getLoginUserID());
            var curSiteUserInfo = pcSiteUserMapper.selectByExample(siteUserExp).stream().findFirst().orElse(null);
            if(curSiteUserInfo != null){
                data.siteUserID = curSiteUserInfo.getSiteUserID();
                data.userId = curSiteUserInfo.getUserID();
            }
        }
        return RespGenerstor.success(data);
    }

    @PostMapping("save")
    public BaseResponse<Integer> save(@RequestBody DoWorkVo data) {
        return RespGenerstor.success(doWorkService.save(data,accountService.getLoginUserID()));
    }

    @GetMapping("export")
    public BaseResponse export(HttpServletResponse response, @RequestParam("pageIndex") int pageIndex, @RequestParam("pageSize") int pageSize,
                               @RequestParam(required = false) List<String> doWorkIDList,
                               @RequestParam(required = false) String userName,@RequestParam(required = false) String workStartDate,  @RequestParam(required = false) String workEndDate,
                               @RequestParam(required = false) Integer doTypeID, @RequestParam(required = false) String regionCode,@RequestParam(required = false) String streetCode) throws Exception{
        PageInfo<DoWorkVo> result = doWorkService.getList(pageIndex, pageSize,doWorkIDList, userName, workStartDate,workEndDate,doTypeID,regionCode,streetCode);
        ExcelHelper.ExcelData data = excelHelper.new ExcelData();
        data.setTitles(Arrays.asList(new String[]{"序号","姓名"," ", "日志类型", "工作情况","所属县区","所属街道"}));
        int i = 0;
        List<List<Object>> rowsData = new ArrayList();
        for (DoWorkVo item : result.getList()) {
            List<Object> row = new ArrayList();
            ++i;
            row.add(i);
            row.add(item.userName);
            row.add(item.workTime);
            row.add(item.logTypeName);
            row.add(item.workContent);
            row.add(item.regionName);
            row.add(item.streetName);
            rowsData.add(row);
        }
        data.setRows(rowsData);
        excelHelper.exportExcel(response, data);
        return null;
    }

    @ResponseBody
    @PostMapping("/delete")
    public BaseResponse<Integer> delete(@RequestBody List<String> idList){
        return RespGenerstor.success(doWorkService.delete(idList));
    }

    /**
     * 获取驿站人员指定工作日期所记录的业务数据数量
     * @param userId 用户ID
     * @param workTime 工作日期时间
     * @return 业务数据记录
     */
    @GetMapping("/getRecordsCount")
    public BaseResponse<RecordsCountVo> getRecordsCount(
            @RequestParam String userId,
            @RequestParam String workTime
    ) {
        RecordsCountVo recordsCount = doWorkService.getRecordsCount(userId, workTime);
        return RespGenerstor.success(recordsCount);
    }
}