/
alarm.go
69 lines (50 loc) · 1.54 KB
/
alarm.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
package types
import (
"context"
"github.com/Juniper/asf/pkg/errutil"
"github.com/Juniper/contrail/pkg/models"
"github.com/Juniper/contrail/pkg/services"
asfmodels "github.com/Juniper/asf/pkg/models"
)
//CreateAlarm does pre check for alarm.
func (sv *ContrailTypeLogicService) CreateAlarm(
ctx context.Context,
request *services.CreateAlarmRequest) (response *services.CreateAlarmResponse, err error) {
alarm := request.GetAlarm()
err = sv.InTransactionDoer.DoInTransaction(
ctx,
func(ctx context.Context) error {
// check if alarm rules are valid
if err = alarm.Validate(); err != nil {
return errutil.ErrorBadRequestf(err.Error())
}
if response, err = sv.BaseService.CreateAlarm(ctx, request); err != nil {
return err
}
return nil
})
return response, err
}
//UpdateAlarm does pre check for alarm.
func (sv *ContrailTypeLogicService) UpdateAlarm(
ctx context.Context,
request *services.UpdateAlarmRequest) (response *services.UpdateAlarmResponse, err error) {
alarm := request.GetAlarm()
fm := request.GetFieldMask()
err = sv.InTransactionDoer.DoInTransaction(
ctx,
func(ctx context.Context) error {
// check if alarm rules was present in JSON
if asfmodels.FieldMaskContains(&fm, models.AlarmFieldAlarmRules) {
// check if alarm rules are valid
if err = alarm.Validate(); err != nil {
return errutil.ErrorBadRequestf(err.Error())
}
}
if response, err = sv.BaseService.UpdateAlarm(ctx, request); err != nil {
return err
}
return nil
})
return response, err
}