95 lines
3.1 KiB
Go
95 lines
3.1 KiB
Go
package repository
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// DeviceTemplateRepository 定义了设备模板数据访问的接口
|
|
type DeviceTemplateRepository interface {
|
|
Create(deviceTemplate *models.DeviceTemplate) error
|
|
FindByID(id uint) (*models.DeviceTemplate, error)
|
|
FindByName(name string) (*models.DeviceTemplate, error)
|
|
ListAll() ([]*models.DeviceTemplate, error)
|
|
Update(deviceTemplate *models.DeviceTemplate) error
|
|
Delete(id uint) error
|
|
IsInUse(id uint) (bool, error)
|
|
}
|
|
|
|
// gormDeviceTemplateRepository 是 DeviceTemplateRepository 的 GORM 实现
|
|
type gormDeviceTemplateRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
// NewGormDeviceTemplateRepository 创建一个新的 gormDeviceTemplateRepository 实例
|
|
func NewGormDeviceTemplateRepository(db *gorm.DB) DeviceTemplateRepository {
|
|
return &gormDeviceTemplateRepository{db: db}
|
|
}
|
|
|
|
// Create 在数据库中创建一个新的设备模板
|
|
func (r *gormDeviceTemplateRepository) Create(deviceTemplate *models.DeviceTemplate) error {
|
|
return r.db.Create(deviceTemplate).Error
|
|
}
|
|
|
|
// FindByID 根据ID查找设备模板
|
|
func (r *gormDeviceTemplateRepository) FindByID(id uint) (*models.DeviceTemplate, error) {
|
|
var deviceTemplate models.DeviceTemplate
|
|
if err := r.db.First(&deviceTemplate, id).Error; err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return nil, fmt.Errorf("设备模板未找到: %w", err)
|
|
}
|
|
return nil, fmt.Errorf("查询设备模板失败: %w", err)
|
|
}
|
|
return &deviceTemplate, nil
|
|
}
|
|
|
|
// FindByName 根据名称查找设备模板
|
|
func (r *gormDeviceTemplateRepository) FindByName(name string) (*models.DeviceTemplate, error) {
|
|
var deviceTemplate models.DeviceTemplate
|
|
if err := r.db.Where("name = ?", name).First(&deviceTemplate).Error; err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return nil, fmt.Errorf("设备模板未找到: %w", err)
|
|
}
|
|
return nil, fmt.Errorf("查询设备模板失败: %w", err)
|
|
}
|
|
return &deviceTemplate, nil
|
|
}
|
|
|
|
// ListAll 获取所有设备模板
|
|
func (r *gormDeviceTemplateRepository) ListAll() ([]*models.DeviceTemplate, error) {
|
|
var deviceTemplates []*models.DeviceTemplate
|
|
if err := r.db.Find(&deviceTemplates).Error; err != nil {
|
|
return nil, fmt.Errorf("获取设备模板列表失败: %w", err)
|
|
}
|
|
return deviceTemplates, nil
|
|
}
|
|
|
|
// Update 更新数据库中的设备模板
|
|
func (r *gormDeviceTemplateRepository) Update(deviceTemplate *models.DeviceTemplate) error {
|
|
return r.db.Save(deviceTemplate).Error
|
|
}
|
|
|
|
// IsInUse 检查设备模板是否正在被设备使用
|
|
func (r *gormDeviceTemplateRepository) IsInUse(id uint) (bool, error) {
|
|
var count int64
|
|
if err := r.db.Model(&models.Device{}).Where("device_template_id = ?", id).Count(&count).Error; err != nil {
|
|
return false, fmt.Errorf("检查设备模板使用情况失败: %w", err)
|
|
}
|
|
return count > 0, nil
|
|
}
|
|
|
|
// Delete 软删除数据库中的设备模板
|
|
func (r *gormDeviceTemplateRepository) Delete(id uint) error {
|
|
inUse, err := r.IsInUse(id)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if inUse {
|
|
return errors.New("设备模板正在被设备使用，无法删除")
|
|
}
|
|
return r.db.Delete(&models.DeviceTemplate{}, id).Error
|
|
}
|