VideoController.java 12.4 KB
package com.skua.modules.app.video;

import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.URLDecoder;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

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

import org.jeecgframework.poi.excel.ExcelImportUtil;
import org.jeecgframework.poi.excel.def.NormalExcelConstants;
import org.jeecgframework.poi.excel.entity.ExportParams;
import org.jeecgframework.poi.excel.entity.ImportParams;
import org.jeecgframework.poi.excel.view.JeecgEntityExcelView;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.DeleteMapping;
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.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;
import org.springframework.web.servlet.ModelAndView;

import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.skua.core.api.vo.Result;
import com.skua.core.api.vo.TransTreeModel;
import com.skua.core.api.vo.TreeData;
import com.skua.core.aspect.annotation.AutoLog;
import com.skua.core.context.BaseContextHandler;
import com.skua.core.query.QueryGenerator;
import com.skua.core.util.ConvertUtils;
import com.skua.core.util.TreeUtils;
import com.skua.modules.remote.service.IsysStructSpotRelationService;
import com.skua.modules.video.entity.TVideo;
import com.skua.modules.video.entity.TVideoRelateRoot;
import com.skua.modules.video.service.ITVideoDiyRelateDiyRootService;
import com.skua.modules.video.service.ITVideoRelateRootService;
import com.skua.modules.video.service.ITVideoRootService;
import com.skua.modules.video.service.ITVideoService;
import com.skua.modules.video.vo.VideoVO;

import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;

/**
* 视频信息
*/
@Slf4j
@Api(tags="视频监控信息")
@RestController
@RequestMapping("/app/video/tVideo")
public class VideoController {
   @Autowired
   private ITVideoService tVideoService;

    @Autowired
    private ITVideoRelateRootService tVideoRelateRootService;
    //主表
    @Autowired
    private ITVideoRootService tVideoRootService;
    //关系表
    @Autowired
    private ITVideoDiyRelateDiyRootService tVideoDiyRelateDiyRootService;
    @Autowired
    private IsysStructSpotRelationService sysStructSpotRelationService;

    @AutoLog(value = "摄像头列表")
    @ApiOperation(value="摄像头列表", notes="摄像头列表")
    @GetMapping(value = "/list")
    public Result<Object> videoMessage(){
        Result<Object> result = new Result<Object>();
        //获取摄像头所有的列表
        String factoryids = BaseContextHandler.getDeparts();
        List<Map<String,Object>> videoList = tVideoService.queryVideoList(factoryids);
        //将业务数据转换为构造树所需的数据结构
        List<TransTreeModel> transTreeModels = videoList.stream().map(bo -> convertToTreeModel(bo)).collect(Collectors.toList());
        //此处工具类构建树通过拼接root构造根节点
        List<TreeData> treeList= TreeUtils.buildTree(transTreeModels,"root");
        result.setResult(treeList);
        return result;
    }

    /**
     * 	将摄像头列表数据填充到TransTreeModel中
     * @param bo
     * @return
     */
    private TransTreeModel convertToTreeModel(Map<String, Object> bo) {
        TransTreeModel transTreeModel = new TransTreeModel();
        transTreeModel.setId(String.valueOf(bo.get("id")));
        transTreeModel.setPid(String.valueOf(bo.get("parent_id")));
        transTreeModel.setTitle(String.valueOf(bo.get("video_name")));
        Map<String, String> map = new HashMap<>(4);
        map.put("is_leaf",String.valueOf(bo.get("is_leaf")));
        map.put("video_camerauuid",String.valueOf(bo.get("video_camerauuid")));
        transTreeModel.setMap(map);
        return transTreeModel;
    }

    @AutoLog(value = "通过构筑物id获取视频配置信息")
    @ApiOperation(value="视频信息-获取视频配置信息查询", notes="视频信息-获取视频配置信息查询")
    @GetMapping(value = "/querystructCodeList")
    public Result<List<TVideo>> querystructCodeList(String structCodeId) {
        Result<List<TVideo>> result = new Result<List<TVideo>>();
        List<TVideo> list = tVideoService.querystructCodeList(structCodeId);
        result.setSuccess(true);
        result.setResult(list);
        return result;
    }

    @AutoLog(value = "通过构筑物id获取视频配置信息")
    @ApiOperation(value="视频信息-获取视频配置信息查询", notes="视频信息-获取视频配置信息查询")
    @GetMapping(value = "/queryByVideoId")
    public Result<List<TVideo>> queryByVideoId(String id) {
        Result<List<TVideo>> result = new Result<List<TVideo>>();
        List<TVideo> list = tVideoService.queryByVideoId(id);
        result.setSuccess(true);
        result.setResult(list);
        return result;
    }
    
    @ApiOperation(value="视频信息-模糊搜索摄像头", notes="视频信息-模糊搜索摄像头")
    @GetMapping(value = "/queryByVideoName")
    public Result<List<VideoVO>> queryByVideoName(String videoName) {
        Result<List<VideoVO>> result = new Result<List<VideoVO>>();
        List<VideoVO> list = tVideoService.queryByVideoName(videoName);
        result.setSuccess(true);
        result.setResult(list);
        return result;
    }


   /**
     *   添加
    * @param tVideo
    * @return
    */
   @AutoLog(value = "视频信息-添加")
   @ApiOperation(value="视频信息-添加", notes="视频信息-添加")
   @PostMapping(value = "/add")
   public Result<TVideo> add(@RequestBody TVideo tVideo) {
       Result<TVideo> result = new Result<TVideo>();
       try {
           Object falg = tVideoService.save(tVideo);
           if((boolean)falg){
               TVideoRelateRoot videoRelateRoot = new TVideoRelateRoot();
               videoRelateRoot.setId(tVideo.getId());
               videoRelateRoot.setVideoRootId(tVideo.getStructCodeId());
               videoRelateRoot.setVideoId(tVideo.getId());
               videoRelateRoot.setFactoryId(tVideo.getFactoryId());
               tVideoRelateRootService.save(videoRelateRoot);
           }
           result.success("添加成功!");
       } catch (Exception e) {
           log.error(e.getMessage(),e);
           result.error500("操作失败");
       }
       return result;
   }

   /**
     *  编辑
    * @param tVideo
    * @return
    */
   @AutoLog(value = "视频信息-编辑")
   @ApiOperation(value="视频信息-编辑", notes="视频信息-编辑")
   @PutMapping(value = "/edit")
   public Result<TVideo> edit(@RequestBody TVideo tVideo) {
       Result<TVideo> result = new Result<TVideo>();
       TVideo tVideoEntity = tVideoService.getById(tVideo.getId());
       if(tVideoEntity==null) {
           result.error500("未找到对应实体");
       }else {
           boolean ok = tVideoService.updateById(tVideo);
           //TODO 返回false说明什么?
           if(ok) {
               TVideoRelateRoot videoRelateRoot = new TVideoRelateRoot();
               videoRelateRoot.setId(tVideo.getId());
               videoRelateRoot.setVideoRootId(tVideo.getStructCodeId());
               videoRelateRoot.setVideoId(tVideo.getId());
               videoRelateRoot.setFactoryId(tVideo.getFactoryId());
               tVideoRelateRootService.updateById(videoRelateRoot);
               result.success("修改成功!");
           }
       }

       return result;
   }

   /**
     *   通过id删除
    * @param id
    * @return
    */
   @AutoLog(value = "视频信息-通过id删除")
   @ApiOperation(value="视频信息-通过id删除", notes="视频信息-通过id删除")
   @DeleteMapping(value = "/delete")
   public Result<?> delete(@RequestParam(name="id",required=true) String id) {
       try {
           tVideoService.removeById(id);
       } catch (Exception e) {
           log.error("删除失败",e.getMessage());
           return Result.error("删除失败!");
       }
       return Result.ok("删除成功!");
   }

   /**
     *  批量删除
    * @param ids
    * @return
    */
   @AutoLog(value = "视频信息-批量删除")
   @ApiOperation(value="视频信息-批量删除", notes="视频信息-批量删除")
   @DeleteMapping(value = "/deleteBatch")
   public Result<TVideo> deleteBatch(@RequestParam(name="ids",required=true) String ids) {
       Result<TVideo> result = new Result<TVideo>();
       if(ids==null || "".equals(ids.trim())) {
           result.error500("参数不识别!");
       }else {
           this.tVideoService.removeByIds(Arrays.asList(ids.split(",")));
           result.success("删除成功!");
       }
       return result;
   }

   /**
     * 通过id查询
    * @param id
    * @return
    */
   @AutoLog(value = "视频信息-通过id查询")
   @ApiOperation(value="视频信息-通过id查询", notes="视频信息-通过id查询")
   @GetMapping(value = "/queryById")
   public Result<TVideo> queryById(@RequestParam(name="id",required=true) String id) {
       Result<TVideo> result = new Result<TVideo>();
       TVideo tVideo = tVideoService.getById(id);
       if(tVideo==null) {
           result.error500("未找到对应实体");
       }else {
           result.setResult(tVideo);
           result.setSuccess(true);
       }
       return result;
   }

 /**
     * 导出excel
  *
  * @param request
  * @param response
  */
 @RequestMapping(value = "/exportXls")
 public ModelAndView exportXls(HttpServletRequest request, HttpServletResponse response) {
     // Step.1 组装查询条件
     QueryWrapper<TVideo> queryWrapper = null;
     try {
         String paramsStr = request.getParameter("paramsStr");
         if (ConvertUtils.isNotEmpty(paramsStr)) {
             String deString = URLDecoder.decode(paramsStr, "UTF-8");
             TVideo tVideo = JSON.parseObject(deString, TVideo.class);
             queryWrapper = QueryGenerator.initQueryWrapper(tVideo, request.getParameterMap());
         }
     } catch (UnsupportedEncodingException e) {
         e.printStackTrace();
     }

     //Step.2 AutoPoi 导出Excel
     ModelAndView mv = new ModelAndView(new JeecgEntityExcelView());
     List<TVideo> pageList = tVideoService.list(queryWrapper);
     //导出文件名称
     mv.addObject(NormalExcelConstants.FILE_NAME, "视频信息列表");
     mv.addObject(NormalExcelConstants.CLASS, TVideo.class);
     mv.addObject(NormalExcelConstants.PARAMS, new ExportParams("视频信息列表数据", "导出人:Jeecg", "导出信息"));
     mv.addObject(NormalExcelConstants.DATA_LIST, pageList);
     return mv;
 }

 /**
     * 通过excel导入数据
  *
  * @param request
  * @param response
  * @return
  */
 @RequestMapping(value = "/importExcel", method = RequestMethod.POST)
 public Result<?> importExcel(HttpServletRequest request, HttpServletResponse response) {
     MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
     Map<String, MultipartFile> fileMap = multipartRequest.getFileMap();
     for (Map.Entry<String, MultipartFile> entity : fileMap.entrySet()) {
         MultipartFile file = entity.getValue();// 获取上传文件对象
         ImportParams params = new ImportParams();
         params.setTitleRows(2);
         params.setHeadRows(1);
         params.setNeedSave(true);
         try {
             List<TVideo> listTVideos = ExcelImportUtil.importExcel(file.getInputStream(), TVideo.class, params);
             tVideoService.saveBatch(listTVideos);
             return Result.ok("文件导入成功!数据行数:" + listTVideos.size());
         } catch (Exception e) {
             log.error(e.getMessage(),e);
             return Result.error("文件导入失败:"+e.getMessage());
         } finally {
             try {
                 file.getInputStream().close();
             } catch (IOException e) {
                 e.printStackTrace();
             }
         }
     }
     return Result.ok("文件导入失败!");
 }

}