package com.gk.hotwork.Controller; import com.gk.hotwork.Controller.Base.BaseController; import com.gk.hotwork.Domain.RiskMap; import com.gk.hotwork.Domain.Utils.Msg; import com.gk.hotwork.Domain.Vo.RiskMapVO; import com.gk.hotwork.Service.RiskMapService; 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 java.util.List; @Api(tags = "安全风险分布图") @RestController @RequestMapping("/riskMap") public class RiskMapController extends BaseController { @Autowired private RiskMapService riskMapService; @ApiOperation("添加一张图") @PostMapping("/add") @ApiImplicitParams({ @ApiImplicitParam(name = "createUid",value = "创建人ID"), @ApiImplicitParam(name = "img",value = "图片路径"), @ApiImplicitParam(name = "descInfo",value = "描述信息") }) public Msg insertOne(@RequestBody RiskMapVO riskMapVO){ Msg msg = new Msg(); if(getUser()!=null && getUser().getId()!=null && riskMapVO.getCreateUid()!=null && getUser().getId().equals(riskMapVO.getCreateUid())){ String rs = riskMapService.saveOne(riskMapVO); if(rs == null){ msg.setCode("200"); }else { msg.setCode("500"); msg.setMessage(rs); } }else { msg.setCode("301"); msg.setMessage("权限错误"); } return msg; } @ApiOperation("查找列表") @GetMapping("/list") @ApiImplicitParams({ @ApiImplicitParam(name = "status",value = "状态 0-有效 , 1-删除 ,不传查所有状态",required = false) }) public Msg getList(@RequestParam(required = false) Byte status){ Msg msg = new Msg(); msg.setCode("200"); List voList = riskMapService.listByStatus(status); if(voList == null || voList.size() == 0){ msg.setMessage("未找到数据"); }else { msg.setResult(voList); } return msg; } @ApiOperation("删除图片") @GetMapping("/del") @ApiImplicitParams({ @ApiImplicitParam(name = "id",value = "图片ID"), }) public Msg delete(@RequestParam Long id){ Msg msg = new Msg(); boolean rs = riskMapService.deleteOne(id,getUser().getId()); if(rs == true){ msg.setCode("200"); }else { msg.setCode("300"); msg.setMessage("删除失败"); } return msg; } }