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.RiskEvent;
|
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.RiskEventExportVo;
|
import com.gk.hotwork.Service.ExcelExportService;
|
import com.gk.hotwork.Service.RiskEventService;
|
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.File;
|
import java.io.FileInputStream;
|
import java.io.InputStream;
|
import java.text.SimpleDateFormat;
|
import java.util.Date;
|
import java.util.List;
|
|
import static com.gk.hotwork.Domain.Utils.Properties.filePath;
|
|
|
@Api(tags = "安全风险事件")
|
@RestController
|
@RequestMapping("/riskEvent")
|
public class RiskEventController extends BaseController {
|
|
@Autowired
|
private RiskEventService riskEventService;
|
|
@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 = "{}"),
|
})
|
public Msg selectPage(@RequestBody FilterObject filterObject) {
|
Integer pageIndex = filterObject.getPageIndex();
|
Integer pageSize = filterObject.getPageSize();
|
IPage page = riskEventService.selectPage(new Page<>(pageIndex, pageSize), filterObject.getFilter(), getUser());
|
return success(page);
|
}
|
|
@ApiOperation("/新增")
|
@PostMapping("/add")
|
@ApiImplicitParams({
|
@ApiImplicitParam(name = "riskUnitId",value = "所属安全风险单元id"),
|
@ApiImplicitParam(name = "riskEventName",value = "安全风险事件名称"),
|
})
|
public Msg add(@RequestBody RiskEvent riskEventVo) {
|
riskEventService.addOne(riskEventVo, getUser());
|
return success();
|
}
|
|
@ApiOperation("/修改")
|
@PostMapping("/mod")
|
@ApiImplicitParams({
|
@ApiImplicitParam(name = "id",value = "id"),
|
@ApiImplicitParam(name = "riskUnitId",value = "所属安全风险单元id"),
|
@ApiImplicitParam(name = "riskEventName",value = "安全风险事件名称"),
|
})
|
public Msg mod(@RequestBody RiskEvent riskEventVo) {
|
riskEventService.modOne(riskEventVo, getUser());
|
return success();
|
}
|
|
@ApiOperation("/删除")
|
@PostMapping("/del")
|
@ApiImplicitParams({
|
@ApiImplicitParam(name = "id",value = "id"),
|
})
|
public Msg mod(@RequestBody JSONObject jsonObject) {
|
Long id = jsonObject.getLong("id");
|
riskEventService.delOne(id, getUser());
|
return success();
|
}
|
|
@ApiOperation("安全风险事件导出")
|
@GetMapping("/export")
|
@ApiImplicitParams({
|
@ApiImplicitParam(name = "filter.riskEventName",value = "{}"),
|
@ApiImplicitParam(name = "filter.riskUnitName",value = "{}"),
|
})
|
public Msg infoExport(HttpServletRequest request, HttpServletResponse response) throws Exception {
|
|
Msg msg = new Msg();
|
msg.setCode("200");
|
msg.setMessage("success");
|
|
List<RiskEventExportVo> riskEventList = riskEventService.getRiskEventList(request, response);
|
msg.setResult(riskEventList);
|
|
return msg;
|
}
|
|
@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.importRiskEventExcel(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;
|
//
|
}
|
}
|
