fxkh-txl-service/src/main/java/com/whdc/controller/ShCallsController.java

package com.whdc.controller;

import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
import com.whdc.exception.MyException;
import com.whdc.model.dto.ShAddressBookDto;
import com.whdc.model.dto.ShCallsDto;
import com.whdc.model.entity.ShAddressBook;
import com.whdc.model.entity.ShCallWord;
import com.whdc.model.entity.ShCalls;
import com.whdc.model.group.Insert;
import com.whdc.model.vo.ShCallsVo;
import com.whdc.service.ShAddressBookService;
import com.whdc.service.ShCallsService;
import com.whdc.utils.DateUtils;
import com.whdc.utils.ExcelCommon;
import com.whdc.utils.ResultJson;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.HttpServletResponse;
import java.time.format.DateTimeFormatter;
import java.util.Date;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;

/**
 * 描述: 抽查日志
 * author: xusan
 * date: 2024-07-29 17:27:25
 */
@Tag(name = "抽查日志")
@RestController
@RequestMapping(value="/shCalls")
public class ShCallsController {

    @Autowired
    private ShCallsService service;

    @Autowired
    private ShAddressBookService shAddressBookService;


    @Operation(summary = "新增")
    @PostMapping("/insert")
    public ResultJson<ShCallWord> insert(@Validated(Insert.class) @RequestBody ShCalls dto) {
        if (shAddressBookService.lambdaQuery().eq(ShAddressBook::getId, dto.getShAbId()).count() != 1){
            throw new MyException("当前通讯录不存在");
        }

        dto.setCallTime(new Date());
        if (StringUtils.isNotBlank(dto.getStatus()) && "0".equals(dto.getCallType())){
            boolean update = shAddressBookService.lambdaUpdate()
                    .set(ShAddressBook::getCallStatus, dto.getCallType())
                    .eq(ShAddressBook::getId, dto.getShAbId())
                    .update();
            if (!update){
                throw new MyException("更新通讯录状态失败");
            }
        }
        boolean result = service.save(dto);
        return ResultJson.ok(result ? dto : null);
    }

    @Operation(summary = "列表")
    @PostMapping("/list")
    public ResultJson<List<ShCalls>> list(@RequestBody ShCallsDto dto) {
        LambdaQueryChainWrapper<ShCalls> query = service.lambdaQuery();
        if (Objects.nonNull(dto.getDate())){

            query.between(ShCalls::getCallTime, dto.getDate() + " 00:00:00",dto.getDate() + " 23:59:59");
        }
        if (Objects.nonNull(dto.getShAbId())){
            query.eq(ShCalls::getShAbId, dto.getShAbId());
        }else{
            throw new RuntimeException("山洪通讯录编号不能为空");
        }
        return ResultJson.ok(query.list());
    }

    @Operation(summary = "抽查日志分页")
    @PostMapping("/page")
    public ResultJson<IPage<ShCallsVo>> page(@RequestBody ShCallsDto dto) {
        return ResultJson.ok(service.page(dto));
    }

    @Operation(summary = "综述")
    @PostMapping("/statistics")
    public ResultJson<String> statistics(@RequestBody ShCallsDto dto) {

        List<ShCallsVo> list = service.list(dto);

        String dateStr = dto.getDate().format(DateTimeFormatter.ofPattern(DateUtils.DATE_PATTERN));
        List<String> citys = list.stream().map(ShCallsVo::getCity).distinct().collect(Collectors.toList());
        List<String> countys = list.stream().map(ShCallsVo::getCounty).distinct().collect(Collectors.toList());

        return ResultJson.ok(String.format("%%抽查了%%个市（%%）%%个县（%%）县乡村山洪灾害防御责任人和包保责任人共计%%人。",
                dateStr,
                citys.size(),
                StringUtils.join(citys, ","),
                countys.size(),
                StringUtils.join(countys, ","),
                list.stream().map(ShCallsVo::getPhone).distinct().count()
        ));
    }



    @Operation(summary = "抽查日志导出(未完成)")
    @PostMapping("/export")
    public void export(@RequestBody ShAddressBookDto dto, HttpServletResponse response) {

        LambdaQueryChainWrapper<ShCalls> query = service.lambdaQuery();

        query.orderByDesc(ShCalls::getCallTime);

        List<ShCalls> list = service.list(query);

        ExcelCommon.exportExcel(list,
                null, "抽查日志", ShCalls.class, "抽查日志.xlsx",
                response);
    }


}