package com.gk.hotwork.Controller;
|
|
|
import com.alibaba.fastjson.JSONObject;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.gk.hotwork.Controller.Base.BaseController;
|
import com.gk.hotwork.Domain.Enum.ErrorCode;
|
import com.gk.hotwork.Domain.Exception.BusinessException;
|
import com.gk.hotwork.Domain.RiskControlMeasure;
|
import com.gk.hotwork.Domain.Utils.FileOptUtils;
|
import com.gk.hotwork.Domain.Utils.FilterObject;
|
import com.gk.hotwork.Domain.Utils.Msg;
|
import com.gk.hotwork.Domain.Vo.RiskControlMeasureExportVo;
|
import com.gk.hotwork.Service.ExcelExportService;
|
import com.gk.hotwork.Service.RiskControlMeasureService;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiImplicitParam;
|
import io.swagger.annotations.ApiImplicitParams;
|
import io.swagger.annotations.ApiOperation;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.*;
|
import java.util.List;
|
|
import static com.gk.hotwork.Domain.Utils.Properties.filePath;
|
|
|
@Api(tags = "安全风险管控措施")
|
@RestController
|
@RequestMapping("/riskControlMeasure")
|
public class RiskControlMeasureController extends BaseController {
|
|
@Autowired
|
private RiskControlMeasureService riskControlMeasureService;
|
@Autowired
|
private ExcelExportService excelExportService;
|
|
@ApiOperation("分页")
|
@PostMapping("/page")
|
@ApiImplicitParams({
|
@ApiImplicitParam(name = "pageIndex",value = "当前页码"),
|
@ApiImplicitParam(name = "pageSize",value = "每页行数"),
|
@ApiImplicitParam(name = "filter.riskEventName",value = "{}"),
|
@ApiImplicitParam(name = "filter.riskUnitName",value = "{}"),
|
@ApiImplicitParam(name = "filter.dataSrc",value = "{}"),
|
})
|
public Msg selectPage(@RequestBody FilterObject filterObject) {
|
Integer pageIndex = filterObject.getPageIndex();
|
Integer pageSize = filterObject.getPageSize();
|
IPage page = riskControlMeasureService.selectPage(new Page<>(pageIndex, pageSize), filterObject.getFilter(), getUser());
|
return success(page);
|
}
|
|
@ApiOperation("ID查找管控措施")
|
@ApiImplicitParams({
|
@ApiImplicitParam(name = "id",value = "管控措施ID")
|
})
|
@GetMapping("/get/id")
|
public Msg selectById(Long id){
|
Msg msg = new Msg();
|
RiskControlMeasure measure = riskControlMeasureService.getById(id);
|
if(measure != null && measure.getId() > 0){
|
msg.setCode("200");
|
msg.setResult(measure);
|
}else {
|
msg.setCode(ErrorCode.ERROR_50001.getCode());
|
msg.setMessage(ErrorCode.ERROR_50001.getMsg());
|
}
|
return msg;
|
}
|
|
|
@ApiOperation("新增")
|
@PostMapping("/add")
|
@ApiImplicitParams({
|
@ApiImplicitParam(name = "riskEventId",value = "所属安全事件id"),
|
@ApiImplicitParam(name = "dataSrc",value = "自动化监控:1;隐患排查2 (数据字典)"),
|
@ApiImplicitParam(name = "riskMeasureDesc",value = "管控措施描述"),
|
@ApiImplicitParam(name = "classify1",value = "分类1"),
|
@ApiImplicitParam(name = "classify2",value = "分类2"),
|
@ApiImplicitParam(name = "classify3",value = "企业自定义分类"),
|
@ApiImplicitParam(name = "troubleshootContent",value = "隐患排查内容"),
|
})
|
public Msg add(@RequestBody RiskControlMeasure riskControlMeasureVo) {
|
riskControlMeasureService.addOne(riskControlMeasureVo, getUser());
|
return success();
|
}
|
|
@ApiOperation("修改")
|
@PostMapping("/mod")
|
@ApiImplicitParams({
|
@ApiImplicitParam(name = "id",value = "id"),
|
@ApiImplicitParam(name = "riskEventId",value = "所属安全事件id"),
|
@ApiImplicitParam(name = "dataSrc",value = "自动化监控:1;隐患排查2 (数据字典)"),
|
@ApiImplicitParam(name = "riskMeasureDesc",value = "管控措施描述"),
|
@ApiImplicitParam(name = "classify1",value = "分类1"),
|
@ApiImplicitParam(name = "classify3",value = "企业自定义分类"),
|
@ApiImplicitParam(name = "troubleshootContent",value = "隐患排查内容"),
|
})
|
public Msg mod(@RequestBody RiskControlMeasure riskControlMeasureVo) {
|
riskControlMeasureService.modOne(riskControlMeasureVo, getUser());
|
return success();
|
}
|
|
@ApiOperation("删除")
|
@PostMapping("/del")
|
@ApiImplicitParams({
|
@ApiImplicitParam(name = "id",value = "id"),
|
})
|
public Msg del(@RequestBody JSONObject jsonObject) {
|
Long id = jsonObject.getLong("id");
|
riskControlMeasureService.delOne(id, getUser());
|
return success();
|
}
|
|
@ApiOperation("安全风险管控措施数据导出")
|
@GetMapping("/export")
|
@ApiImplicitParams({
|
@ApiImplicitParam(name = "filter.riskEventName",value = "{}"),
|
@ApiImplicitParam(name = "filter.dataSrc",value = "{}"),
|
})
|
public Msg infoExport(HttpServletRequest request, HttpServletResponse response) throws Exception {
|
Msg msg = new Msg();
|
msg.setCode("200");
|
msg.setMessage("success");
|
|
List<RiskControlMeasureExportVo> list = riskControlMeasureService.getRiskControlMeasureList(request, response);
|
|
msg.setResult(list);
|
|
return msg;
|
}
|
//List<RiskControlMeasureExportVo> list = riskControlMeasureService.getRiskControlMeasureList();
|
|
|
@ApiOperation("安全风险管控措施数据导入")
|
@PostMapping("/import")
|
public Msg infoImport(MultipartFile file) {
|
Msg msg = new Msg();
|
msg.setCode("200");
|
msg.setMessage("success");
|
|
String filesave ="";
|
try {
|
if (file == null)
|
{
|
msg.setCode("404");
|
msg.setMessage("上传文件未找到");
|
return msg;
|
}
|
|
long size = file.getSize();
|
if(0 == size)
|
{
|
msg.setCode("404");
|
msg.setMessage("上传文件大小为空");
|
return msg;
|
}
|
|
if (!FileOptUtils.isDirExists(filePath)) {
|
msg.setCode(ErrorCode.ERROR_10004.getCode());
|
msg.setMessage("发生错误或不为目录");
|
return msg;
|
}
|
|
if (!FileOptUtils.isDirExists(filePath)) {
|
msg.setCode(ErrorCode.ERROR_10004.getCode());
|
msg.setMessage("发生错误或不为目录");
|
return msg;
|
}
|
|
filesave = filePath + getUser().getRealname() + "_" + ".xlsx";
|
|
file.transferTo(new File(filesave));
|
InputStream in = new FileInputStream(filesave);
|
String name = file.getOriginalFilename();
|
Boolean isExcel2007 = name.substring(name.lastIndexOf(".") + 1).endsWith("xlsx") ? true : false;
|
//riskEventService.importRiskEventData(param, getUser());
|
excelExportService.importRiskControlMeasureExcel(in, getUser().getRealname(), isExcel2007);
|
|
} catch (BusinessException e) {
|
e.printStackTrace();
|
msg.setCode(ErrorCode.ERROR_10004.getCode());
|
msg.setMessage(e.getMessage());
|
} catch (Exception e) {
|
e.printStackTrace();
|
msg.setCode(ErrorCode.ERROR_10004.getCode());
|
msg.setMessage("导入发生错误");
|
}
|
|
return msg;
|
}
|
|
}
|
