feat: persist notify cur number (#1013)

Co-authored-by: tanxiao <tanxiao@asiainfo.com>

docker/initsql/a-n9e.sql

@@ -377,6 +377,7 @@ CREATE TABLE `alert_cur_event` (
     `notify_channels` varchar(255) not null default '' comment 'split by space: sms voice email dingtalk wecom',
     `notify_groups` varchar(255) not null default '' comment 'split by space: 233 43',
     `notify_repeat_next` bigint not null default 0 comment 'next timestamp to notify, get repeat settings from rule',
+    `notify_cur_number` int not null default 0 comment '',
     `target_ident` varchar(191) not null default '' comment 'target ident, also in tags',
     `target_note` varchar(191) not null default '' comment 'target note',
     `trigger_time` bigint not null,
@@ -410,6 +411,7 @@ CREATE TABLE `alert_his_event` (
     `notify_recovered` tinyint(1) not null comment 'whether notify when recovery',
     `notify_channels` varchar(255) not null default '' comment 'split by space: sms voice email dingtalk wecom',
     `notify_groups` varchar(255) not null default '' comment 'split by space: 233 43',
+    `notify_cur_number` int not null default 0 comment '',
     `target_ident` varchar(191) not null default '' comment 'target ident, also in tags',
     `target_note` varchar(191) not null default '' comment 'target note',
     `trigger_time` bigint not null,

docker/initsql_for_postgres/a-n9e-for-Postgres.sql

@@ -404,6 +404,7 @@ CREATE TABLE alert_cur_event (
     notify_channels varchar(255) NOT NULL DEFAULT ''::character varying,
     notify_groups varchar(255) NOT NULL DEFAULT ''::character varying,
     notify_repeat_next int8 NOT NULL DEFAULT 0,
+    notify_cur_number int4 not null default 0,
     target_ident varchar(191) NOT NULL DEFAULT ''::character varying,
     target_note varchar(191) NOT NULL DEFAULT ''::character varying,
     trigger_time int8 NOT NULL,
@@ -454,6 +455,7 @@ CREATE TABLE alert_his_event (
     notify_recovered int2 NOT NULL,
     notify_channels varchar(255) NOT NULL DEFAULT ''::character varying,
     notify_groups varchar(255) NOT NULL DEFAULT ''::character varying,
+    notify_cur_number int4 not null default 0,
     target_ident varchar(191) NOT NULL DEFAULT ''::character varying,
     target_note varchar(191) NOT NULL DEFAULT ''::character varying,
     trigger_time int8 NOT NULL,

src/models/alert_cur_event.go

@@ -45,7 +45,7 @@ type AlertCurEvent struct {
 	NotifyUsersObj     []*User           `json:"notify_users_obj" gorm:"-"` // for notify.py
 	LastEvalTime       int64             `json:"last_eval_time" gorm:"-"`   // for notify.py 上次计算的时间
 	LastSentTime       int64             `json:"last_sent_time" gorm:"-"`   // 上次发送时间
-	NotifyCurNumber    int               `json:"notify_cur_number" gorm:"-"` // notify: current number
+	NotifyCurNumber    int               `json:"notify_cur_number"`         // notify: current number
 }
 
 func (e *AlertCurEvent) TableName() string {
@@ -179,6 +179,7 @@ func (e *AlertCurEvent) ToHis() *AlertHisEvent {
 		Tags:             e.Tags,
 		RecoverTime:      recoverTime,
 		LastEvalTime:     e.LastEvalTime,
+		NotifyCurNumber:  e.NotifyCurNumber,
 	}
 }
 

src/models/alert_his_event.go

@@ -38,6 +38,7 @@ type AlertHisEvent struct {
 	LastEvalTime       int64       `json:"last_eval_time"`
 	Tags               string      `json:"-"`
 	TagsJSON           []string    `json:"tags" gorm:"-"`
+	NotifyCurNumber    int         `json:"notify_cur_number"` // notify: current number
 }
 
 func (e *AlertHisEvent) TableName() string {

src/server/engine/effective.go

@@ -29,9 +29,5 @@ func isNoneffective(timestamp int64, alertRule *models.AlertRule) bool {
 
 	alertRule.EnableDaysOfWeek = strings.Replace(alertRule.EnableDaysOfWeek, "7", "0", 1)
 
-	if !strings.Contains(alertRule.EnableDaysOfWeek, triggerWeek) {
+	return !strings.Contains(alertRule.EnableDaysOfWeek, triggerWeek)
-		return true
-	}
-
-	return false
 }

src/server/engine/worker.go

@@ -365,6 +365,7 @@ func (r RuleEval) fireEvent(event *models.AlertCurEvent) {
 		if event.LastEvalTime > fired.LastSentTime+int64(r.rule.NotifyRepeatStep)*60 {
 			if r.rule.NotifyMaxNumber == 0 {
 				// 最大可以发送次数如果是0，表示不想限制最大发送次数，一直发即可
+				event.NotifyCurNumber = fired.NotifyCurNumber + 1
 				r.pushEventToQueue(event)
 			} else {
 				// 有最大发送次数的限制，就要看已经发了几次了，是否达到了最大发送次数