You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
74 lines
2.2 KiB
74 lines
2.2 KiB
package monitor |
|
|
|
import ( |
|
"context" |
|
"encoding/json" |
|
"github.com/pkg/errors" |
|
"go-common/app/job/main/aegis/model/monitor" |
|
xsql "go-common/library/database/sql" |
|
"go-common/library/log" |
|
"time" |
|
) |
|
|
|
const ( |
|
_rulesByBid = "SELECT id,type,bid,name,state,stime,etime,rule,uid,ctime,mtime FROM monitor_rule WHERE bid = ? AND state = 1 AND stime < ? AND etime > ?" |
|
_allValidRules = "SELECT id,type,bid,name,state,stime,etime,rule,uid,ctime,mtime FROM monitor_rule WHERE state = 1 AND stime < ? AND etime > ?" |
|
) |
|
|
|
// RulesByBid 获取某业务的监控 |
|
func (d *Dao) RulesByBid(c context.Context, bid int64) (rules []*monitor.Rule, err error) { |
|
var ( |
|
rows *xsql.Rows |
|
now = time.Now() |
|
) |
|
if rows, err = d.db.Query(c, _rulesByBid, bid, now, now); err != nil { |
|
log.Error("d.db.Exec error(%v)", errors.WithStack(err)) |
|
return |
|
} |
|
defer rows.Close() |
|
for rows.Next() { |
|
rule := &monitor.Rule{} |
|
var confStr string |
|
if err = rows.Scan(&rule.ID, &rule.Type, &rule.BID, &rule.Name, &rule.State, &rule.STime, &rule.ETime, &confStr, &rule.UID, &rule.CTime, &rule.MTime); err != nil { |
|
log.Error("rows.Scan error(%v)", err) |
|
return |
|
} |
|
conf := &monitor.RuleConf{} |
|
if err = json.Unmarshal([]byte(confStr), conf); err != nil { |
|
log.Error("json.Unmarshal(%s) error(%v)", confStr, err) |
|
return |
|
} |
|
rule.RuleConf = conf |
|
rules = append(rules, rule) |
|
} |
|
return |
|
} |
|
|
|
// ValidRules 获取有效的监控 |
|
func (d *Dao) ValidRules(c context.Context) (rules []*monitor.Rule, err error) { |
|
var ( |
|
rows *xsql.Rows |
|
now = time.Now() |
|
) |
|
if rows, err = d.db.Query(c, _allValidRules, now, now); err != nil { |
|
log.Error("d.db.Exec error(%v)", errors.WithStack(err)) |
|
return |
|
} |
|
defer rows.Close() |
|
for rows.Next() { |
|
rule := &monitor.Rule{} |
|
var confStr string |
|
if err = rows.Scan(&rule.ID, &rule.Type, &rule.BID, &rule.Name, &rule.State, &rule.STime, &rule.ETime, &confStr, &rule.UID, &rule.CTime, &rule.MTime); err != nil { |
|
log.Error("rows.Scan error(%v)", err) |
|
return |
|
} |
|
conf := &monitor.RuleConf{} |
|
if err = json.Unmarshal([]byte(confStr), conf); err != nil { |
|
log.Error("json.Unmarshal(%s) error(%v)", confStr, err) |
|
return |
|
} |
|
rule.RuleConf = conf |
|
rules = append(rules, rule) |
|
} |
|
return |
|
}
|
|
|