nightingale/src/model/stra.go

package model

import (
	"fmt"
	"strconv"
	"strings"
	"time"

	"xorm.io/xorm"
)

type Stra struct {
	Id                  int64     `json:"id"`
	Name                string    `json:"name"`
	Category            int       `json:"category"`                     //机器，非机器
	Nid                 int64     `json:"nid"`                          //服务树节点id
	ExclNidStr          string    `xorm:"excl_nid" json:"-"`            //排除的叶子节点
	AlertDur            int       `json:"alert_dur"`                    //单位秒，持续异常10分钟则产生异常event
	RecoveryDur         int       `json:"recovery_dur"`                 //单位秒，持续正常2分钟则产生恢复event，0表示立即产生恢复event
	RecoveryNotify      int       `json:"recovery_notify"`              //0 发送恢复通知 1不发送恢复通知
	ExprsStr            string    `xorm:"exprs" json:"-"`               //多个条件的监控实例需要相同，并且同时满足才产生event
	TagsStr             string    `xorm:"tags" json:"-"`                //tag过滤条件
	EnableStime         string    `json:"enable_stime"`                 //策略生效开始时间
	EnableEtime         string    `json:"enable_etime"`                 //策略生效终止时间 支持23:00-02:00
	EnableDaysOfWeekStr string    `xorm:"enable_days_of_week" json:"-"` //策略生效日期
	ConvergeStr         string    `xorm:"converge" json:"-"`            //告警通知收敛，第1个值表示收敛周期，单位秒，第2个值表示周期内允许发送告警次数
	Priority            int       `json:"priority"`
	Callback            string    `json:"callback"`
	NotifyGroupStr      string    `xorm:"notify_group" json:"-"`
	NotifyUserStr       string    `xorm:"notify_user" json:"-"`
	Creator             string    `json:"creator"`
	Created             time.Time `xorm:"created" json:"created"`
	LastUpdator         string    `xorm:"last_updator" json:"last_updator"`
	LastUpdated         time.Time `xorm:"<-" json:"last_updated"`
	NeedUpgrade         int       `xorm:"need_upgrade" json:"need_upgrade"`
	AlertUpgradeStr     string    `xorm:"alert_upgrade" json:"-"`

	ExclNid          []int64      `xorm:"-" json:"excl_nid"`
	Exprs            []Exp        `xorm:"-" json:"exprs"`
	Tags             []Tag        `xorm:"-" json:"tags"`
	EnableDaysOfWeek []int        `xorm:"-" json:"enable_days_of_week"`
	Converge         []int        `xorm:"-" json:"converge"`
	NotifyGroup      []int        `xorm:"-" json:"notify_group"`
	NotifyUser       []int        `xorm:"-" json:"notify_user"`
	LeafNids         []int64      `xorm:"-" json:"leaf_nids"` //叶子节点id
	Endpoints        []string     `xorm:"-" json:"endpoints"`
	AlertUpgrade     AlertUpgrade `xorm:"-" json:"alert_upgrade"`
	JudgeInstance    string       `xorm:"-" json:"judge_instance"`
}

type StraLog struct {
	Id      int64     `json:"id"`
	Sid     int64     `json:"sid"`
	Action  string    `json:"action"` // update|delete
	Body    string    `json:"body"`
	Creator string    `json:"creator"`
	Created time.Time `json:"created" xorm:"created"`
}

type Exp struct {
	Eopt      string  `json:"eopt"`
	Func      string  `json:"func"`      //all,max,min
	Metric    string  `json:"metric"`    //metric
	Params    []int   `json:"params"`    //连续n秒
	Threshold float64 `json:"threshold"` //阈值
}

type Tag struct {
	Tkey string   `json:"tkey"`
	Topt string   `json:"topt"`
	Tval []string `json:"tval"` //修改为数组
}

type AlertUpgrade struct {
	Users    []int64 `json:"users"`
	Groups   []int64 `json:"groups"`
	Duration int     `json:"duration"`
	Level    int     `json:"level"`
}

var MathOperators = map[string]bool{
	">":  true,
	"<":  true,
	">=": true,
	"<=": true,
	"!=": true,
	"=":  true,
}

func (s *Stra) Save() error {
	session := DB["mon"].NewSession()
	defer session.Close()

	err := session.Begin()
	if err != nil {
		session.Rollback()
		return err
	}

	_, err = session.Insert(s)
	if err != nil {
		session.Rollback()
		return err
	}

	straByte, err := json.Marshal(s)
	if err != nil {
		session.Rollback()
		return err
	}

	err = SaveStraCommit(s.Id, "add", s.Creator, string(straByte), session)
	if err != nil {
		session.Rollback()
		return err
	}

	session.Commit()
	return nil
}

func (s *Stra) Update() error {
	var obj Stra

	session := DB["mon"].NewSession()
	defer session.Close()

	err := session.Begin()
	if err != nil {
		session.Rollback()
		return err
	}

	exists, err := session.ID(s.Id).Get(&obj)
	if err != nil {
		session.Rollback()
		return err
	}

	if !exists {
		session.Rollback()
		return fmt.Errorf("%d not exists", s.Id)
	}

	_, err = session.ID(s.Id).AllCols().Update(s)
	if err != nil {
		session.Rollback()
		return err
	}

	straByte, err := json.Marshal(s)
	if err != nil {
		session.Rollback()
		return err
	}

	err = SaveStraCommit(s.Id, "update", s.Creator, string(straByte), session)
	if err != nil {
		session.Rollback()
		return err
	}

	session.Commit()
	return nil
}

func StraGet(col string, val interface{}) (*Stra, error) {
	var obj Stra
	has, err := DB["mon"].Where(col+"=?", val).Get(&obj)
	if err != nil {
		return nil, err
	}

	if !has {
		return nil, nil
	}

	return &obj, nil
}

func StraDel(id int64) error {
	session := DB["mon"].NewSession()
	defer session.Close()
	var obj Stra

	if err := session.Begin(); err != nil {
		return err
	}

	exists, err := session.ID(id).Get(&obj)
	if err != nil {
		session.Rollback()
		return err
	}

	if !exists {
		session.Rollback()
		return fmt.Errorf("%d not exists", obj.Id)
	}

	if _, err := session.ID(id).Delete(new(Stra)); err != nil {
		session.Rollback()
		return err
	}

	straByte, err := json.Marshal(obj)
	if err != nil {
		session.Rollback()
		return err
	}

	err = SaveStraCommit(obj.Id, "delete", obj.Creator, string(straByte), session)
	if err != nil {
		session.Rollback()
		return err
	}

	return session.Commit()
}

func StrasList(name string, priority int, nid int64) ([]*Stra, error) {
	session := DB["mon"].NewSession()
	defer session.Close()

	objs := make([]*Stra, 0)

	whereClause := "1 = 1"
	params := []interface{}{}

	if name != "" {
		whereClause += " AND name LIKE ?"
		params = append(params, "%"+name+"%")
	}

	if priority <= 3 {
		whereClause += " AND priority = ?"
		params = append(params, priority)
	}

	var err error
	if nid != 0 {
		err = session.Where(whereClause, params...).Where("nid=?", nid).Find(&objs)
	} else {
		err = session.Where(whereClause, params...).Find(&objs)
	}
	if err != nil {
		return objs, err
	}

	stras := make([]*Stra, 0)
	for _, obj := range objs {
		err = obj.Decode()
		if err != nil {
			return stras, err
		}
		stras = append(stras, obj)
	}
	return stras, err
}

func StrasAll() ([]*Stra, error) {
	objs := make([]*Stra, 0)

	err := DB["mon"].Find(&objs)
	if err != nil {
		return objs, err
	}

	stras := make([]*Stra, 0)
	for _, obj := range objs {
		err = obj.Decode()
		if err != nil {
			return stras, err
		}
		stras = append(stras, obj)
	}
	return stras, err
}

func EffectiveStrasList() ([]*Stra, error) {
	session := DB["mon"].NewSession()
	defer session.Close()

	objs := make([]*Stra, 0)
	t := time.Now()

	now := t.Format("15:04")
	weekday := strconv.Itoa(int(t.Weekday()))

	err := session.Where("((enable_stime <= ? and enable_etime >= ? or (enable_stime > enable_etime and !(enable_stime > ? and enable_etime < ?))) and enable_days_of_week like ?)", now, now, now, now, "%"+weekday+"%").Find(&objs)
	if err != nil {
		return objs, err
	}

	stras := make([]*Stra, 0)
	for _, obj := range objs {
		err = obj.Decode()
		if err != nil {
			return stras, err
		}
		stras = append(stras, obj)
	}
	return stras, err
}

func SaveStraCommit(id int64, action, username, body string, session *xorm.Session) error {
	strategyLog := StraLog{
		Sid:     id,
		Action:  action,
		Body:    body,
		Creator: username,
	}

	if _, err := session.Insert(&strategyLog); err != nil {
		session.Rollback()
		return err
	}

	return nil
}

func (s *Stra) HasPermssion() error {
	return nil
}

func (s *Stra) Encode() error {
	alertUpgrade, err := AlertUpgradeMarshal(s.AlertUpgrade)
	if err != nil {
		return fmt.Errorf("encode alert_upgrade err:%v", err)
	}

	if s.NeedUpgrade == 1 {
		if len(s.AlertUpgrade.Users) == 0 && len(s.AlertUpgrade.Groups) == 0 {
			return fmt.Errorf("alert upgrade: users and groups is blank")
		}
	}

	s.AlertUpgradeStr = alertUpgrade

	exclNid, err := json.Marshal(s.ExclNid)
	if err != nil {
		return fmt.Errorf("encode excl_nid err:%v", err)
	}
	s.ExclNidStr = string(exclNid)

	exprs, err := json.Marshal(s.Exprs)
	if err != nil {
		return fmt.Errorf("encode exprs err:%v", err)
	}
	s.ExprsStr = string(exprs)

	// 校验 exprs
	var exprsTmp []Exp
	err = json.Unmarshal(exprs, &exprsTmp)
	if err != nil {
		return fmt.Errorf("unmarshal exprs err:%v", err)
	}

	for _, exp := range exprsTmp {
		if _, found := MathOperators[exp.Eopt]; !found {
			return fmt.Errorf("unknown exp.eopt:%v", exp)
		}
	}

	tags, err := json.Marshal(s.Tags)
	if err != nil {
		return fmt.Errorf("encode Tags err:%v", err)
	}
	s.TagsStr = string(tags)

	// 校验 tags
	var tagsTmp []Tag
	err = json.Unmarshal(tags, &tagsTmp)
	if err != nil {
		return fmt.Errorf("unmarshal Tags err:%v", err)
	}

	for _, tag := range tagsTmp {
		if tag.Topt != "=" && tag.Topt != "!=" {
			return fmt.Errorf("unknown tag.topt")
		}
	}

	//校验时间
	err = checkDurationString(s.EnableStime)
	if err != nil {
		return fmt.Errorf("unknown enable_stime: %s", s.EnableStime)
	}

	err = checkDurationString(s.EnableEtime)
	if err != nil {
		return fmt.Errorf("unknown enable_etime: %s", s.EnableEtime)
	}

	for _, day := range s.EnableDaysOfWeek {
		if day > 7 || day < 0 {
			return fmt.Errorf("illegal period_days_of_week %v", s.EnableDaysOfWeek)
		}
	}
	enableDaysOfWeek, err := json.Marshal(s.EnableDaysOfWeek)
	if err != nil {
		return fmt.Errorf("encode EnableDaysOfWeek err:%v", err)
	}
	s.EnableDaysOfWeekStr = string(enableDaysOfWeek)

	//校验收敛配置
	if len(s.Converge) != 2 {
		return fmt.Errorf("illegal converge %v", s.Converge)
	}
	Converge, err := json.Marshal(s.Converge)
	if err != nil {
		return err
	}
	s.ConvergeStr = string(Converge)

	notifyGroup, err := json.Marshal(s.NotifyGroup)
	if err != nil {
		return err
	}
	s.NotifyGroupStr = string(notifyGroup)

	notifyUser, err := json.Marshal(s.NotifyUser)
	if err != nil {
		return err
	}
	s.NotifyUserStr = string(notifyUser)

	return nil
}

func (s *Stra) Decode() error {
	var err error

	s.AlertUpgrade, err = AlertUpgradeUnMarshal(s.AlertUpgradeStr)
	if err != nil {
		return err
	}

	err = json.Unmarshal([]byte(s.ExclNidStr), &s.ExclNid)
	if err != nil {
		return err
	}

	err = json.Unmarshal([]byte(s.ExprsStr), &s.Exprs)
	if err != nil {
		return err
	}

	err = json.Unmarshal([]byte(s.TagsStr), &s.Tags)
	if err != nil {
		return err
	}

	err = json.Unmarshal([]byte(s.EnableDaysOfWeekStr), &s.EnableDaysOfWeek)
	if err != nil {
		return err
	}

	err = json.Unmarshal([]byte(s.ConvergeStr), &s.Converge)
	if err != nil {
		return err
	}

	err = json.Unmarshal([]byte(s.NotifyUserStr), &s.NotifyUser)
	if err != nil {
		return err
	}

	err = json.Unmarshal([]byte(s.NotifyGroupStr), &s.NotifyGroup)
	if err != nil {
		return err
	}

	return nil
}

// 00:00-23:59
func checkDurationString(str string) error {
	slice := strings.Split(str, ":")
	if len(slice) != 2 {
		return fmt.Errorf("illegal duration: %s", str)
	}

	hour, err := strconv.Atoi(slice[0])
	if err != nil {
		return fmt.Errorf("illegal duration: %s", str)
	}
	if hour < 0 || hour > 23 {
		return fmt.Errorf("illegal duration: %s", str)
	}
	minute, err := strconv.Atoi(slice[1])
	if err != nil {
		return fmt.Errorf("illegal duration: %s", str)
	}
	if minute < 0 || minute > 59 {
		return fmt.Errorf("illegal duration: %s", str)
	}

	return nil
}

func AlertUpgradeMarshal(alterUpgrade AlertUpgrade) (string, error) {
	dat := AlertUpgrade{
		Duration: alterUpgrade.Duration,
		Level:    alterUpgrade.Level,
	}

	if alterUpgrade.Duration == 0 {
		dat.Duration = 60
	}

	if alterUpgrade.Level == 0 {
		dat.Level = 1
	}

	if alterUpgrade.Groups == nil {
		dat.Groups = []int64{}
	} else {
		dat.Groups = alterUpgrade.Groups
	}

	if alterUpgrade.Users == nil {
		dat.Users = []int64{}
	} else {
		dat.Users = alterUpgrade.Users
	}

	data, err := json.Marshal(dat)
	return string(data), err
}

func AlertUpgradeUnMarshal(str string) (AlertUpgrade, error) {
	var obj AlertUpgrade
	if strings.TrimSpace(str) == "" {
		return AlertUpgrade{
			Users:    []int64{},
			Groups:   []int64{},
			Duration: 0,
			Level:    0,
		}, nil
	}

	err := json.Unmarshal([]byte(str), &obj)
	return obj, err
}