From 5de7d78bc629f316749b9e11b9c6113d5e0d0b8f Mon Sep 17 00:00:00 2001
From: skyouc
Date: 星期四, 27 三月 2025 11:20:23 +0800
Subject: [PATCH] 新增站点初始化功能

---
 rsf-server/src/main/java/com/vincent/rsf/server/manager/service/impl/DeviceSiteServiceImpl.java |   49 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 49 insertions(+), 0 deletions(-)

diff --git a/rsf-server/src/main/java/com/vincent/rsf/server/manager/service/impl/DeviceSiteServiceImpl.java b/rsf-server/src/main/java/com/vincent/rsf/server/manager/service/impl/DeviceSiteServiceImpl.java
index e19660b..9328ef1 100644
--- a/rsf-server/src/main/java/com/vincent/rsf/server/manager/service/impl/DeviceSiteServiceImpl.java
+++ b/rsf-server/src/main/java/com/vincent/rsf/server/manager/service/impl/DeviceSiteServiceImpl.java
@@ -1,12 +1,61 @@
 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.mapper.DeviceSiteMapper;
 import com.vincent.rsf.server.manager.entity.DeviceSite;
 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.stereotype.Service;
+
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
+import java.util.Objects;
 
 @Service("deviceSiteService")
 public class DeviceSiteServiceImpl extends ServiceImpl<DeviceSiteMapper, DeviceSite> implements DeviceSiteService {
 
+    /**
+     * 鍒濆鍖栫珯鐐�
+     * @param param
+     * @return
+     */
+    @Override
+    public boolean initSites(DeviceSiteParame param) {
+        if (param.getFlagInit() == 1) {
+            if (!this.remove(new LambdaQueryWrapper<>())) {
+                throw new CoolException("鍒濆鍖栧け璐ワ細 鍒犻櫎绔欑偣澶辫触锛侊紒");
+            }
+        }
+        if (Objects.isNull(param.getDeviceSites()) || param.getDeviceSites().isEmpty()) {
+            throw new CoolException("鍒濆鍖栧け璐ワ細 璁惧浣滀笟绔欑偣涓嶈兘涓虹┖锛侊紒");
+        }
+        if (Objects.isNull(param.getSites()) || param.getSites().isEmpty()) {
+            throw new CoolException("鍒濆鍖栧け璐�: 浣滀笟绔欑偣涓嶈兘涓虹┖锛侊紒");
+        }
+        if (Objects.isNull(param.getTypeIds()) || param.getTypeIds().isEmpty()) {
+            throw new CoolException("鍒濆鍖栧け璐ワ細 浣滀笟绫诲瀷涓嶈兘涓虹┖锛侊紒");
+        }
+        List<DeviceSite> deviceSites =  new ArrayList<>();
+        for (String site : param.getSites()) {
+            for (String deviceSite : param.getDeviceSites()) {
+                for (Long id : param.getTypeIds()) {
+                    DeviceSite site1 = new DeviceSite();
+                    site1.setType(id + "")
+                            .setSite(site)
+                            .setDevice(param.getDeviceType())
+                            .setDeviceSite(deviceSite)
+                            .setDeviceCode(param.getDeviceCode());
+                    deviceSites.add(site1);
+                }
+            }
+        }
+        if (!this.saveBatch(deviceSites)) {
+            throw new CoolException("鍒濆鍖栧け璐ワ細 鏁版嵁淇濆瓨澶辫触锛侊紒");
+        }
+        return true;
+    }
 }

--
Gitblit v1.9.1