iailab-plat.git

潘志宝

2024-11-15 4be7d863a161b64f8592a789d699e807545e7dc6

提交 \| 用户 \| 时间
a6de49	1	package com.iailab.module.data.channel.opcua.service.impl;
H	2
03e8ac	3	import cn.hutool.core.collection.CollUtil;
cfbd83	4	import com.baomidou.dynamic.datasource.annotation.DSTransactional;
a6de49	5	import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
H	6	import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
	7	import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
aecc49	8	import com.iailab.framework.common.pojo.PageResult;
03e8ac	9	import com.iailab.framework.common.util.object.BeanUtils;
a6de49	10	import com.iailab.framework.common.util.object.ConvertUtils;
H	11	import com.iailab.module.data.channel.opcua.dao.ChannelOPCUATagDao;
	12	import com.iailab.module.data.channel.opcua.dto.ChannelOPCUATagDTO;
	13	import com.iailab.module.data.channel.opcua.entity.ChannelOPCUATagEntity;
	14	import com.iailab.module.data.channel.opcua.service.ChannelOPCUATagService;
03e8ac	15	import com.iailab.module.data.channel.opcua.vo.OpcUaTagImportExcelVO;
aecc49	16	import com.iailab.module.data.channel.opcua.vo.OpcUaTagPageReqVO;
03e8ac	17	import com.iailab.module.data.channel.tag.vo.TagImportRespVO;
J	18	import com.iailab.module.data.common.enums.CommonConstant;
a6de49	19	import lombok.extern.slf4j.Slf4j;
H	20	import org.apache.poi.ss.usermodel.CellType;
	21	import org.apache.poi.xssf.usermodel.XSSFRow;
	22	import org.apache.poi.xssf.usermodel.XSSFSheet;
	23	import org.apache.poi.xssf.usermodel.XSSFWorkbook;
	24	import org.springframework.beans.factory.annotation.Value;
	25	import org.springframework.stereotype.Service;
	26	import org.springframework.web.multipart.MultipartFile;
	27
	28	import javax.annotation.Resource;
	29	import java.io.File;
	30	import java.io.FileInputStream;
03e8ac	31	import java.util.*;
J	32
	33	import static com.iailab.framework.common.exception.util.ServiceExceptionUtil.exception;
	34	import static com.iailab.module.data.enums.ErrorCodeConstants.*;
a6de49	35
H	36	/**
aecc49	37	* @author lirm
a6de49	38	* @Description
aecc49	39	* @createTime 2024年08月26日
a6de49	40	*/
H	41	@Slf4j
	42	@Service
	43	public class ChannelOPCUATagServiceImpl extends ServiceImpl<ChannelOPCUATagDao, ChannelOPCUATagEntity> implements ChannelOPCUATagService {
	44	@Resource
	45	private ChannelOPCUATagDao channelOPCUATagDao;
	46
	47	@Value("${iems.upload-dir}")
	48	private String uploadDir;
	49
	50	@Override
aecc49	51	public PageResult<ChannelOPCUATagEntity> queryPage(OpcUaTagPageReqVO reqVO) {
L	52	return channelOPCUATagDao.selectPage(reqVO);
a6de49	53	}
H	54
	55	@Override
	56	public ChannelOPCUATagEntity info(String id) {
	57	return channelOPCUATagDao.selectById(id);
	58	}
	59
	60	@Override
	61	public void add(ChannelOPCUATagEntity channelOPCUATagEntity) {
	62	channelOPCUATagDao.insert(channelOPCUATagEntity);
	63	}
	64
	65	@Override
	66	public void update(ChannelOPCUATagEntity channelOPCUATagEntity) {
	67	channelOPCUATagDao.updateById(channelOPCUATagEntity);
	68	}
	69
	70	@Override
	71	public void delete(String id) {
	72	channelOPCUATagDao.deleteById(id);
	73	}
	74
aecc49	75
L	76	@Override
	77	public List<ChannelOPCUATagEntity> getByDevice(String device) {
	78	QueryWrapper<ChannelOPCUATagEntity> queryWrapper = new QueryWrapper<>();
	79	queryWrapper.eq("device", device).orderByDesc ("create_time");
	80	return channelOPCUATagDao.selectList(queryWrapper);
	81	}
	82
a6de49	83	@Override
H	84	public List<ChannelOPCUATagDTO> selectAll() {
	85	List<ChannelOPCUATagEntity> entityList = baseMapper.selectList(
	86	null
	87	);
	88	return ConvertUtils.sourceToTarget(entityList, ChannelOPCUATagDTO.class);
	89	}
	90
	91	@Override
	92	public List<ChannelOPCUATagEntity> listByIds(List<String> ids) {
	93	return baseMapper.selectList(new QueryWrapper<ChannelOPCUATagEntity>().in("id", ids));
	94	}
	95
	96	@Override
	97	public void deleteByDeviceName(String name) {
	98	baseMapper.delete(new QueryWrapper<ChannelOPCUATagEntity>().eq("device",name));
	99	}
	100
	101	@Override
cfbd83	102	@DSTransactional(rollbackFor = Exception.class)
a6de49	103	public void importTag(String device, MultipartFile file) throws Exception {
H	104	try {
	105	String suffix = file.getOriginalFilename().substring(file.getOriginalFilename().lastIndexOf("."));
	106	String fileName = UUID.randomUUID().toString() + suffix;
	107	String path = uploadDir + fileName;
	108	file.transferTo(new File(path));
	109
	110	XSSFWorkbook hssfWorkbook = new XSSFWorkbook(new FileInputStream(path));
	111	XSSFSheet sheet = hssfWorkbook.getSheetAt(0);
	112	int lastRowNum = sheet.getLastRowNum();
	113	log.info("最后一行：" + lastRowNum);
	114	int lastCellNum = 4;
	115	List<ChannelOPCUATagEntity> dangerList = new ArrayList<>();
	116	for (int i = 2; i <= lastRowNum; i++) {
	117	XSSFRow row = sheet.getRow(i);
	118	for (int j = row.getFirstCellNum(); j < lastCellNum; j++) {
	119	row.getCell(j).setCellType(CellType.STRING);
	120	}
	121	ChannelOPCUATagEntity tagEntity = new ChannelOPCUATagEntity();
	122	tagEntity.setId(UUID.randomUUID().toString());
	123	tagEntity.setTagName(row.getCell(1).getStringCellValue());
	124	tagEntity.setDataType(row.getCell(2).getStringCellValue());
	125	row.getCell(4).setCellType(CellType.STRING);
	126	if(row.getCell(3).getStringCellValue().equals("1")){
	127	tagEntity.setAddress(String.format("1%04d",Integer.parseInt(row.getCell(4).getStringCellValue())));
	128	}else if(row.getCell(3).getStringCellValue().equals("3")){
	129	tagEntity.setAddress(String.format("4%04d",Integer.parseInt(row.getCell(4).getStringCellValue())));
	130	}
03e8ac	131	tagEntity.setEnabled(1);
a6de49	132	tagEntity.setDevice(device);
H	133	tagEntity.setSamplingRate(1);
	134	dangerList.add(tagEntity);
	135	}
	136	if (CollectionUtils.isEmpty(dangerList)) {
	137	return;
	138	}
	139	//getBaseMapper().insertList(dangerList);
	140	dangerList.forEach(item -> {
	141	try {
	142	getBaseMapper().insert(item);
	143	} catch (Exception ex) {
	144	log.warn("插入异常：" + item.getTagName());
	145	}
	146	});
	147	} catch (Exception ex) {
	148	ex.printStackTrace();
	149	log.warn("导入失败！");
	150	throw ex;
	151	}
	152	}
03e8ac	153
J	154	@Override
	155	public TagImportRespVO importOpcUaTagList(List<OpcUaTagImportExcelVO> importTags, boolean isUpdateSupport, String device) {
	156	// 1.1 参数校验
	157	if (CollUtil.isEmpty(importTags)) {
	158	throw exception(TAG_IMPORT_LIST_IS_EMPTY);
	159	}
	160	// 2. 遍历，逐个创建 or 更新
	161	TagImportRespVO respVO = TagImportRespVO.builder().createTagNames(new ArrayList<>())
	162	.updateTagNames(new ArrayList<>()).failureTagNames(new LinkedHashMap<>()).build();
	163	importTags.forEach(importTag -> {
	164	// 判断如果不存在，再进行插入
	165	ChannelOPCUATagEntity existTag = channelOPCUATagDao.selectOne(new QueryWrapper<ChannelOPCUATagEntity>()
	166	.eq("device", device)
	167	.eq("tag_name",importTag.getTagName()));
	168	if (existTag == null) {
	169	ChannelOPCUATagEntity channelOpCuaTagEntity = ConvertUtils.sourceToTarget(importTag, ChannelOPCUATagEntity.class);
	170	channelOpCuaTagEntity.setId(UUID.randomUUID().toString());
	171	channelOpCuaTagEntity.setEnabled(CommonConstant.IS_ENABLE);
	172	channelOpCuaTagEntity.setDevice(device);
	173	channelOpCuaTagEntity.setCreateTime(new Date());
	174	channelOPCUATagDao.insert(channelOpCuaTagEntity);
	175
	176	respVO.getCreateTagNames().add(channelOpCuaTagEntity.getTagName());
	177	return;
	178	}
	179
	180	// 如果存在，判断是否允许更新
	181	if (!isUpdateSupport) {
	182	respVO.getFailureTagNames().put(importTag.getTagName(), TAG_EXISTS.getMsg());
	183	return;
	184	}
	185
	186	ChannelOPCUATagEntity updateTag = BeanUtils.toBean(importTag, ChannelOPCUATagEntity.class);
	187	updateTag.setId(existTag.getId());
	188	baseMapper.updateById(updateTag);
	189	respVO.getUpdateTagNames().add(importTag.getTagName());
	190	});
	191	return respVO;
	192	}
a6de49	193	}