89 lines
2.9 KiB
Go
89 lines
2.9 KiB
Go
package repository
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// AreaControllerRepository 定义了对 AreaController 模型的数据库操作接口
|
|
type AreaControllerRepository interface {
|
|
FindByID(id uint) (*models.AreaController, error)
|
|
FindByNetworkID(networkID string) (*models.AreaController, error)
|
|
Create(ac *models.AreaController) error
|
|
ListAll() ([]*models.AreaController, error)
|
|
Update(ac *models.AreaController) error
|
|
Delete(id uint) error
|
|
}
|
|
|
|
// gormAreaControllerRepository 是 AreaControllerRepository 的 GORM 实现。
|
|
type gormAreaControllerRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
// NewGormAreaControllerRepository 创建一个新的 AreaControllerRepository GORM 实现实例。
|
|
func NewGormAreaControllerRepository(db *gorm.DB) AreaControllerRepository {
|
|
return &gormAreaControllerRepository{db: db}
|
|
}
|
|
|
|
// Create 创建一个新的 AreaController 记录。
|
|
func (r *gormAreaControllerRepository) Create(ac *models.AreaController) error {
|
|
return r.db.Create(ac).Error
|
|
}
|
|
|
|
// ListAll 返回所有 AreaController 的列表。
|
|
func (r *gormAreaControllerRepository) ListAll() ([]*models.AreaController, error) {
|
|
var areaControllers []*models.AreaController
|
|
if err := r.db.Find(&areaControllers).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return areaControllers, nil
|
|
}
|
|
|
|
// Update 更新一个已存在的 AreaController 记录。
|
|
func (r *gormAreaControllerRepository) Update(ac *models.AreaController) error {
|
|
return r.db.Save(ac).Error
|
|
}
|
|
|
|
// Delete 删除一个 AreaController 记录。
|
|
// 在删除前会检查是否有设备关联到该主控，如果有，则不允许删除。
|
|
func (r *gormAreaControllerRepository) Delete(id uint) error {
|
|
return r.db.Transaction(func(tx *gorm.DB) error {
|
|
// 检查是否有设备关联到这个区域主控
|
|
var count int64
|
|
if err := tx.Model(&models.Device{}).Where("area_controller_id = ?", id).Count(&count).Error; err != nil {
|
|
return fmt.Errorf("检查关联设备失败: %w", err)
|
|
}
|
|
|
|
if count > 0 {
|
|
return fmt.Errorf("无法删除区域主控，因为仍有 %d 个设备关联到它", count)
|
|
}
|
|
|
|
// 如果没有关联设备，则执行删除操作
|
|
if err := tx.Delete(&models.AreaController{}, id).Error; err != nil {
|
|
return fmt.Errorf("删除区域主控失败: %w", err)
|
|
}
|
|
|
|
return nil
|
|
})
|
|
}
|
|
|
|
// FindByID 通过 ID 查找一个 AreaController。
|
|
func (r *gormAreaControllerRepository) FindByID(id uint) (*models.AreaController, error) {
|
|
var areaController models.AreaController
|
|
if err := r.db.First(&areaController, id).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &areaController, nil
|
|
}
|
|
|
|
// FindByNetworkID 通过 NetworkID 查找一个 AreaController。
|
|
func (r *gormAreaControllerRepository) FindByNetworkID(networkID string) (*models.AreaController, error) {
|
|
var areaController models.AreaController
|
|
if err := r.db.Where("network_id = ?", networkID).First(&areaController).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &areaController, nil
|
|
}
|