基于Spring Cloud分布式物联网(IOT)平台源码.zip

/* * Copyright 2016-2021 Pnoker. All Rights Reserved. * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * http://www.apache.org/licenses/LICENSE-2.0 * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package com.dc3.center.manager.service.impl; import cn.hutool.core.util.StrUtil; import com.dc3.center.manager.service.*; import com.dc3.common.bean.batch.*; import com.dc3.common.bean.driver.AttributeInfo; import com.dc3.common.bean.driver.DriverMetadata; import com.dc3.common.constant.Common; import com.dc3.common.exception.NotFoundException; import com.dc3.common.exception.ServiceException; import com.dc3.common.model.*; import lombok.extern.slf4j.Slf4j; import org.springframework.beans.BeanUtils; import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; import javax.annotation.Resource; import java.util.*; import java.util.concurrent.ConcurrentHashMap; import java.util.stream.Collectors; /** * <p>BatchService Impl * * @author pnoker */ @Slf4j @Service public class BatchServiceImpl implements BatchService { @Resource private DriverService driverService; @Resource private DriverAttributeService driverAttributeService; @Resource private DriverInfoService driverInfoService; @Resource private ProfileService profileService; @Resource private ProfileBindService profileBindService; @Resource private GroupService groupService; @Resource private DeviceService deviceService; @Resource private PointService pointService; @Resource private PointAttributeService pointAttributeService; @Resource private PointInfoService pointInfoService; @Resource private NotifyService notifyService; @Override @Transactional public void batchImport(List<BatchDriver> batchDrivers) { batchDrivers.forEach(batchDriver -> { if (StrUtil.isBlank(batchDriver.getServiceName())) { throw new ServiceException("Driver service name is blank"); } Driver driver = driverService.selectByServiceName(batchDriver.getServiceName()); if (null == driver) { throw new ServiceException("Driver service does not exist: " + batchDriver.getServiceName()); } batchDriver.getProfiles().forEach(batchProfile -> { // import Profile Profile profile = importProfile(driver, batchProfile); // import Driver Info Array // todo bug importDriverInfo(driver, null, batchProfile.getDriverConfig()); // import Point Array importPoint(profile, batchProfile.getPoints()); // import Device Array importDevice(driver, profile, batchProfile.getGroups(), batchProfile.getPointConfig()); }); }); } @Override public BatchDriver batchExport(String serviceName) { Driver driver = driverService.selectByServiceName(serviceName); BatchDriver batchDriver = new BatchDriver(); batchDriver.setServiceName(serviceName); // export profile List<BatchProfile> profiles = exportProfile(driver.getId()); if (profiles.size() > 0) { batchDriver.setProfiles(profiles); } return batchDriver; } @Override public DriverMetadata batchDriverMetadata(String serviceName) { DriverMetadata driverMetadata = new DriverMetadata(); Driver driver = driverService.selectByServiceName(serviceName); try { Map<Long, DriverAttribute> driverAttributeMap = getDriverAttributeMap(driver.getId()); driverMetadata.setDriverAttributeMap(driverAttributeMap); Map<Long, PointAttribute> pointAttributeMap = getPointAttributeMap(driver.getId()); driverMetadata.setPointAttributeMap(pointAttributeMap); List<Device> devices = deviceService.selectByDriverId(driver.getId()); Set<Long> deviceIds = devices.stream().map(Description::getId).collect(Collectors.toSet()); Map<Long, Map<String, AttributeInfo>> driverInfoMap = getDriverInfoMap(deviceIds, driverAttributeMap); driverMetadata.setDriverInfoMap(driverInfoMap); Map<Long, Device> deviceMap = getDeviceMap(devices); driverMetadata.setDeviceMap(deviceMap); Map<Long, Map<Long, Point>> profilePointMap = getProfileMap(deviceIds); driverMetadata.setProfileMap(profilePointMap); Map<Long, Map<Long, Map<String, AttributeInfo>>> devicePointInfoMap = getPointInfoMap(devices, profilePointMap, pointAttributeMap); driverMetadata.setPointInfoMap(devicePointInfoMap); return driverMetadata; } catch (NotFoundException ignored) { } return driverMetadata; } /** * 导入 Profile * * @param driver Driver * @param batchProfile BatchProfile */ private Profile importProfile(Driver driver, BatchProfile batchProfile) { if (StrUtil.isBlank(batchProfile.getName())) { throw new ServiceException("Profile name is blank"); } Profile profile; try { profile = profileService.selectByName(batchProfile.getName(), driver.getTenantId()); profile.setShare(batchProfile.getShare()); profile.setDescription("批量导入：更新操作"); profile = profileService.update(profile); notifyService.notifyDriverProfile(Common.Driver.Profile.UPDATE, profile); } catch (NotFoundException notFoundException) { profile = new Profile(batchProfile.getName(), batchProfile.getShare(), driver.getId(), driver.getTenantId()); profile.setDescription("批量导入：新增操作"); profile = profileService.add(profile); notifyService.notifyDriverProfile(Common.Driver.Profile.ADD, profile); } return profile; } /** * 导入 Driver Info 列表 * * @param driver Driver * @param device Device * @param driverConfig Driver Config */ private void importDriverInfo(Driver driver, Device device, final Map<String, String> driverConfig) { List<String> driverInfoList = new ArrayList<>(16); if (null == driverConfig) { return; } driverConfig.forEach((name, value) -> { DriverAttribute driverAttribute = driverAttributeService.selectByNameAndDriverId(name, driver.getId()); if (driverInfoList.contains(name)) { throw new ServiceException("Repeatedly driver info: " + name); } driverInfoList.add(name); try { DriverInfo driverInfo = driverInfoService.selectByAttributeIdAndDeviceId(driverAttribute.getId(), device.getId()); driverInfo.setDescription("批量导入：更新操作"); driverInfo = driverInfoService.update(driverInfo.setValue(value)); notifyService.notifyDriverDriverInfo(Common.Driver.DriverInfo.UPDATE, driverInfo); } catch (NotFoundException notFoundException) { DriverInfo driverInfo = new DriverInfo(driverAttribute.getId(), value, device.getId()); driverInfo.setDescription("批量导入：新增操作"); driverInfo = driverInfoService.add(driverInfo); notifyService.notifyDriverDriverInfo(Common.Driver.DriverInfo.ADD, driverInfo);