package com.vincent.rsf.server.manager.service.impl;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.vincent.rsf.framework.exception.CoolException;
|
import com.vincent.rsf.server.manager.controller.params.DeviceSiteParame;
|
import com.vincent.rsf.server.manager.controller.params.DeviceSiteRowParam;
|
import com.vincent.rsf.server.manager.entity.BasStation;
|
import com.vincent.rsf.server.manager.mapper.DeviceSiteMapper;
|
import com.vincent.rsf.server.manager.entity.DeviceSite;
|
import com.vincent.rsf.server.manager.service.BasStationService;
|
import com.vincent.rsf.server.manager.service.DeviceSiteService;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import org.apache.commons.lang3.StringUtils;
|
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.Collections;
|
import java.util.List;
|
import java.util.Objects;
|
|
@Service("deviceSiteService")
|
public class DeviceSiteServiceImpl extends ServiceImpl<DeviceSiteMapper, DeviceSite> implements DeviceSiteService {
|
|
/** 与表 man_device_site.target 列长度一致，超长截断避免 Data too long */
|
private static final int TARGET_MAX_LENGTH = 255;
|
|
@Autowired
|
private BasStationService basStationService;
|
|
|
/**
|
* 初始化站点：多行（设备站点、作业站点、目标站点为一组），巷道英文逗号分隔，每组×巷道×作业类型生成多条记录
|
*/
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public boolean initSites(DeviceSiteParame param) {
|
if (param.getFlagInit() != null && param.getFlagInit() == 1) {
|
List<DeviceSite> list = this.list(new LambdaQueryWrapper<DeviceSite>().select(DeviceSite::getId).last("limit 1"));
|
if (!list.isEmpty()) {
|
if (!this.remove(new LambdaQueryWrapper<>())) {
|
throw new CoolException("初始化失败： 删除站点失败！！");
|
}
|
}
|
}
|
if (Objects.isNull(param.getTypeIds()) || param.getTypeIds().isEmpty()) {
|
throw new CoolException("初始化失败： 作业类型不能为空！！");
|
}
|
if (StringUtils.isBlank(param.getChannel())) {
|
throw new CoolException("初始化失败： 巷道不能为空！！");
|
}
|
List<Integer> channels = parseChannels(param.getChannel());
|
if (channels.isEmpty()) {
|
throw new CoolException("初始化失败： 巷道格式错误，请用英文逗号分隔，如 1,2,3！！");
|
}
|
|
List<DeviceSiteRowParam> rows = param.getRows();
|
if (Objects.isNull(rows) || rows.isEmpty()) {
|
throw new CoolException("初始化失败： 请至少添加一行（设备站点、作业站点、目标站点）！！");
|
}
|
|
List<DeviceSite> deviceSites = new ArrayList<>();
|
for (DeviceSiteRowParam row : rows) {
|
if (StringUtils.isBlank(row.getDeviceSite()) || StringUtils.isBlank(row.getSite()) || StringUtils.isBlank(row.getTarget())) {
|
throw new CoolException("初始化失败： 每行的设备站点、作业站点、目标站点均不能为空！！");
|
}
|
BasStation siteStation = basStationService.getById(Long.parseLong(row.getSite().trim()));
|
if (siteStation == null) {
|
throw new CoolException("初始化失败： 作业站点未找到！！");
|
}
|
BasStation deviceStation = basStationService.getById(Long.parseLong(row.getDeviceSite().trim()));
|
if (deviceStation == null) {
|
throw new CoolException("初始化失败： 设备站点未找到！！");
|
}
|
for (Long typeId : param.getTypeIds()) {
|
for (Integer ch : channels) {
|
DeviceSite ds = new DeviceSite();
|
String siteName = siteStation.getStationName();
|
String deviceSiteName = deviceStation.getStationName();
|
String targetVal = truncate(row.getTarget().trim(), TARGET_MAX_LENGTH);
|
String commonName = StringUtils.isNotBlank(param.getName()) ? param.getName().trim() : null;
|
String commonLabel = StringUtils.isNotBlank(param.getLabel()) ? param.getLabel().trim() : null;
|
ds.setType(String.valueOf(typeId))
|
.setSite(siteName)
|
.setDevice(param.getDeviceType())
|
.setDeviceSite(deviceSiteName)
|
.setTarget(targetVal)
|
.setDeviceCode(param.getDeviceCode())
|
.setAreaIdStart(param.getAreaIdStart())
|
.setAreaIdEnd(param.getAreaIdEnd())
|
.setChannel(ch)
|
.setName(commonName != null ? commonName : (deviceSiteName + "-" + siteName + "-" + targetVal + "-" + ch))
|
.setLabel(commonLabel);
|
deviceSites.add(ds);
|
}
|
}
|
}
|
if (!this.saveBatch(deviceSites)) {
|
throw new CoolException("初始化失败： 数据保存失败！！");
|
}
|
return true;
|
}
|
|
private static String truncate(String s, int maxLen) {
|
if (s == null) return null;
|
return s.length() <= maxLen ? s : s.substring(0, maxLen);
|
}
|
|
/** 巷道英文逗号分割，解析为整数列表 */
|
private List<Integer> parseChannels(String channelStr) {
|
if (StringUtils.isBlank(channelStr)) {
|
return Collections.emptyList();
|
}
|
String[] parts = channelStr.split(",");
|
List<Integer> list = new ArrayList<>();
|
for (String p : parts) {
|
String t = (p == null) ? "" : p.trim();
|
if (t.isEmpty()) continue;
|
try {
|
list.add(Integer.parseInt(t));
|
} catch (NumberFormatException e) {
|
return Collections.emptyList();
|
}
|
}
|
return list;
|
}
|
}
|
