141 lines
4.5 KiB
Go
141 lines
4.5 KiB
Go
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)
|
||
}
|