package com.iailab.module.data.channel.opcua.service.impl;
|
|
import cn.hutool.core.collection.CollUtil;
|
import com.baomidou.dynamic.datasource.annotation.DSTransactional;
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.iailab.framework.common.pojo.PageResult;
|
import com.iailab.framework.common.util.object.BeanUtils;
|
import com.iailab.framework.common.util.object.ConvertUtils;
|
import com.iailab.module.data.channel.opcua.dao.ChannelOPCUATagDao;
|
import com.iailab.module.data.channel.opcua.dto.ChannelOPCUATagDTO;
|
import com.iailab.module.data.channel.opcua.entity.ChannelOPCUATagEntity;
|
import com.iailab.module.data.channel.opcua.service.ChannelOPCUATagService;
|
import com.iailab.module.data.channel.opcua.vo.OpcUaTagImportExcelVO;
|
import com.iailab.module.data.channel.opcua.vo.OpcUaTagPageReqVO;
|
import com.iailab.module.data.channel.tag.vo.TagImportRespVO;
|
import com.iailab.module.data.common.enums.CommonConstant;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.poi.ss.usermodel.CellType;
|
import org.apache.poi.xssf.usermodel.XSSFRow;
|
import org.apache.poi.xssf.usermodel.XSSFSheet;
|
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.stereotype.Service;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import javax.annotation.Resource;
|
import java.io.File;
|
import java.io.FileInputStream;
|
import java.util.*;
|
|
import static com.iailab.framework.common.exception.util.ServiceExceptionUtil.exception;
|
import static com.iailab.module.data.enums.ErrorCodeConstants.*;
|
|
/**
|
* @author lirm
|
* @Description
|
* @createTime 2024年08月26日
|
*/
|
@Slf4j
|
@Service
|
public class ChannelOPCUATagServiceImpl extends ServiceImpl<ChannelOPCUATagDao, ChannelOPCUATagEntity> implements ChannelOPCUATagService {
|
@Resource
|
private ChannelOPCUATagDao channelOPCUATagDao;
|
|
@Value("${iems.upload-dir}")
|
private String uploadDir;
|
|
@Override
|
public PageResult<ChannelOPCUATagEntity> queryPage(OpcUaTagPageReqVO reqVO) {
|
return channelOPCUATagDao.selectPage(reqVO);
|
}
|
|
@Override
|
public ChannelOPCUATagEntity info(String id) {
|
return channelOPCUATagDao.selectById(id);
|
}
|
|
@Override
|
public void add(ChannelOPCUATagEntity channelOPCUATagEntity) {
|
channelOPCUATagDao.insert(channelOPCUATagEntity);
|
}
|
|
@Override
|
public void update(ChannelOPCUATagEntity channelOPCUATagEntity) {
|
channelOPCUATagDao.updateById(channelOPCUATagEntity);
|
}
|
|
@Override
|
public void delete(String id) {
|
channelOPCUATagDao.deleteById(id);
|
}
|
|
|
@Override
|
public List<ChannelOPCUATagEntity> getByDevice(String device) {
|
QueryWrapper<ChannelOPCUATagEntity> queryWrapper = new QueryWrapper<>();
|
queryWrapper.eq("device", device).orderByDesc ("create_time");
|
return channelOPCUATagDao.selectList(queryWrapper);
|
}
|
|
@Override
|
public List<ChannelOPCUATagDTO> selectAll() {
|
List<ChannelOPCUATagEntity> entityList = baseMapper.selectList(
|
null
|
);
|
return ConvertUtils.sourceToTarget(entityList, ChannelOPCUATagDTO.class);
|
}
|
|
@Override
|
public List<ChannelOPCUATagEntity> listByIds(List<String> ids) {
|
return baseMapper.selectList(new QueryWrapper<ChannelOPCUATagEntity>().in("id", ids));
|
}
|
|
@Override
|
public void deleteByDeviceName(String name) {
|
baseMapper.delete(new QueryWrapper<ChannelOPCUATagEntity>().eq("device",name));
|
}
|
|
@Override
|
@DSTransactional(rollbackFor = Exception.class)
|
public void importTag(String device, MultipartFile file) throws Exception {
|
try {
|
String suffix = file.getOriginalFilename().substring(file.getOriginalFilename().lastIndexOf("."));
|
String fileName = UUID.randomUUID().toString() + suffix;
|
String path = uploadDir + fileName;
|
file.transferTo(new File(path));
|
|
XSSFWorkbook hssfWorkbook = new XSSFWorkbook(new FileInputStream(path));
|
XSSFSheet sheet = hssfWorkbook.getSheetAt(0);
|
int lastRowNum = sheet.getLastRowNum();
|
log.info("最后一行：" + lastRowNum);
|
int lastCellNum = 4;
|
List<ChannelOPCUATagEntity> dangerList = new ArrayList<>();
|
for (int i = 2; i <= lastRowNum; i++) {
|
XSSFRow row = sheet.getRow(i);
|
for (int j = row.getFirstCellNum(); j < lastCellNum; j++) {
|
row.getCell(j).setCellType(CellType.STRING);
|
}
|
ChannelOPCUATagEntity tagEntity = new ChannelOPCUATagEntity();
|
tagEntity.setId(UUID.randomUUID().toString());
|
tagEntity.setTagName(row.getCell(1).getStringCellValue());
|
tagEntity.setDataType(row.getCell(2).getStringCellValue());
|
row.getCell(4).setCellType(CellType.STRING);
|
if(row.getCell(3).getStringCellValue().equals("1")){
|
tagEntity.setAddress(String.format("1%04d",Integer.parseInt(row.getCell(4).getStringCellValue())));
|
}else if(row.getCell(3).getStringCellValue().equals("3")){
|
tagEntity.setAddress(String.format("4%04d",Integer.parseInt(row.getCell(4).getStringCellValue())));
|
}
|
tagEntity.setEnabled(1);
|
tagEntity.setDevice(device);
|
tagEntity.setSamplingRate(1);
|
dangerList.add(tagEntity);
|
}
|
if (CollectionUtils.isEmpty(dangerList)) {
|
return;
|
}
|
//getBaseMapper().insertList(dangerList);
|
dangerList.forEach(item -> {
|
try {
|
getBaseMapper().insert(item);
|
} catch (Exception ex) {
|
log.warn("插入异常：" + item.getTagName());
|
}
|
});
|
} catch (Exception ex) {
|
ex.printStackTrace();
|
log.warn("导入失败！");
|
throw ex;
|
}
|
}
|
|
@Override
|
public TagImportRespVO importOpcUaTagList(List<OpcUaTagImportExcelVO> importTags, boolean isUpdateSupport, String device) {
|
// 1.1 参数校验
|
if (CollUtil.isEmpty(importTags)) {
|
throw exception(TAG_IMPORT_LIST_IS_EMPTY);
|
}
|
// 2. 遍历，逐个创建 or 更新
|
TagImportRespVO respVO = TagImportRespVO.builder().createTagNames(new ArrayList<>())
|
.updateTagNames(new ArrayList<>()).failureTagNames(new LinkedHashMap<>()).build();
|
importTags.forEach(importTag -> {
|
// 判断如果不存在，再进行插入
|
ChannelOPCUATagEntity existTag = channelOPCUATagDao.selectOne(new QueryWrapper<ChannelOPCUATagEntity>()
|
.eq("device", device)
|
.eq("tag_name",importTag.getTagName()));
|
if (existTag == null) {
|
ChannelOPCUATagEntity channelOpCuaTagEntity = ConvertUtils.sourceToTarget(importTag, ChannelOPCUATagEntity.class);
|
channelOpCuaTagEntity.setId(UUID.randomUUID().toString());
|
channelOpCuaTagEntity.setEnabled(CommonConstant.IS_ENABLE);
|
channelOpCuaTagEntity.setDevice(device);
|
channelOpCuaTagEntity.setCreateTime(new Date());
|
channelOPCUATagDao.insert(channelOpCuaTagEntity);
|
|
respVO.getCreateTagNames().add(channelOpCuaTagEntity.getTagName());
|
return;
|
}
|
|
// 如果存在，判断是否允许更新
|
if (!isUpdateSupport) {
|
respVO.getFailureTagNames().put(importTag.getTagName(), TAG_EXISTS.getMsg());
|
return;
|
}
|
|
ChannelOPCUATagEntity updateTag = BeanUtils.toBean(importTag, ChannelOPCUATagEntity.class);
|
updateTag.setId(existTag.getId());
|
baseMapper.updateById(updateTag);
|
respVO.getUpdateTagNames().add(importTag.getTagName());
|
});
|
return respVO;
|
}
|
}
|
