55 lines
1.5 KiB
Go
55 lines
1.5 KiB
Go
package alarm
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log/slog"
|
|
|
|
"bj_power_mes/constants"
|
|
"bj_power_mes/ent"
|
|
"bj_power_mes/ent/alarm"
|
|
"bj_power_mes/internal/eventbus"
|
|
)
|
|
|
|
// Service 报警服务，负责创建报警并发布事件。
|
|
type Service struct {
|
|
entClient *ent.Client
|
|
bus eventbus.Bus
|
|
}
|
|
|
|
// NewService 创建报警服务。
|
|
func NewService(entClient *ent.Client, bus eventbus.Bus) *Service {
|
|
return &Service{entClient: entClient, bus: bus}
|
|
}
|
|
|
|
// Raise 创建报警并发布 ALARM_RAISED 事件。
|
|
func (s *Service) Raise(ctx context.Context, alarmCode string, alarmMessage string,
|
|
level constants.AlarmLevel, equipmentID int, jobID int, source string) (*ent.Alarm, error) {
|
|
|
|
a, err := s.entClient.Alarm.Create().
|
|
SetAlarmCode(alarmCode).
|
|
SetAlarmMessage(alarmMessage).
|
|
SetLevel(level).
|
|
SetNillableEquipmentId(&equipmentID).
|
|
SetNillableJobId(&jobID).
|
|
SetSource(source).
|
|
Save(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("create alarm: %w", err)
|
|
}
|
|
|
|
evt := eventbus.NewAlarmRaisedEvent(a.ID, alarmCode, string(level), equipmentID, source)
|
|
if err := s.bus.Publish(ctx, evt); err != nil {
|
|
slog.Error("alarm: publish event failed", "alarmId", a.ID, "error", err)
|
|
}
|
|
|
|
return a, nil
|
|
}
|
|
|
|
// FindActive 检查是否已存在同编码的未处理报警（去重）。
|
|
func (s *Service) FindActive(ctx context.Context, alarmCode string) (*ent.Alarm, error) {
|
|
return s.entClient.Alarm.Query().
|
|
Where(alarm.AlarmCodeEQ(alarmCode), alarm.ResolvedEQ(false)).
|
|
First(ctx)
|
|
}
|