1. 增加程序启动时自动迁移表

2. 增加设备信息model和基础操作函数
This commit is contained in:
2025-09-08 10:58:26 +08:00
parent 0895cf42b2
commit bab44e7e3a
6 changed files with 168 additions and 10 deletions

View File

@@ -7,6 +7,27 @@ import (
"gorm.io/gorm"
)
// DeviceRepo 设备仓库接口
type DeviceRepo interface {
// Create 创建设备
Create(device *model.Device) error
// FindByID 根据ID查找设备
FindByID(id uint) (*model.Device, error)
// FindByParentID 根据上级设备ID查找设备
FindByParentID(parentID uint) ([]*model.Device, error)
// FindByType 根据设备类型查找设备
FindByType(deviceType model.DeviceType) ([]*model.Device, error)
// Update 更新设备信息
Update(device *model.Device) error
// Delete 删除设备
Delete(id uint) error
}
// DeviceControlRepo 设备控制仓库接口
type DeviceControlRepo interface {
// Create 创建设备控制记录
@@ -22,11 +43,23 @@ type DeviceControlRepo interface {
List(offset, limit int) ([]*model.DeviceControl, error)
}
// deviceRepo 设备仓库实现
type deviceRepo struct {
db *gorm.DB
}
// deviceControlRepo 设备控制仓库实现
type deviceControlRepo struct {
db *gorm.DB
}
// NewDeviceRepo 创建设备仓库实例
func NewDeviceRepo(db *gorm.DB) DeviceRepo {
return &deviceRepo{
db: db,
}
}
// NewDeviceControlRepo 创建设备控制仓库实例
func NewDeviceControlRepo(db *gorm.DB) DeviceControlRepo {
return &deviceControlRepo{
@@ -34,6 +67,54 @@ func NewDeviceControlRepo(db *gorm.DB) DeviceControlRepo {
}
}
// Create 创建设备
func (r *deviceRepo) Create(device *model.Device) error {
result := r.db.Create(device)
return result.Error
}
// FindByID 根据ID查找设备
func (r *deviceRepo) FindByID(id uint) (*model.Device, error) {
var device model.Device
result := r.db.First(&device, id)
if result.Error != nil {
return nil, result.Error
}
return &device, nil
}
// FindByParentID 根据上级设备ID查找设备
func (r *deviceRepo) FindByParentID(parentID uint) ([]*model.Device, error) {
var devices []*model.Device
result := r.db.Where("parent_id = ?", parentID).Find(&devices)
if result.Error != nil {
return nil, result.Error
}
return devices, nil
}
// FindByType 根据设备类型查找设备
func (r *deviceRepo) FindByType(deviceType model.DeviceType) ([]*model.Device, error) {
var devices []*model.Device
result := r.db.Where("type = ?", deviceType).Find(&devices)
if result.Error != nil {
return nil, result.Error
}
return devices, nil
}
// Update 更新设备信息
func (r *deviceRepo) Update(device *model.Device) error {
result := r.db.Save(device)
return result.Error
}
// Delete 删除设备
func (r *deviceRepo) Delete(id uint) error {
result := r.db.Delete(&model.Device{}, id)
return result.Error
}
// Create 创建设备控制记录
func (r *deviceControlRepo) Create(control *model.DeviceControl) error {
result := r.db.Create(control)