jihe-dc/zc-business/src/main/java/com/zc/business/controller/DcSdhsEventController.java

package com.zc.business.controller;

import com.ruoyi.common.config.RuoYiConfig;
import com.ruoyi.common.core.domain.AjaxResult;
import com.zc.business.domain.DcSdhsEvent;
import com.zc.business.domain.DcSdhsEventQuery;
import com.zc.business.enums.UniversalEnum;
import com.zc.business.domain.DcSdhsEventQuery;
import com.zc.business.enums.UniversalEnum;
import com.ruoyi.common.utils.SecurityUtils;
import com.zc.business.domain.DcSdhsEvent;
import com.zc.business.domain.DcSdhsEventQuery;
import com.zc.business.enums.UniversalEnum;
import com.zc.business.service.IDcSdhsEventService;
import com.zc.business.utils.PoiUtil;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.CellType;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.apache.poi.xwpf.usermodel.XWPFChart;
import org.apache.poi.xwpf.usermodel.XWPFDocument;
import org.apache.poi.xwpf.usermodel.XWPFTable;
import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTTblWidth;
import org.openxmlformats.schemas.wordprocessingml.x2006.main.STTblWidth;
import io.swagger.annotations.ApiParam;
import org.apache.poi.xwpf.usermodel.ParagraphAlignment;
import org.apache.poi.xwpf.usermodel.XWPFDocument;
import org.apache.poi.xwpf.usermodel.XWPFParagraph;
import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTBr;
import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTP;
import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTR;
import org.openxmlformats.schemas.wordprocessingml.x2006.main.STBrType;
import org.apache.catalina.security.SecurityUtil;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.apache.poi.xwpf.usermodel.*;
import org.openxmlformats.schemas.wordprocessingml.x2006.main.*;
import org.springframework.beans.factory.annotation.Autowired;
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 org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import sun.reflect.generics.tree.VoidDescriptor;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;
import java.math.BigInteger;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import static com.zc.business.utils.PoiUtil.mergeChart;
import static com.zc.business.utils.PoiUtil.setTableFonts;

import javax.servlet.http.HttpServletResponse;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.List;
import java.util.Map;

import static com.zc.business.utils.PoiUtil.insertTOC;

import javax.servlet.http.HttpServletResponse;
import java.io.InputStream;
import java.math.BigInteger;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.List;
import java.util.Map;

import static com.zc.business.utils.PoiUtil.*;
import static com.zc.business.utils.PoiUtil.createLineBreak;

/**
 * @Description 高速云事件分析Controller
 *
 * @author liuwenge
 * @date 2024/08/15 16:38
 */
@Api(tags = "高速云事件分析")
@RestController
@RequestMapping("/business/sdhsEvent")
public class DcSdhsEventController {


    @Autowired
    private IDcSdhsEventService dcSdhsEventService;
    @Autowired
    private WordController wordController;

    /*@ApiOperation("导入数据")
    @PostMapping("/importData")
    public AjaxResult importData(MultipartFile file) throws Exception
    {
        return dcSdhsEventService.importData(file);
    }*/
    @ApiOperation("导入数据")
    @PostMapping("/importData")
    public AjaxResult importData(@ApiParam(value="文件名称", name="fileName", required=true) @RequestParam("fileName") String fileName,
                                 @ApiParam(value="文件路径", name="fileUrl", required=true) @RequestParam("fileUrl") String fileUrl) throws Exception
    {
        return dcSdhsEventService.importData(fileName,fileUrl);
    }

    //事故多发路段分析
    @ApiOperation("查询事故多发路段分析")
    @PostMapping("/sectionAnalysis")
    public AjaxResult sectionAnalysis(@RequestBody  DcSdhsEventQuery dcSdhsEventQuery){
        if (dcSdhsEventQuery.getStartTime()==null||dcSdhsEventQuery.getEndTime()==null){
            return AjaxResult.error("参数异常");
        }
        return AjaxResult.success(dcSdhsEventService.accidentSectionAnalysis(dcSdhsEventQuery));
    }
    //事件类型分析
    @ApiOperation("查询事件类型分析")
    @PostMapping("/typeAnalysis")
    public AjaxResult accidentTypeAnalysis(@RequestBody  DcSdhsEventQuery dcSdhsEventQuery){
        if (dcSdhsEventQuery.getStartTime()==null||dcSdhsEventQuery.getEndTime()==null){
            return AjaxResult.error("参数异常");
        }
        return AjaxResult.success(dcSdhsEventService.accidentTypeAnalysis(dcSdhsEventQuery));
    }
    //事故拥堵时间
    @ApiOperation("查询事故拥堵时间分析")
    @PostMapping("/congestedTime")
    public AjaxResult accidentCongestedTime(@RequestBody  DcSdhsEventQuery dcSdhsEventQuery){
        if (dcSdhsEventQuery.getStartTime()==null||dcSdhsEventQuery.getEndTime()==null){
            return AjaxResult.error("参数异常");
        }
        return AjaxResult.success(dcSdhsEventService.accidentCongestedTime(dcSdhsEventQuery));
    }
    //事故拥堵地点
    @ApiOperation("查询事故拥堵地点分析")
    @PostMapping("/congestedLocation")
    public AjaxResult accidentCongestedLocation(@RequestBody  DcSdhsEventQuery dcSdhsEventQuery){
        if (dcSdhsEventQuery.getStartTime()==null||dcSdhsEventQuery.getEndTime()==null){
            return AjaxResult.error("参数异常");
        }
        return AjaxResult.success(dcSdhsEventService.accidentCongestedLocation(dcSdhsEventQuery));
    }

    //事故多发路段、类型分析导出
    @PostMapping("/sectionTypeAnalysisExport")
    public void  sectionTypeAnalysisExport(HttpServletResponse response,@RequestBody  DcSdhsEventQuery dcSdhsEventQuery) throws IOException {
        // 读取Word模板
        InputStream inputStream = this.getClass().getClassLoader().getResourceAsStream(UniversalEnum.EXPORT_TRAFFIC_REPORT.getValue());
        XWPFDocument newDoc = new XWPFDocument(inputStream);
        dcSdhsEventService.sectionAnalysisExport(newDoc,dcSdhsEventQuery);
        dcSdhsEventService.accidentTypeAnalysisExport(newDoc,dcSdhsEventQuery);
        dcSdhsEventService.accidentCongestedTimeExport(newDoc,dcSdhsEventQuery);
        dcSdhsEventService.accidentCongestedLocationExport(newDoc,dcSdhsEventQuery);
        // 不保存直接返回文件流
        newDoc.write(response.getOutputStream());
        // 关闭文档
        newDoc.close();
        System.out.println("生成通行情况快报成功！");
    }
    @PostMapping("/export")
    @ApiOperation("导出")
    public void trafficSituationReport(HttpServletResponse response,DcSdhsEventQuery dcSdhsEventQuery) throws Exception {
        try {
            SimpleDateFormat df = new SimpleDateFormat(UniversalEnum.TIME_FORMAT_YEARS_MONTH_DAY.getValue());
            String date = df.format(new Date());
            String outputPath = RuoYiConfig.getDownloadPath() +"/高速云事件分析报告" + date + ".docx";
            // 读取Word模板
            InputStream inputStream = this.getClass().getClassLoader().getResourceAsStream("wordTemplate/wordTemplate2.docx");
            XWPFDocument newDoc = new XWPFDocument(inputStream);

            // 创建并插入带超链接的目录
            insertTOC(newDoc);


            //换页
            XWPFParagraph breakPara = newDoc.createParagraph();
            breakPara.setAlignment(ParagraphAlignment.CENTER);
            CTP ctP = breakPara.getCTP();
            CTR ctr = ctP.addNewR();
            CTBr ctBr = ctr.addNewBr();
            ctBr.setType(STBrType.PAGE);

            List<String> exportKey = dcSdhsEventQuery.getExportKey();
            if (exportKey != null && exportKey.size() > 0){
                //交通事件
                if (exportKey.contains("1") || exportKey.contains("2") || exportKey.contains("3")
                        || exportKey.contains("4") || exportKey.contains("5") || exportKey.contains("6")){
                    PoiUtil.createHeading1(newDoc,"交通事件");
                    for (String trafficIncidentType : exportKey) {
                        if ("1".equals(trafficIncidentType)){
                            //事故多发路段
                            dcSdhsEventService.sectionAnalysisExport(newDoc,dcSdhsEventQuery);
                        } else if ("2".equals(trafficIncidentType)){
                            //事件类型分析
                            dcSdhsEventService.accidentTypeAnalysisExport(newDoc,dcSdhsEventQuery);
                        } else if ("3".equals(trafficIncidentType)){
                            //事故车型分析
                            //事故车型
                            dcSdhsEventService.exportAccidentModelAnalysis(newDoc, dcSdhsEventQuery);

                        } else if ("4".equals(trafficIncidentType)){
                            //事故时间分析
                            //事故时间
                            dcSdhsEventService.exportAccidentTimeAnalysis(newDoc, dcSdhsEventQuery);

                        } else if ("5".equals(trafficIncidentType)){
                            //事故地市分布
                            dcSdhsEventService.exportRegionAnalysis(newDoc,dcSdhsEventQuery);
                        } else if ("6".equals(trafficIncidentType)){
                            //交通管制情况统计
                            dcSdhsEventService.exportStationAnalysis(newDoc,dcSdhsEventQuery);
                        }

                    }
                }
                if (exportKey.contains("7")){
                    PoiUtil.createHeading1(newDoc,"天气情况统计");
                    wordController.weather(newDoc);
                }
                if (exportKey.contains("8") || exportKey.contains("9")){
                    PoiUtil.createHeading1(newDoc,"交通量统计");
                    for (String trafficIncidentType : exportKey) {
                        if ("8".equals(trafficIncidentType)){
                            //门架车流量
                            dcSdhsEventService.exporFrameDataAnalysis(newDoc,dcSdhsEventQuery);
                        } else if ("9".equals(trafficIncidentType)){
                            //收费站车流量
                            dcSdhsEventService.exporttollGateEntrance(newDoc,dcSdhsEventQuery);
                        }

                    }
                }
                if (exportKey.contains("10")){
                    PoiUtil.createHeading1(newDoc,"交通事故情况");
                    //交通事故类型统计
                    dcSdhsEventService.exportAccidentType(newDoc,dcSdhsEventQuery);

                }
                if (exportKey.contains("11") || exportKey.contains("12")){
                    PoiUtil.createHeading1(newDoc,"拥堵情况");
                    for (String trafficIncidentType : exportKey) {
                        if ("11".equals(trafficIncidentType)){
                            //拥堵地点统计
                            dcSdhsEventService.accidentCongestedLocationExport(newDoc,dcSdhsEventQuery);
                        } else if ("12".equals(trafficIncidentType)){
                            //路段平均拥堵时长
                            dcSdhsEventService.accidentCongestedTimeExport(newDoc,dcSdhsEventQuery);
                        }

                    }
                }

            }




            // 保存新生成的Word文件
            /*FileOutputStream out = new FileOutputStream(outputPath);
            newDoc.write(out);
            out.close();*/

            // 不保存直接返回文件流
            newDoc.write(response.getOutputStream());
            // 关闭文档
            newDoc.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    @ApiOperation("查询上次上传的文件信息")
    @GetMapping("/getFileInfo")
    public AjaxResult getFileInfo()
    {
        return dcSdhsEventService.getFileInfo();
    }

    @ApiOperation("查询事故地市分布")
    @PostMapping("/selectRegionAnalysis")
    public AjaxResult selectRegionAnalysis(@RequestBody DcSdhsEventQuery dcSdhsEventQuery)
    {
        return dcSdhsEventService.selectRegionAnalysis(dcSdhsEventQuery);
    }

    /**
     * 事故时间分析
     * @param dcSdhsEventQuery
     * @return
     */
    @PostMapping("/accidentTimeAnalysis")
    @ApiOperation("事故时间分析")
    public AjaxResult accidentTimeAnalysis(@RequestBody DcSdhsEventQuery dcSdhsEventQuery){
        return AjaxResult.success( dcSdhsEventService.accidentTimeAnalysis(dcSdhsEventQuery));
    }
    /**
     * 事故车型分析
     * @param dcSdhsEventQuery
     * @return
     */
    @PostMapping("/accidentModelAnalysis")
    @ApiOperation("事故车型分析")
    public AjaxResult accidentModelAnalysis(@RequestBody DcSdhsEventQuery dcSdhsEventQuery){
        return AjaxResult.success( dcSdhsEventService.accidentModelAnalysis(dcSdhsEventQuery));
    }    /**
     * 收费站出入口分析
     * @param dcSdhsEventQuery
     * @return
     */
    @PostMapping("/TollBoothStatisticsList")
    @ApiOperation("收费站出入口分析")
    public AjaxResult TollBoothStatisticsList(@RequestBody DcSdhsEventQuery dcSdhsEventQuery){
        return AjaxResult.success( dcSdhsEventService.TollBoothStatisticsList(dcSdhsEventQuery));
    }

       /**
     * 门架分析
     * @param dcSdhsEventQuery
     * @return
     */
    @PostMapping("/gantryAnalysis")
    @ApiOperation("门架分析")
    public AjaxResult gantryAnalysis(@RequestBody DcSdhsEventQuery dcSdhsEventQuery){
        return AjaxResult.success( dcSdhsEventService.trafficFlowStatistics(dcSdhsEventQuery));
    }


    @ApiOperation("查询交通管制情况分析")
    @PostMapping("/selectStationAnalysis")
    public AjaxResult selectStationAnalysis(@RequestBody DcSdhsEventQuery dcSdhsEventQuery)
    {
        return dcSdhsEventService.selectStationAnalysis(dcSdhsEventQuery);
    }

    @ApiOperation("查询交通事故类型统计")
    @PostMapping("/selectAccidentType")
    public AjaxResult selectAccidentType(@RequestBody DcSdhsEventQuery dcSdhsEventQuery)
    {
        return dcSdhsEventService.selectAccidentType(dcSdhsEventQuery);
    }

    @ApiOperation("查询天气情况统计")
    @PostMapping("/selectWeather")
    public AjaxResult selectWeather()
    {
        return dcSdhsEventService.selectWeather();
    }



}