102 lines
4.0 KiB
Go
102 lines
4.0 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
|
|
"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"
|
|
"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// DeviceTemplateRepository 定义了设备模板数据访问的接口
|
|
type DeviceTemplateRepository interface {
|
|
Create(ctx context.Context, deviceTemplate *models.DeviceTemplate) error
|
|
FindByID(ctx context.Context, id uint) (*models.DeviceTemplate, error)
|
|
FindByName(ctx context.Context, name string) (*models.DeviceTemplate, error)
|
|
ListAll(ctx context.Context) ([]*models.DeviceTemplate, error)
|
|
Update(ctx context.Context, deviceTemplate *models.DeviceTemplate) error
|
|
Delete(ctx context.Context, id uint) error
|
|
IsInUse(ctx context.Context, id uint) (bool, error)
|
|
}
|
|
|
|
// gormDeviceTemplateRepository 是 DeviceTemplateRepository 的 GORM 实现
|
|
type gormDeviceTemplateRepository struct {
|
|
ctx context.Context
|
|
db *gorm.DB
|
|
}
|
|
|
|
// NewGormDeviceTemplateRepository 创建一个新的 gormDeviceTemplateRepository 实例
|
|
func NewGormDeviceTemplateRepository(ctx context.Context, db *gorm.DB) DeviceTemplateRepository {
|
|
return &gormDeviceTemplateRepository{ctx: ctx, db: db}
|
|
}
|
|
|
|
// Create 在数据库中创建一个新的设备模板
|
|
func (r *gormDeviceTemplateRepository) Create(ctx context.Context, deviceTemplate *models.DeviceTemplate) error {
|
|
repoCtx := logs.AddFuncName(ctx, r.ctx, "Create")
|
|
return r.db.WithContext(repoCtx).Create(deviceTemplate).Error
|
|
}
|
|
|
|
// FindByID 根据ID查找设备模板
|
|
func (r *gormDeviceTemplateRepository) FindByID(ctx context.Context, id uint) (*models.DeviceTemplate, error) {
|
|
repoCtx := logs.AddFuncName(ctx, r.ctx, "FindByID")
|
|
var deviceTemplate models.DeviceTemplate
|
|
if err := r.db.WithContext(repoCtx).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(ctx context.Context, name string) (*models.DeviceTemplate, error) {
|
|
repoCtx := logs.AddFuncName(ctx, r.ctx, "FindByName")
|
|
var deviceTemplate models.DeviceTemplate
|
|
if err := r.db.WithContext(repoCtx).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(ctx context.Context) ([]*models.DeviceTemplate, error) {
|
|
repoCtx := logs.AddFuncName(ctx, r.ctx, "ListAll")
|
|
var deviceTemplates []*models.DeviceTemplate
|
|
if err := r.db.WithContext(repoCtx).Find(&deviceTemplates).Error; err != nil {
|
|
return nil, fmt.Errorf("获取设备模板列表失败: %w", err)
|
|
}
|
|
return deviceTemplates, nil
|
|
}
|
|
|
|
// Update 更新数据库中的设备模板
|
|
func (r *gormDeviceTemplateRepository) Update(ctx context.Context, deviceTemplate *models.DeviceTemplate) error {
|
|
repoCtx := logs.AddFuncName(ctx, r.ctx, "Update")
|
|
return r.db.WithContext(repoCtx).Save(deviceTemplate).Error
|
|
}
|
|
|
|
// IsInUse 检查设备模板是否正在被设备使用
|
|
func (r *gormDeviceTemplateRepository) IsInUse(ctx context.Context, id uint) (bool, error) {
|
|
repoCtx := logs.AddFuncName(ctx, r.ctx, "IsInUse")
|
|
var count int64
|
|
if err := r.db.WithContext(repoCtx).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(ctx context.Context, id uint) error {
|
|
repoCtx := logs.AddFuncName(ctx, r.ctx, "Delete")
|
|
if err := r.db.WithContext(repoCtx).Delete(&models.DeviceTemplate{}, id).Error; err != nil {
|
|
return fmt.Errorf("删除设备模板失败: %w", err)
|
|
}
|
|
return nil
|
|
}
|