修复图表时间和查询区间不匹配问题
This commit is contained in:
135
backend/internal/device/device.go
Normal file
135
backend/internal/device/device.go
Normal file
@@ -0,0 +1,135 @@
|
||||
package device
|
||||
|
||||
import (
|
||||
"sync"
|
||||
)
|
||||
|
||||
// Device 设备信息
|
||||
type Device struct {
|
||||
ID string `json:"id"`
|
||||
Name string `json:"name"`
|
||||
IP string `json:"ip"`
|
||||
Token string `json:"token"` // 设备认证令牌
|
||||
Status string `json:"status"` // active, inactive, offline
|
||||
CreatedAt int64 `json:"created_at"`
|
||||
UpdatedAt int64 `json:"updated_at"`
|
||||
}
|
||||
|
||||
// Storage 设备存储接口
|
||||
type Storage interface {
|
||||
// GetDevices 获取所有设备
|
||||
GetDevices() []Device
|
||||
// GetDevice 获取指定设备
|
||||
GetDevice(id string) (Device, bool)
|
||||
// GetDeviceByToken 通过令牌获取设备
|
||||
GetDeviceByToken(token string) (Device, bool)
|
||||
// AddDevice 添加设备
|
||||
AddDevice(device Device) error
|
||||
// UpdateDevice 更新设备
|
||||
UpdateDevice(device Device) error
|
||||
// DeleteDevice 删除设备
|
||||
DeleteDevice(id string) error
|
||||
// UpdateDeviceStatus 更新设备状态
|
||||
UpdateDeviceStatus(id string, status string) error
|
||||
}
|
||||
|
||||
// MemoryStorage 内存设备存储
|
||||
type MemoryStorage struct {
|
||||
devices map[string]Device
|
||||
mutex sync.RWMutex
|
||||
}
|
||||
|
||||
// NewMemoryStorage 创建内存设备存储实例
|
||||
func NewMemoryStorage() *MemoryStorage {
|
||||
return &MemoryStorage{
|
||||
devices: make(map[string]Device),
|
||||
}
|
||||
}
|
||||
|
||||
// GetDevices 获取所有设备
|
||||
func (s *MemoryStorage) GetDevices() []Device {
|
||||
s.mutex.RLock()
|
||||
defer s.mutex.RUnlock()
|
||||
|
||||
devices := make([]Device, 0, len(s.devices))
|
||||
for _, device := range s.devices {
|
||||
devices = append(devices, device)
|
||||
}
|
||||
|
||||
return devices
|
||||
}
|
||||
|
||||
// GetDevice 获取指定设备
|
||||
func (s *MemoryStorage) GetDevice(id string) (Device, bool) {
|
||||
s.mutex.RLock()
|
||||
defer s.mutex.RUnlock()
|
||||
|
||||
device, ok := s.devices[id]
|
||||
return device, ok
|
||||
}
|
||||
|
||||
// AddDevice 添加设备
|
||||
func (s *MemoryStorage) AddDevice(device Device) error {
|
||||
s.mutex.Lock()
|
||||
defer s.mutex.Unlock()
|
||||
|
||||
// 检查设备是否已存在
|
||||
if _, ok := s.devices[device.ID]; ok {
|
||||
return nil // 设备已存在,不重复添加
|
||||
}
|
||||
|
||||
s.devices[device.ID] = device
|
||||
return nil
|
||||
}
|
||||
|
||||
// UpdateDevice 更新设备
|
||||
func (s *MemoryStorage) UpdateDevice(device Device) error {
|
||||
s.mutex.Lock()
|
||||
defer s.mutex.Unlock()
|
||||
|
||||
// 检查设备是否存在
|
||||
if _, ok := s.devices[device.ID]; !ok {
|
||||
return nil // 设备不存在,不更新
|
||||
}
|
||||
|
||||
s.devices[device.ID] = device
|
||||
return nil
|
||||
}
|
||||
|
||||
// DeleteDevice 删除设备
|
||||
func (s *MemoryStorage) DeleteDevice(id string) error {
|
||||
s.mutex.Lock()
|
||||
defer s.mutex.Unlock()
|
||||
|
||||
delete(s.devices, id)
|
||||
return nil
|
||||
}
|
||||
|
||||
// GetDeviceByToken 通过令牌获取设备
|
||||
func (s *MemoryStorage) GetDeviceByToken(token string) (Device, bool) {
|
||||
s.mutex.RLock()
|
||||
defer s.mutex.RUnlock()
|
||||
|
||||
for _, device := range s.devices {
|
||||
if device.Token == token {
|
||||
return device, true
|
||||
}
|
||||
}
|
||||
|
||||
return Device{}, false
|
||||
}
|
||||
|
||||
// UpdateDeviceStatus 更新设备状态
|
||||
func (s *MemoryStorage) UpdateDeviceStatus(id string, status string) error {
|
||||
s.mutex.Lock()
|
||||
defer s.mutex.Unlock()
|
||||
|
||||
device, ok := s.devices[id]
|
||||
if !ok {
|
||||
return nil // 设备不存在,不更新
|
||||
}
|
||||
|
||||
device.Status = status
|
||||
s.devices[id] = device
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user