重构设备控制器

internal/app/service/area_controller_service.go

@@ -0,0 +1,140 @@
+package service
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+
+	"git.huangwc.com/pig/pig-farm-controller/internal/app/dto"
+	"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"
+	"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
+	"git.huangwc.com/pig/pig-farm-controller/internal/infra/repository"
+)
+
+// AreaControllerService 定义了应用层的区域主控服务接口。
+type AreaControllerService interface {
+	CreateAreaController(ctx context.Context, req *dto.CreateAreaControllerRequest) (*dto.AreaControllerResponse, error)
+	GetAreaController(ctx context.Context, id uint32) (*dto.AreaControllerResponse, error)
+	ListAreaControllers(ctx context.Context) ([]*dto.AreaControllerResponse, error)
+	UpdateAreaController(ctx context.Context, id uint32, req *dto.UpdateAreaControllerRequest) (*dto.AreaControllerResponse, error)
+	DeleteAreaController(ctx context.Context, id uint32) error
+}
+
+// areaControllerService 是 AreaControllerService 接口的具体实现。
+type areaControllerService struct {
+	ctx                   context.Context
+	areaControllerRepo    repository.AreaControllerRepository
+	thresholdAlarmService ThresholdAlarmService
+}
+
+// NewAreaControllerService 创建一个新的 AreaControllerService 实例。
+func NewAreaControllerService(
+	ctx context.Context,
+	areaControllerRepo repository.AreaControllerRepository,
+	thresholdAlarmService ThresholdAlarmService,
+) AreaControllerService {
+	return &areaControllerService{
+		ctx:                   ctx,
+		areaControllerRepo:    areaControllerRepo,
+		thresholdAlarmService: thresholdAlarmService,
+	}
+}
+
+func (s *areaControllerService) CreateAreaController(ctx context.Context, req *dto.CreateAreaControllerRequest) (*dto.AreaControllerResponse, error) {
+	serviceCtx := logs.AddFuncName(ctx, s.ctx, "CreateAreaController")
+	propertiesJSON, err := json.Marshal(req.Properties)
+	if err != nil {
+		return nil, err
+	}
+
+	ac := &models.AreaController{
+		Name:       req.Name,
+		NetworkID:  req.NetworkID,
+		Location:   req.Location,
+		Properties: propertiesJSON,
+	}
+
+	if err := ac.SelfCheck(); err != nil {
+		return nil, err
+	}
+
+	if err := s.areaControllerRepo.Create(serviceCtx, ac); err != nil {
+		return nil, err
+	}
+
+	return dto.NewAreaControllerResponse(ac)
+}
+
+func (s *areaControllerService) GetAreaController(ctx context.Context, id uint32) (*dto.AreaControllerResponse, error) {
+	serviceCtx := logs.AddFuncName(ctx, s.ctx, "GetAreaController")
+	ac, err := s.areaControllerRepo.FindByID(serviceCtx, id)
+	if err != nil {
+		return nil, err
+	}
+	return dto.NewAreaControllerResponse(ac)
+}
+
+func (s *areaControllerService) ListAreaControllers(ctx context.Context) ([]*dto.AreaControllerResponse, error) {
+	serviceCtx := logs.AddFuncName(ctx, s.ctx, "ListAreaControllers")
+	acs, err := s.areaControllerRepo.ListAll(serviceCtx)
+	if err != nil {
+		return nil, err
+	}
+	return dto.NewListAreaControllerResponse(acs)
+}
+
+func (s *areaControllerService) UpdateAreaController(ctx context.Context, id uint32, req *dto.UpdateAreaControllerRequest) (*dto.AreaControllerResponse, error) {
+	serviceCtx := logs.AddFuncName(ctx, s.ctx, "UpdateAreaController")
+	existingAC, err := s.areaControllerRepo.FindByID(serviceCtx, id)
+	if err != nil {
+		return nil, err
+	}
+
+	propertiesJSON, err := json.Marshal(req.Properties)
+	if err != nil {
+		return nil, err
+	}
+
+	existingAC.Name = req.Name
+	existingAC.NetworkID = req.NetworkID
+	existingAC.Location = req.Location
+	existingAC.Properties = propertiesJSON
+
+	if err := existingAC.SelfCheck(); err != nil {
+		return nil, err
+	}
+
+	if err := s.areaControllerRepo.Update(serviceCtx, existingAC); err != nil {
+		return nil, err
+	}
+
+	return dto.NewAreaControllerResponse(existingAC)
+}
+
+func (s *areaControllerService) DeleteAreaController(ctx context.Context, id uint32) error {
+	serviceCtx := logs.AddFuncName(ctx, s.ctx, "DeleteAreaController")
+
+	// 1. 检查是否存在
+	_, err := s.areaControllerRepo.FindByID(serviceCtx, id)
+	if err != nil {
+		return err // 如果未找到，gorm会返回 ErrRecordNotFound
+	}
+
+	// 2. 检查是否被使用（业务逻辑）
+	inUse, err := s.areaControllerRepo.IsAreaControllerUsedByTasks(serviceCtx, id, []models.TaskType{models.TaskTypeAreaCollectorThresholdCheck})
+	if err != nil {
+		return err // 返回数据库检查错误
+	}
+	if inUse {
+		return ErrAreaControllerInUse // 返回业务错误
+	}
+
+	// TODO 这个应该用事务处理
+	err = s.thresholdAlarmService.DeleteAreaThresholdAlarmByAreaControllerID(serviceCtx, id)
+	if err != nil {
+		return fmt.Errorf("删除区域阈值告警失败: %w", err)
+	}
+
+	// 3. 执行删除
+	return s.areaControllerRepo.Delete(serviceCtx, id)
+}