package com.gunshi.project.hsz.controller; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.toolkit.IdWorker; import com.baomidou.mybatisplus.core.toolkit.ObjectUtils; import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.fasterxml.jackson.databind.ObjectMapper; import com.gunshi.core.result.R; import com.gunshi.project.hsz.entity.so.AttCctvBasePage; import com.gunshi.project.hsz.entity.vo.CctvControlVo; import com.gunshi.project.hsz.model.AttCctvBase; import com.gunshi.project.hsz.model.CctvBMenu; import com.gunshi.project.hsz.service.AttCctvBaseService; import com.gunshi.project.hsz.service.CctvBMenuService; import com.gunshi.project.hsz.util.OkHttpUtil; import com.gunshi.project.hsz.common.validate.markers.Insert; import com.gunshi.project.hsz.common.validate.markers.Update; import io.swagger.v3.oas.annotations.Operation; import io.swagger.v3.oas.annotations.media.Schema; import io.swagger.v3.oas.annotations.tags.Tag; import okhttp3.OkHttpClient; import okhttp3.Request; import okhttp3.Response; import org.apache.commons.collections4.CollectionUtils; import org.apache.commons.lang3.StringUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.validation.annotation.Validated; import org.springframework.web.bind.annotation.*; import java.io.IOException; import java.io.Serializable; import java.util.Date; import java.util.List; import java.util.Map; import java.util.Objects; import java.util.stream.Collectors; /** * 描述: 视频基本信息表 * author: xusan * date: 2024-07-08 17:40:37 */ @Tag(name = "视频基本信息表") @RestController @RequestMapping(value="/attCctvBase") public class AttCctvBaseController { @Autowired private AttCctvBaseService service; @Autowired private CctvBMenuService menuService; @Operation(summary = "新增") @PostMapping("/insert") public R insert(@Validated(Insert.class) @RequestBody AttCctvBase dto) { dto.setId(IdWorker.getId()); dto.setCreateTime(new Date()); if (StringUtils.isNotBlank(dto.getName()) && service.lambdaQuery().eq(AttCctvBase::getName,dto.getName()) .count() > 0){ throw new IllegalArgumentException("当前名称已存在"); } if (Objects.nonNull(dto.getMenuId()) && menuService.lambdaQuery().eq(CctvBMenu::getId,dto.getMenuId()) .count() == 0 ){ throw new IllegalArgumentException("当前视频点目录不存在"); } dto.setCreateTime(new Date()); boolean result = service.save(dto); return R.ok(result ? dto : null); } @Operation(summary = "修改") @PostMapping("/update") public R update(@Validated(Update.class) @RequestBody AttCctvBase dto) { if (Objects.isNull(service.getById(dto.getId()))){ throw new RuntimeException("当前数据不存在"); } if (StringUtils.isNotBlank(dto.getName()) && service.lambdaQuery().eq(AttCctvBase::getName,dto.getName()) .ne(AttCctvBase::getId,dto.getId()) .count() > 0){ throw new IllegalArgumentException("当前名称已存在"); } if (Objects.nonNull(dto.getMenuId()) && menuService.lambdaQuery().eq(CctvBMenu::getId,dto.getMenuId()) .count() == 0 ){ throw new IllegalArgumentException("当前视频点目录不存在"); } dto.setCreateTime(null); boolean result = service.updateById(dto); return R.ok(result ? dto : null); } @Operation(summary = "删除") @GetMapping("/del/{id}") public R del(@Schema(name = "id") @PathVariable("id") Serializable id) { if (Objects.isNull(service.getById(id))){ throw new RuntimeException("当前数据不存在"); } return R.ok(service.removeById(id)); } @Operation(summary = "列表") @PostMapping("/list") public R> list() { List res = service.queryList(); return R.ok(res); } @Operation(summary = "分页") @PostMapping("/page") public R> page(@RequestBody @Validated AttCctvBasePage page) { LambdaQueryWrapper query = Wrappers.lambdaQuery(); if (ObjectUtils.isNotNull(page.getCode())) { query.like(AttCctvBase::getIndexCode, page.getCode()); } if (ObjectUtils.isNotNull(page.getMenuId())) { query.eq(AttCctvBase::getMenuId, page.getMenuId()); } if (ObjectUtils.isNotNull(page.getName())) { query.like(AttCctvBase::getName, page.getName()); } if (ObjectUtils.isNotNull(page.getType())) { query.eq(AttCctvBase::getType, Integer.valueOf(page.getType())); } query.orderByDesc(AttCctvBase::getCreateTime); Page basePage = service.page(page.getPageSo().toPage(), query); List records = basePage.getRecords(); if (CollectionUtils.isNotEmpty(records)){ List list = menuService.list(); if (CollectionUtils.isNotEmpty(list)){ Map> listMap = list.stream(). collect(Collectors.groupingBy(CctvBMenu::getId, Collectors.mapping(CctvBMenu::getName, Collectors.toList()))); records.forEach(item -> { if (Objects.nonNull(item.getMenuId())){ item.setMenuName(listMap.get(item.getMenuId()).getFirst()); } }); } basePage.setRecords(records); } return R.ok(basePage); } @Operation(summary = "获取预览地址") @GetMapping("/preview/{indexCode}") public R preview(@PathVariable("indexCode") String indexCode) { String api = "http://10.42.6.68:7000/liveAddressLimited?cameraIndexCode=%s&protocol=ws&token=111"; OkHttpClient client = OkHttpUtil.build(); String ret = null; try { Response resp = client.newCall(new Request.Builder().url(String.format(api, indexCode)).build()).execute(); String respStr = resp.body().string(); ObjectMapper om = new ObjectMapper(); Map map = om.readValue(respStr, Map.class); ret = map.get("data").toString(); } catch (IOException e) { throw new RuntimeException(e); } return R.ok(ret); } @Operation(summary = "云台控制") @PostMapping("/control") public R control(@RequestBody CctvControlVo vo) { String indexCode = vo.getIndexCode(); Integer action = vo.getAction(); Integer speed = vo.getSpeed(); String command = vo.getCommand(); String api = "http://10.42.6.68:7000/controlling?cameraIndexCode="+indexCode+"&action="+action+"&speed="+speed+"&command="+command+"&token=111"; OkHttpClient client = OkHttpUtil.build(); String ret = null; try { Response resp = client.newCall(new Request.Builder().url(api).build()).execute(); ret = resp.body().string(); } catch (IOException e) { throw new RuntimeException(e); } return R.ok(ret); } }