package com.ruoyi.business.uav.service.impl;
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.ruoyi.business.aidetection.config.Analyzer;
|
import com.ruoyi.business.aidetection.config.ZLMediaKit;
|
import com.ruoyi.business.aidetection.domain.AvControl;
|
import com.ruoyi.business.aidetection.service.AvControlService;
|
import com.ruoyi.business.uav.config.MqttConsumerConfig;
|
import com.ruoyi.business.uav.domain.UavConfig;
|
import com.ruoyi.business.uav.domain.UavConfigMessage;
|
import com.ruoyi.business.uav.domain.UavMessage;
|
import com.ruoyi.business.uav.domain.vo.UavConfigVo;
|
import com.ruoyi.business.uav.mapper.UavConfigMapper;
|
import com.ruoyi.business.uav.service.UavConfigMessageService;
|
import com.ruoyi.business.uav.service.UavConfigService;
|
import com.ruoyi.business.uav.service.UavMessageService;
|
import com.ruoyi.common.utils.StringUtils;
|
import org.hamcrest.text.IsEmptyString;
|
import org.springframework.beans.BeanUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
|
|
import java.util.ArrayList;
|
import java.util.Arrays;
|
import java.util.List;
|
import java.util.Map;
|
|
/**
|
* @Author：yuankun
|
* @Package：com.ruoyi.business.uav.service.impl
|
* @Project：ruoyi-vue-service
|
* @name：UavConfigServiceImpl
|
* @Date：2024/5/12 15:27
|
* @Filename：UavConfigServiceImpl
|
*/
|
@Service
|
public class UavConfigServiceImpl extends ServiceImpl<UavConfigMapper, UavConfig> implements UavConfigService {
|
|
@Autowired
|
private UavConfigMessageService uavConfigMessageService;
|
@Autowired
|
private UavMessageService uavMessageService;
|
@Autowired
|
private MqttConsumerConfig mqttConsumerConfig;
|
|
@Autowired
|
private AvControlService avControlService;
|
@Autowired
|
private Analyzer analyzer;
|
@Autowired
|
private ZLMediaKit zlMediaKit;
|
|
|
@Override
|
public List<UavConfigVo> queryList(UavConfigVo entity) {
|
//查询符合条件的无人机配置信息
|
QueryWrapper<UavConfig> queryWrapper = new QueryWrapper<>();
|
if(entity.getUavBrand() != null||"".equals(entity.getUavBrand()))
|
queryWrapper.like("uav_brand", entity.getUavBrand());
|
if(entity.getUavModel() != null||"".equals(entity.getUavBrand()))
|
queryWrapper.like("uav_model", entity.getUavModel());
|
if(entity.getState() != null)
|
queryWrapper.eq("state", entity.getState());
|
|
List<UavConfig> uavConfigList = baseMapper.selectList(queryWrapper);
|
List<UavConfigVo> uavConfigVoList = new ArrayList<>();
|
//查找无人机的订阅信息
|
for (UavConfig uavConfig : uavConfigList) {
|
UavConfigVo uavConfigVo = new UavConfigVo();
|
BeanUtils.copyProperties(uavConfig, uavConfigVo);
|
//先查询关联表中无人机配置的所有订阅信息的id
|
List<UavConfigMessage> uavConfigMessageList = uavConfigMessageService.queryUavConfigMessageByConfigId(uavConfigVo.getId());
|
List<String> messageTopic = new ArrayList<>();
|
for (UavConfigMessage uavConfigMessage:uavConfigMessageList) {
|
//根据订阅信息id查询订阅topic
|
messageTopic.add(uavMessageService.queryById(uavConfigMessage.getMessageId()).getMessageTopic());
|
}
|
//如果有订阅信息就拼接成字符串
|
if(messageTopic.size() > 0)
|
uavConfigVo.setMessageTopic(String.join(",", messageTopic));
|
uavConfigVoList.add(uavConfigVo);
|
//查询无人机的布控信息
|
String[] split = uavConfigVo.getUavControlId().split(",");
|
List<String> ControlName = new ArrayList<>();
|
for (String ControlId:split) {
|
AvControl avControl = avControlService.getById(ControlId);
|
if(avControl!=null){
|
ControlName.add(avControl.getName());
|
}
|
}
|
if(ControlName.size() > 0){
|
uavConfigVo.setUavControlName(String.join(",", ControlName));
|
}
|
|
}
|
return uavConfigVoList;
|
}
|
|
@Override
|
public List<UavConfigVo> queryAll(UavConfigVo entity) {
|
return this.baseMapper.queryList(entity);
|
}
|
|
@Override
|
public UavConfigVo queryById(Long id) {
|
//查询符合条件的无人机配置信息
|
UavConfig uavConfig = baseMapper.selectById(id);
|
//查找无人机的订阅信息
|
UavConfigVo uavConfigVo = new UavConfigVo();
|
BeanUtils.copyProperties(uavConfig, uavConfigVo);
|
//先查询关联表中无人机配置的所有订阅信息的id
|
List<UavConfigMessage> uavConfigMessageList = uavConfigMessageService.queryUavConfigMessageByConfigId(uavConfigVo.getId());
|
List<String> messageTopic = new ArrayList<>();
|
for (UavConfigMessage uavConfigMessage:uavConfigMessageList) {
|
//根据订阅信息id查询订阅topic
|
messageTopic.add(uavMessageService.queryById(uavConfigMessage.getMessageId()).getMessageTopic());
|
}
|
//如果有订阅信息就拼接成字符串
|
if(messageTopic.size() > 0)
|
uavConfigVo.setMessageTopic(String.join(",", messageTopic));
|
|
|
|
return uavConfigVo;
|
}
|
|
@Override
|
@Transactional//防止出现脏数据
|
public boolean updateConfigById(UavConfigVo uavConfigVo) throws Exception {
|
|
UavConfig uavConfig = baseMapper.selectById(uavConfigVo.getId());
|
BeanUtils.copyProperties(uavConfigVo, uavConfig);
|
baseMapper.updateById(uavConfig);
|
//删除所有的关联表信息
|
uavConfigMessageService.deleteByConfigId(uavConfigVo.getId());
|
List<String> split = new ArrayList<>();
|
if(uavConfigVo.getMessageTopic()!=null){
|
split= Arrays.asList(uavConfigVo.getMessageTopic().split(","));
|
}else{
|
throw new Exception("请添加无人机订阅的主题！");
|
}
|
|
|
|
for (String messageTopic:split) {
|
if(uavMessageService.queryByTopic(messageTopic)==null){
|
throw new Exception("您所添加的订阅主题不存在！请先添加订阅主题！");
|
}
|
uavConfigMessageService.save(new UavConfigMessage(uavMessageService.queryByTopic(messageTopic).getId(),uavConfigVo.getId()));
|
|
|
|
}
|
return true;
|
}
|
|
@Override
|
@Transactional//防止出现脏数据
|
public boolean saveUavConfigVo(UavConfigVo uavConfigVo) throws Exception {
|
QueryWrapper<UavConfig> queryWrapper = new QueryWrapper<>();
|
if(!StringUtils.isEmpty(uavConfigVo.getUavId())&&!StringUtils.isEmpty(uavConfigVo.getAddress())&&!StringUtils.isEmpty(uavConfigVo.getUavClient())&&
|
!StringUtils.isEmpty(uavConfigVo.getUsername())&&!StringUtils.isEmpty(uavConfigVo.getPassword())&&
|
!StringUtils.isEmpty(uavConfigVo.getPort().toString())){
|
queryWrapper.eq("uav_id", uavConfigVo.getUavId());
|
queryWrapper.eq("address", uavConfigVo.getAddress());
|
queryWrapper.eq("uav_client", uavConfigVo.getUavClient());
|
queryWrapper.eq("username", uavConfigVo.getUsername());
|
queryWrapper.eq("port", uavConfigVo.getPort());
|
if(baseMapper.selectList(queryWrapper).size() > 0){
|
throw new Exception("该无人机配置信息已存在！");
|
}else{
|
if(!StringUtils.isEmpty(uavConfigVo.getMessageTopic())){//检测是否已经存在订阅消息主题
|
if(uavMessageService.queryByTopic(uavConfigVo.getMessageTopic())==null){
|
throw new Exception("您无人机所添加的订阅主题不存在！请先去消息管理添加订阅主题！");
|
}
|
uavConfigMessageService.save(new UavConfigMessage(uavMessageService.queryByTopic(uavConfigVo.getMessageTopic()).getId(),uavConfigVo.getId()));
|
}
|
|
baseMapper.insert(uavConfigVo);
|
}
|
}else{
|
throw new Exception("请填写完整无人机配置信息！");
|
}
|
return true;
|
}
|
|
@Override
|
@Transactional
|
public boolean connectUav(Long id) throws Exception {
|
UavConfigVo uavConfigVo = queryById(id);
|
if(StringUtils.isEmpty(uavConfigVo.getMessageTopic())){
|
throw new Exception("无订阅消息，先添加订阅再连接！");
|
}
|
|
//连接Mqtt网关
|
mqttConsumerConfig.connect(uavConfigVo);
|
//订阅主题
|
String[] split = uavConfigVo.getMessageTopic().split(",");
|
int[] qos=new int[split.length];
|
int i=0;
|
for (String messageTopic:split) {
|
UavMessage uavMessage = uavMessageService.queryByTopic(messageTopic);
|
qos[i]=uavMessage.getMessageQos().intValue();;
|
i++;
|
}
|
|
mqttConsumerConfig.subscribe(split,qos);
|
mqttConsumerConfig.setIsConnected(true);
|
uavConfigVo.setState(1L);
|
baseMapper.updateById(uavConfigVo);
|
|
return true;
|
}
|
|
@Override
|
@Transactional
|
public boolean disconnectUav(Long id) throws Exception {
|
UavConfigVo uavConfigVo = queryById(id);
|
if(uavConfigVo.getState()!=1){
|
throw new Exception("当前不是处于连接状态！");
|
}
|
if(StringUtils.isEmpty(uavConfigVo.getMessageTopic())){
|
throw new Exception("无订阅消息，先添加订阅再取消连接！");
|
}
|
|
//设定无人机离线
|
uavConfigVo.setUavState(0);
|
//设定无人机识别服务状态
|
uavConfigVo.setUavAiState(0);
|
//取消识别服务
|
if(uavConfigVo.getUavId()!=null){
|
QueryWrapper<AvControl> avControlQueryWrapper=new QueryWrapper<>();
|
avControlQueryWrapper.eq("uav_sn",uavConfigVo.getUavId());
|
avControlService.list(avControlQueryWrapper).forEach(avControl->{
|
Map<String, Object> map = analyzer.controlCancel(avControl.getCode());
|
if("200".equals(map.get("code").toString()))
|
avControl.setState(0L);
|
avControlService.updateById(avControl);
|
});
|
}
|
//取消连接Mqtt网关
|
System.out.println(mqttConsumerConfig.getIsConnected());
|
if(mqttConsumerConfig.getIsConnected()){
|
|
mqttConsumerConfig.disConnect();
|
}
|
//设定离线状态
|
uavConfigVo.setState(0L);
|
mqttConsumerConfig.setIsConnected(false);
|
baseMapper.updateById(uavConfigVo);
|
return true;
|
}
|
}
|
