84 lines
1.9 KiB
Go
84 lines
1.9 KiB
Go
// Package service 提供各种业务服务功能
|
|
package service
|
|
|
|
import (
|
|
"sync"
|
|
)
|
|
|
|
// DeviceStatus 设备状态信息
|
|
type DeviceStatus struct {
|
|
// Active 设备是否启动
|
|
Active bool
|
|
}
|
|
|
|
// DeviceStatusPool 设备状态池，用于管理所有设备的当前状态
|
|
type DeviceStatusPool struct {
|
|
// statuses 设备状态映射 设备ID:状态
|
|
statuses map[string]*DeviceStatus
|
|
|
|
// mutex 读写锁，保证并发安全
|
|
mutex sync.RWMutex
|
|
}
|
|
|
|
// NewDeviceStatusPool 创建设备状态池实例
|
|
func NewDeviceStatusPool() *DeviceStatusPool {
|
|
return &DeviceStatusPool{
|
|
statuses: make(map[string]*DeviceStatus),
|
|
}
|
|
}
|
|
|
|
// SetStatus 设置设备状态
|
|
func (dsp *DeviceStatusPool) SetStatus(deviceID string, status *DeviceStatus) {
|
|
dsp.mutex.Lock()
|
|
defer dsp.mutex.Unlock()
|
|
|
|
dsp.statuses[deviceID] = status
|
|
}
|
|
|
|
// GetStatus 获取设备状态
|
|
func (dsp *DeviceStatusPool) GetStatus(deviceID string) (*DeviceStatus, bool) {
|
|
dsp.mutex.RLock()
|
|
defer dsp.mutex.RUnlock()
|
|
|
|
status, exists := dsp.statuses[deviceID]
|
|
return status, exists
|
|
}
|
|
|
|
// DeleteStatus 删除设备状态
|
|
func (dsp *DeviceStatusPool) DeleteStatus(deviceID string) {
|
|
dsp.mutex.Lock()
|
|
defer dsp.mutex.Unlock()
|
|
|
|
delete(dsp.statuses, deviceID)
|
|
}
|
|
|
|
// GetAllStatuses 获取所有设备状态
|
|
func (dsp *DeviceStatusPool) GetAllStatuses() map[string]*DeviceStatus {
|
|
dsp.mutex.RLock()
|
|
defer dsp.mutex.RUnlock()
|
|
|
|
// 创建副本以避免外部修改
|
|
result := make(map[string]*DeviceStatus)
|
|
for id, status := range dsp.statuses {
|
|
result[id] = status
|
|
}
|
|
|
|
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
|
|
}
|
|
}
|