package com.ruoyi.aibrain.controller; import java.util.List; import javax.servlet.http.HttpServletResponse; import com.ruoyi.HIKVISION.GetPicFromIPV; import org.springframework.security.access.prepost.PreAuthorize; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.GetMapping; import org.springframework.web.bind.annotation.PostMapping; import org.springframework.web.bind.annotation.PutMapping; import org.springframework.web.bind.annotation.DeleteMapping; import org.springframework.web.bind.annotation.PathVariable; import org.springframework.web.bind.annotation.RequestBody; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RestController; import com.ruoyi.common.annotation.Log; import com.ruoyi.common.core.controller.BaseController; import com.ruoyi.common.core.domain.AjaxResult; import com.ruoyi.common.enums.BusinessType; import com.ruoyi.aibrain.domain.AiCamera; import com.ruoyi.aibrain.service.IAiCameraService; import com.ruoyi.common.utils.poi.ExcelUtil; import com.ruoyi.common.core.page.TableDataInfo; /** * 设备管理Controller * * @author wf * @date 2023-03-08 */ @RestController @RequestMapping("/aibrain/camera") public class AiCameraController extends BaseController { @Autowired private IAiCameraService aiCameraService; /** * 查询设备管理列表 */ @PreAuthorize("@ss.hasPermi('aibrain:camera:list')") @GetMapping("/list") public TableDataInfo list(AiCamera aiCamera) { // startPage(); List list = aiCameraService.selectAiCameraList(aiCamera); return getDataTable(list); } /** * 导出设备管理列表 */ @PreAuthorize("@ss.hasPermi('aibrain:camera:export')") @Log(title = "设备管理", businessType = BusinessType.EXPORT) @PostMapping("/export") public void export(HttpServletResponse response, AiCamera aiCamera) { List list = aiCameraService.selectAiCameraList(aiCamera); ExcelUtil util = new ExcelUtil(AiCamera.class); util.exportExcel(response, list, "设备管理数据"); } /** * 获取设备管理详细信息 */ @PreAuthorize("@ss.hasPermi('aibrain:camera:query')") @GetMapping(value = "/{id}") public AjaxResult getInfo(@PathVariable("id") Long id) { return success(aiCameraService.selectAiCameraById(id)); } /** * 新增设备管理 */ @PreAuthorize("@ss.hasPermi('aibrain:camera:add')") @Log(title = "设备管理", businessType = BusinessType.INSERT) @PostMapping public AjaxResult add(@RequestBody AiCamera aiCamera) { return toAjax(aiCameraService.insertAiCamera(aiCamera)); } /** * 修改设备管理 */ @PreAuthorize("@ss.hasPermi('aibrain:camera:edit')") @Log(title = "设备管理", businessType = BusinessType.UPDATE) @PutMapping public AjaxResult edit(@RequestBody AiCamera aiCamera) { return toAjax(aiCameraService.updateAiCamera(aiCamera)); } /** * 删除设备管理 */ @PreAuthorize("@ss.hasPermi('aibrain:camera:remove')") @Log(title = "设备管理", businessType = BusinessType.DELETE) @DeleteMapping("/{ids}") public AjaxResult remove(@PathVariable Long[] ids) { return toAjax(aiCameraService.deleteAiCameraByIds(ids)); } /**String index out of range: -2 * 重启设备 */ @PostMapping("startCamera") public AjaxResult startCamera(AiCamera aiCamera) { return AjaxResult.success(); } /** * 设备全部重启 */ @PostMapping("startCameraAll") public AjaxResult startCameraAll() { return AjaxResult.success(); } /** * 抓取摄像头图片 */ @PostMapping("getCameraPic") public AjaxResult getCameraPic(AiCamera aiCamera) { AiCamera camera = aiCameraService.selectAiCameraById(aiCamera.getId()); GetPicFromIPV getPicFromIPV = new GetPicFromIPV(); String msg = getPicFromIPV.getDVRPic(camera, false); if(!"200".equals(msg)){ return AjaxResult.error(msg); } return AjaxResult.success("抓取成功"); } }