123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169 |
- package pinglog
- import (
- "github.com/gogf/gf/v2/database/gdb"
- "github.com/gogf/gf/v2/frame/g"
- "github.com/gogf/gf/v2/os/glog"
- "github.com/gogf/gf/v2/os/gtime"
- "golang.org/x/net/context"
- "nodeMonitor/internal/consts"
- "nodeMonitor/internal/dao"
- "nodeMonitor/internal/model"
- "nodeMonitor/internal/model/do"
- "nodeMonitor/internal/service"
- )
- type (
- sPing struct{}
- )
- func init() {
- service.RegisterPing(New())
- }
- func New() *sPing {
- return &sPing{}
- }
- func (c *sPing) Create(ctx context.Context, ping model.PingSt, setverid int) error {
- return dao.Pinglog.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
- _, err := dao.Pinglog.Ctx(ctx).Data(do.Pinglog{
- Maxdelay: ping.MaxDelay,
- Mindelay: ping.MinDelay,
- Avgdelay: ping.AvgDelay,
- Losspk: ping.LossPk,
- Serverid: setverid,
- }).Insert()
- return err
- })
- }
- func (c *sPing) GetHyStatus(ctx context.Context, ping model.PingSt, serverid int) (bool, error) {
- pingConfig, err := service.PingConfig().Get(ctx)
- if err != nil {
- glog.Debug(ctx, err.Error())
- return false, err
- }
-
- var lastDownTime *gtime.Time
- err = dao.Pinglog.Ctx(ctx).
- Where("avgdelay = 0").
- Where("losspk > ?", pingConfig.NodeLoos).
- Where("maxdelay = 0").
- Where("serverid = ?", serverid).
- OrderDesc("create_at").
- Limit(1).
- Scan(&lastDownTime)
- if err != nil {
- glog.Debug(ctx, err.Error())
- return false, err
- }
-
- if lastDownTime == nil {
- return false, nil
- }
- return true, nil
- }
- func (c *sPing) GetStatus(ctx context.Context, serverid int) (consts.NodeStatus, error) {
-
- startTime, err := g.Cfg().Get(ctx, "node.startTime")
- if err != nil {
- glog.Debug(ctx, err.Error())
- return consts.Normal, err
- }
-
-
-
-
-
-
-
- pingConfig, err := service.PingConfig().Get(ctx)
- if err != nil {
- glog.Debug(ctx, err.Error())
- return consts.Normal, err
- }
-
- failCount, err := dao.Pinglog.Ctx(ctx).
- Where("create_at > DATE_SUB(NOW(), INTERVAL ? MINUTE)", startTime).
- Where("avgdelay = 0").
- Where("losspk > ?", pingConfig.NodeLoos).
- Where("maxdelay = 0").
- Where("serverid = ?", serverid).
- Count()
- if err != nil {
- glog.Debug(ctx, err.Error())
- return consts.Normal, err
- }
-
- if failCount >= pingConfig.NodeDie {
-
- return consts.Down, nil
- }
-
- var lastDownTime *gtime.Time
- err = dao.Pinglog.Ctx(ctx).
- Where("avgdelay = 0").
- Where("losspk > ?", pingConfig.NodeLoos).
- Where("maxdelay = 0").
- Where("serverid = ?", serverid).
- OrderDesc("create_at").
- Limit(1).
- Scan(&lastDownTime)
- if err != nil {
- glog.Debug(ctx, err.Error())
- return consts.Normal, err
- }
-
- if lastDownTime == nil {
- return consts.Normal, nil
- }
-
- successCount, err := dao.Pinglog.Ctx(ctx).
- Where("create_at > ?", lastDownTime).
- Where("avgdelay > 0").
- Where("serverid = ?", serverid).
- Count()
- if err != nil {
- glog.Debug(ctx, err.Error())
- return consts.Normal, err
- }
-
- if successCount >= pingConfig.NodeRecover {
- return consts.Recovered, nil
- }
-
- return consts.Normal, nil
- }
- func (c *sPing) ClearLog(ctx context.Context) {
- clearLogTime, err := g.Cfg().Get(ctx, "node.clearLogTime")
- if err != nil {
- glog.Debug(ctx, err.Error())
- return
- }
- timeStartStr := gtime.Now().AddDate(0, 0, -clearLogTime.Int()).String()
- glog.Info(ctx, timeStartStr)
- _, err = dao.Pinglog.Ctx(ctx).Where("create_at <= ?", timeStartStr).Delete()
- if err != nil {
- glog.Error(ctx, err)
- return
- }
- }
|