实现在心跳中采集各设备信息

internal/service/device_status.go

@@ -65,3 +65,19 @@ func (dsp *DeviceStatusPool) GetAllStatuses() map[string]*DeviceStatus {
 
 	return result
 }
+
+// SetAllStatuses 全量更新设备状态池
+func (dsp *DeviceStatusPool) SetAllStatuses(statuses map[string]*DeviceStatus) {
+	dsp.mutex.Lock()
+	defer dsp.mutex.Unlock()
+
+	// 清空现有状态
+	for id := range dsp.statuses {
+		delete(dsp.statuses, id)
+	}
+
+	// 添加新状态
+	for id, status := range statuses {
+		dsp.statuses[id] = status
+	}
+}