stat.go

package egorm

import (
	"net/http"
	"sync"
	"time"

	"github.com/gotomicro/ego/core/elog"
	"github.com/gotomicro/ego/core/emetric"
	"github.com/gotomicro/ego/server/egovernor"
	jsoniter "github.com/json-iterator/go"
)

var instances = sync.Map{}

func init() {
	egovernor.HandleFunc("/debug/gorm/stats", func(w http.ResponseWriter, r *http.Request) {
		_ = jsoniter.NewEncoder(w).Encode(stats())
	})
	go monitor()
}

func monitor() {
	for {
		instances.Range(func(key, val interface{}) bool {
			name := key.(string)
			db := val.(*Component)

			sqlDB, err := db.DB()
			if err != nil {
				elog.EgoLogger.With(elog.FieldComponent(PackageName)).Panic("monitor db error", elog.FieldErr(err))
				return false
			}
			sqlStats := sqlDB.Stats()
			// Gauge指标
			emetric.ClientStatsGauge.Set(float64(sqlStats.MaxOpenConnections), emetric.TypeGorm, name, "max_open_connections")
			emetric.ClientStatsGauge.Set(float64(sqlStats.OpenConnections), emetric.TypeGorm, name, "open_connections")
			emetric.ClientStatsGauge.Set(float64(sqlStats.InUse), emetric.TypeGorm, name, "in_use")
			emetric.ClientStatsGauge.Set(float64(sqlStats.Idle), emetric.TypeGorm, name, "idle")
			emetric.ClientStatsGauge.Set(float64(sqlStats.MaxIdleClosed), emetric.TypeGorm, name, "max_idle_closed")
			emetric.ClientStatsGauge.Set(float64(sqlStats.MaxIdleTimeClosed), emetric.TypeGorm, name, "max_idle_time_closed")
			emetric.ClientStatsGauge.Set(float64(sqlStats.MaxLifetimeClosed), emetric.TypeGorm, name, "max_lifetime_closed")
			// 以下数据为db里的累加值，如果要看瞬时的，需要在metric里使用irate
			emetric.ClientStatsGauge.Set(float64(sqlStats.WaitCount), emetric.TypeGorm, name, "wait_count")
			emetric.ClientStatsGauge.Set(float64(sqlStats.WaitDuration.Milliseconds()/1000), emetric.TypeGorm, name, "wait_duration")
			return true
		})
		time.Sleep(time.Second * 10)
	}
}

// stats
func stats() (stats map[string]interface{}) {
	stats = make(map[string]interface{})
	instances.Range(func(key, val interface{}) bool {
		name := key.(string)
		db := val.(*Component)

		sqlDB, err := db.DB()
		if err != nil {
			elog.EgoLogger.With(elog.FieldComponent(PackageName)).Panic("stats db error", elog.FieldErr(err))
			return false
		}
		stats[name] = sqlDB.Stats()
		return true
	})

	return
}