From 1571cd0f137ced4345fa8785e166a29dc31b6ad1 Mon Sep 17 00:00:00 2001
From: feiyu02 <risaku@163.com>
Date: 星期二, 13 五月 2025 17:42:39 +0800
Subject: [PATCH] 1. 新增动态污染溯源的数据异常判断逻辑 2. 新增动态污染溯源websocket连接功能

---
 src/main/kotlin/com/flightfeather/uav/socket/DeviceSession.kt |   54 ++++++++++++++++++++++++++----------------------------
 1 files changed, 26 insertions(+), 28 deletions(-)

diff --git a/src/main/kotlin/com/flightfeather/uav/socket/DeviceSession.kt b/src/main/kotlin/com/flightfeather/uav/socket/DeviceSession.kt
index d387fb3..8ee5393 100644
--- a/src/main/kotlin/com/flightfeather/uav/socket/DeviceSession.kt
+++ b/src/main/kotlin/com/flightfeather/uav/socket/DeviceSession.kt
@@ -11,38 +11,36 @@
  */
 class DeviceSession {
 
-    companion object{
-        private val deviceMap = ConcurrentHashMap<String, ChannelHandlerContext?>()
-        private val typeMap = ConcurrentHashMap<String, List<AirTypeData>>()
+    companion object {
         private const val DEFAULT_DEVICE = "default_device"
+    }
+    private val deviceMap = ConcurrentHashMap<String?, ChannelHandlerContext?>()
+    private val typeMap = ConcurrentHashMap<String, List<AirTypeData>>()
 
-        fun saveDevice(deviceCode: String?, channel: ChannelHandlerContext?) {
-            deviceCode?.let {
-                deviceMap.put(deviceCode, channel)
-            }
+    fun saveDevice(deviceCode: String?, channel: ChannelHandlerContext?) {
+        deviceMap[deviceCode] = channel
+    }
+
+    fun getDevice(deviceCode: String?): ChannelHandlerContext? {
+        return if (deviceMap.containsKey(deviceCode))
+            deviceMap[deviceCode]
+        else
+            null
+    }
+
+    fun saveAirType(deviceCode: String?, types: List<AirTypeData>) {
+        if (deviceCode == null) {
+            typeMap[DEFAULT_DEVICE] = types
+        } else {
+            typeMap[deviceCode] = types
         }
+    }
 
-        fun getDevice(deviceCode: String?): ChannelHandlerContext? {
-            return if (deviceMap.containsKey(deviceCode))
-                deviceMap[deviceCode]
-            else
-                null
-        }
-
-        fun saveAirType(deviceCode: String?, types: List<AirTypeData>) {
-            if (deviceCode == null) {
-                typeMap[DEFAULT_DEVICE] = types
-            } else {
-                typeMap[deviceCode] = types
-            }
-        }
-
-        fun getAirType(deviceCode: String?): List<AirTypeData>? {
-            return when {
-                deviceCode == null -> typeMap[DEFAULT_DEVICE]
-                typeMap.containsKey(deviceCode) -> typeMap[deviceCode]
-                else -> null
-            }
+    fun getAirType(deviceCode: String?): List<AirTypeData>? {
+        return when {
+            deviceCode == null -> typeMap[DEFAULT_DEVICE]
+            typeMap.containsKey(deviceCode) -> typeMap[deviceCode]
+            else -> null
         }
     }
 }
\ No newline at end of file

--
Gitblit v1.9.3