newcpms/master/src/main/java/com/ruoyi/project/sems/controller/TMeasureThicknessController.java

package com.ruoyi.project.sems.controller;

import java.io.IOException;
import java.io.OutputStream;
import java.math.BigDecimal;
import java.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.*;

import com.ruoyi.common.utils.document.ChartUtil;
import com.ruoyi.common.utils.document.DocumentHandler;
import com.ruoyi.common.utils.document.PDFTemplateUtil;
import com.ruoyi.common.utils.file.FileUploadUtils;
import com.ruoyi.framework.config.RuoYiConfig;
import com.ruoyi.project.sems.domain.TMeasureRecord;
import com.ruoyi.project.sems.service.ITMeasureRecordService;
import freemarker.template.Template;
import org.apache.commons.lang.StringUtils;
import org.jfree.data.category.DefaultCategoryDataset;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import com.ruoyi.framework.aspectj.lang.annotation.Log;
import com.ruoyi.framework.aspectj.lang.enums.BusinessType;
import com.ruoyi.project.sems.domain.TMeasureThickness;
import com.ruoyi.project.sems.service.ITMeasureThicknessService;
import com.ruoyi.framework.web.controller.BaseController;
import com.ruoyi.framework.web.domain.AjaxResult;
import com.ruoyi.common.utils.poi.ExcelUtil;
import com.ruoyi.framework.web.page.TableDataInfo;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

/**
 * 定点测厚Controller
 *
 * @author ruoyi
 * @date 2021-07-07
 */
@RestController
@RequestMapping("/sems/thickness")
public class TMeasureThicknessController extends BaseController
{
    @Autowired
    private ITMeasureThicknessService tMeasureThicknessService;
    @Autowired
    private ITMeasureRecordService tMeasureRecordService;

    /**
     * 查询定点测厚列表
     */
    @PreAuthorize("@ss.hasPermi('sems:thickness:list')")
    @GetMapping("/list")
    public TableDataInfo list(TMeasureThickness tMeasureThickness)
    {
        startPage();
        List<TMeasureThickness> list = tMeasureThicknessService.selectTMeasureThicknessList(tMeasureThickness);
        for (TMeasureThickness m : list) {
            List<TMeasureRecord> measureRecords = tMeasureRecordService.queryRecords(m.getId().toString());
            if (measureRecords.size() > 0) {
                m.setFirstMeasureDate(measureRecords.get(measureRecords.size() - 1).getMeasureDate());
                m.setNewMeasureDate(measureRecords.get(0).getMeasureDate());
                BigDecimal measureCycle = new BigDecimal(m.getMeasureCycle());
                BigDecimal yearNum = new BigDecimal("31536000000");
                Long cycle = measureCycle.multiply(yearNum).longValue();
                Long newTime = measureRecords.get(0).getMeasureDate().getTime();
                Long nextTime = newTime + cycle;
                Date dat = new Date(nextTime);
                GregorianCalendar nextMeasureDate = new GregorianCalendar();
                nextMeasureDate.setTime(dat);
                java.text.SimpleDateFormat format = new java.text.SimpleDateFormat("yyyy-MM-dd");
                m.setNextMeasureDate(format.format(nextMeasureDate.getTime()));
            }else {
                m.setFirstMeasureDate(null);
                m.setNewMeasureDate(null);
                m.setNextMeasureDate(null);
            }
            //寿命预警
            if (m.getNextWarnDate() != null && !StringUtils.isEmpty(m.getEstRemain())) {
               int day = this.formatDate(m.getNextWarnDate(),new Date());
               BigDecimal remain = new BigDecimal(m.getEstRemain()).multiply(new BigDecimal("365")).setScale(4,BigDecimal.ROUND_HALF_DOWN);
                logger.info(day +"::::" + remain);
               if (remain.compareTo(new BigDecimal(day)) == 1) {
                    m.setWarnFlag(1);
               }else {
                   m.setWarnFlag(0);
               }
            }else {
                m.setWarnFlag(0);
            }
        }
        return getDataTable(list);
    }

    /**
     * 导出定点测厚列表
     */
    @PreAuthorize("@ss.hasPermi('sems:thickness:export')")
    @Log(title = "定点测厚", businessType = BusinessType.EXPORT)
    @GetMapping("/export")
    public AjaxResult export(TMeasureThickness tMeasureThickness)
    {
        List<TMeasureThickness> list = tMeasureThicknessService.selectTMeasureThicknessList(tMeasureThickness);
        ExcelUtil<TMeasureThickness> util = new ExcelUtil<TMeasureThickness>(TMeasureThickness.class);
        return util.exportExcel(list, "thickness");
    }

    /**
     * 获取定点测厚详细信息
     */
    @PreAuthorize("@ss.hasPermi('sems:thickness:query')")
    @GetMapping(value = "/{id}")
    public AjaxResult getInfo(@PathVariable("id") Long id)
    {
        //生成图表
        String chartUrl = "/profile/" + this.getRecordUrl(id.toString());
        TMeasureThickness t = tMeasureThicknessService.selectTMeasureThicknessById(id);
        t.setRecordUrl(chartUrl);
        return AjaxResult.success(t);
    }

    /**
     * 新增定点测厚
     */
    @PreAuthorize("@ss.hasPermi('sems:thickness:add')")
    @Log(title = "定点测厚", businessType = BusinessType.INSERT)
    @PostMapping
    public AjaxResult add(@RequestBody TMeasureThickness tMeasureThickness)
    {
        return toAjax(tMeasureThicknessService.insertTMeasureThickness(tMeasureThickness));
    }

    /**
     * 修改定点测厚
     */
    @PreAuthorize("@ss.hasPermi('sems:thickness:edit')")
    @Log(title = "定点测厚", businessType = BusinessType.UPDATE)
    @PutMapping
    public AjaxResult edit(@RequestBody TMeasureThickness tMeasureThickness)
    {
        return toAjax(tMeasureThicknessService.updateTMeasureThickness(tMeasureThickness));
    }

    /**
     * 删除定点测厚
     */
    @PreAuthorize("@ss.hasPermi('sems:thickness:remove')")
    @Log(title = "定点测厚", businessType = BusinessType.DELETE)
	@DeleteMapping("/{ids}")
    public AjaxResult remove(@PathVariable Long[] ids)
    {
        return toAjax(tMeasureThicknessService.deleteTMeasureThicknessByIds(ids));
    }

    /**
     * 定点测厚图片上传
     */
    @Log(title = "定点测厚图片上传", businessType = BusinessType.UPDATE)
    @PostMapping("/uploadFile")
    public AjaxResult uploadFile(@RequestParam("file") MultipartFile file) throws IOException
    {
        if (!file.isEmpty())
        {
            String avatar = FileUploadUtils.upload(RuoYiConfig.getFilePath("/thickness"), file);
            return AjaxResult.success(avatar);
        }
        return AjaxResult.error("上传图片异常，请联系管理员");
    }


    @RequestMapping("/measure/exportPDF")
    public String measure(@RequestParam String id, HttpServletRequest request, HttpServletResponse response) {
        OutputStream out= null;
        try {
            getRecordUrl(id);
            out = response.getOutputStream();
            logger.info("id:" + id);
            //获取信息，就是上面的结构
            HashMap<String, Object> map = new HashMap<>();
            DocumentHandler dh = new DocumentHandler();
            Template t = dh.getTemplate();
            SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
            TMeasureThickness measureThicknessEntity = tMeasureThicknessService.selectTMeasureThicknessById(Long.parseLong(id));
            //生成图表
            String chartUrl = this.getRecordUrl(id);
            map.put("chartUrl", chartUrl);
            logger.info(chartUrl);
            //测量方法
            String[] inspectionMethod = measureThicknessEntity.getInspectionMethod().split(",");
            logger.info("inspectionMethod:" + inspectionMethod);
            for (String m : inspectionMethod) {
                if (m.equals("1")){ measureThicknessEntity.setMethodUT(true); }
                if (m.equals("2")){ measureThicknessEntity.setMethodPEC(true); }
                if (m.equals("3")){ measureThicknessEntity.setMethodRT(true); }
                if (m.equals("4")){ measureThicknessEntity.setMethodEMAT(true); }
                if (m.equals("5")){ measureThicknessEntity.setMethodOther(true); }
            }
            map.put("measure", measureThicknessEntity);
            logger.info("measure:" + measureThicknessEntity);
            if(StringUtils.isNotEmpty(measureThicknessEntity.getPhoto())){
                String[] photos =  measureThicknessEntity.getPhoto().split(",");
                if (photos.length == 1) {
                    map.put("photo1", photos[0].replaceAll("/profile/",""));
                }else if (photos.length == 2) {
                    map.put("photo1", photos[0].replaceAll("/profile/",""));
                    map.put("photo2", photos[1].replaceAll("/profile/",""));
                }else if (photos.length == 3) {
                    map.put("photo1", photos[0].replaceAll("/profile/",""));
                    map.put("photo2", photos[1].replaceAll("/profile/",""));
                    map.put("photo3", photos[2].replaceAll("/profile/",""));
                }
            }
            if(StringUtils.isNotEmpty(measureThicknessEntity.getLocationUrl())){
                String[] locationUrls =  measureThicknessEntity.getLocationUrl().split(",");
                if (locationUrls.length == 1) {
                    map.put("locationUrl1", locationUrls[0].replaceAll("/profile/",""));
                }else if (locationUrls.length == 2) {
                    map.put("locationUrl1", locationUrls[0].replaceAll("/profile/",""));
                    map.put("locationUrl2", locationUrls[1].replaceAll("/profile/",""));
                    logger.info(locationUrls[0]);
                }else if (locationUrls.length == 3) {
                    map.put("locationUrl1", locationUrls[0].replaceAll("/profile/",""));
                    map.put("locationUrl2", locationUrls[1].replaceAll("/profile/",""));
                    map.put("locationUrl3", locationUrls[2].replaceAll("/profile/",""));
                }
            }
        //测量记录分行
        List<TMeasureRecord> measureRecords = tMeasureRecordService.queryRecords(id);
        List<List<TMeasureRecord>> result = new ArrayList<List<TMeasureRecord>>();
        int sourceSize = measureRecords.size();
            int size = (measureRecords.size() / 7) + 1;
            if (measureRecords.size() % 7 == 0) {
                size = size - 1;
            }
            for (int i = 0; i < size; i++) {
                List<TMeasureRecord> subset = new ArrayList<TMeasureRecord>();
                for (int j = i * 7; j < (i + 1) * 7; j++) {
                    if (j < sourceSize) {
                        subset.add(measureRecords.get(j));
                    }
                }
                if (subset.size() < 7) {
                    int supplement = 7 - subset.size();
                    for (int m = 0; m < supplement; m++) {
                        TMeasureRecord nullRecord = new TMeasureRecord();
                        subset.add(nullRecord);
                    }
                }
                result.add(subset);
            }
            map.put("result", result);
            logger.info("result:" + result);
            String name = measureThicknessEntity.getPlantCode() +"-" +measureThicknessEntity.getTagno() + "-" + measureThicknessEntity.getPosition();
            PDFTemplateUtil.exportPdf("measureFMaker.ftl", name + ".pdf","file:"+RuoYiConfig.getProfile(), map, response);
//             PDFTemplateUtil.exportPdf("measureFMaker.ftl", name + ".pdf",RuoYiConfig.getProfile(), map, response);

        } catch (Exception e) {
            e.printStackTrace();
        }finally {
            if(out!=null)
                try {
                    out.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
        }
        return null;
    }

    public String getRecordUrl(String id){
        TMeasureThickness measureThicknessEntity = tMeasureThicknessService.selectTMeasureThicknessById(Long.parseLong(id));
        List<TMeasureRecord> measureRecords = tMeasureRecordService.queryRecords(id);
        Collections.reverse(measureRecords);
        //种类数据集
        DefaultCategoryDataset dataset = new DefaultCategoryDataset();
        SimpleDateFormat sdf = new SimpleDateFormat("yyyy/MM/dd");
        for (TMeasureRecord t: measureRecords
        ) {
            dataset.addValue(t.getMeasureValue() , "数值Value" , sdf.format(t.getMeasureDate()));
            if (measureThicknessEntity.getThicknessMin() != null){
                dataset.addValue(Double.parseDouble(measureThicknessEntity.getThicknessMin()) , "最小允许壁厚" , sdf.format(t.getMeasureDate()));
            }
        }
        String url = RuoYiConfig.getProfile() + "/" + id +".jpg";
        ChartUtil.createLinePort("" , dataset ,"" ,"" , url ) ;
        return id +".jpg";
    }

    private int formatDate(Date nowDate, Date sqlDate){
        DateFormat df = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
        String newDate = df.format(nowDate);
        String sqlDate1 = df.format(sqlDate);
        try {
            nowDate = df.parse(newDate);
            sqlDate = df.parse(sqlDate1);
        } catch (Exception e) {
            e.printStackTrace();
        }
        Long time =  nowDate.getTime();
        Long time2 =  sqlDate.getTime();
        int day = (int) ((time - time2) / (24*3600*1000));
        return day;
    }
}