From 34ee53680318cef28c80cf40959b6fd3d55a0c4e Mon Sep 17 00:00:00 2001
From: liuhaonan <konodioda2333@vip.qq.com>
Date: 星期四, 24 二月 2022 17:28:40 +0800
Subject: [PATCH] 功能完善

---
 ximon-admin/src/main/java/com/sandu/ximon/admin/service/PoleBindingService.java |   42 +++++++++++++++++++++++++++++++++++-------
 1 files changed, 35 insertions(+), 7 deletions(-)

diff --git a/ximon-admin/src/main/java/com/sandu/ximon/admin/service/PoleBindingService.java b/ximon-admin/src/main/java/com/sandu/ximon/admin/service/PoleBindingService.java
index 749fd9c..76c0dca 100644
--- a/ximon-admin/src/main/java/com/sandu/ximon/admin/service/PoleBindingService.java
+++ b/ximon-admin/src/main/java/com/sandu/ximon/admin/service/PoleBindingService.java
@@ -1,5 +1,7 @@
 package com.sandu.ximon.admin.service;
 
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
+import com.sandu.common.execption.BusinessException;
 import com.sandu.common.service.impl.BaseServiceImpl;
 import com.sandu.ximon.admin.param.PoleBindingParam;
 import com.sandu.ximon.dao.domain.PoleBinding;
@@ -18,17 +20,43 @@
 
     private final LightReportDataService lightReportDataService;
 
+    /**
+     * 鐏潌缁戝畾璁惧
+     *
+     * @param poleId
+     * @param param
+     * @return
+     */
     public boolean bindPole(Long poleId, PoleBindingParam param) {
         Integer deviceType = param.getDeviceType();
 //        if (PoleBindingEnums.LIGHT.getCode().equals(deviceType)) {
 //        }
-        PoleBinding poleBinding = new PoleBinding();
-        poleBinding.setPoleId(poleId);
-        poleBinding.setDeviceType(deviceType);
-        poleBinding.setDeviceName(param.getDeviceName());
-        poleBinding.setDeviceCode(param.getDeviceCode());
-        return save(poleBinding);
 
-//        return false;
+        PoleBinding one = getOne(Wrappers.lambdaQuery(PoleBinding.class).eq(PoleBinding::getDeviceCode, param.getDeviceCode()));
+        if (one == null) {
+            PoleBinding poleBinding = new PoleBinding();
+            poleBinding.setPoleId(poleId);
+            poleBinding.setDeviceType(deviceType);
+            poleBinding.setDeviceName(param.getDeviceName());
+            poleBinding.setDeviceCode(param.getDeviceCode());
+            return save(poleBinding);
+        } else {
+            throw new BusinessException("璇ヨ澶囧凡缁戝畾杩�");
+        }
+    }
+
+    /**
+     * 鐏潌瑙g粦璁惧锛屽垹闄よ澶囧墠闇�瑕佽В缁�
+     *
+     * @param deviceCode
+     */
+    public void unBindPole(String deviceCode) {
+        if (deviceCode == null)
+            throw new BusinessException("璁惧缂栧彿涓嶈兘涓虹┖");
+
+        PoleBinding one = getOne(Wrappers.lambdaQuery(PoleBinding.class).eq(PoleBinding::getDeviceCode, deviceCode));
+        if (one != null) {
+            removeById(one.getId());
+        }
     }
 }

--
Gitblit v1.9.3