diag.go

package restful

import (
	"github.com/Cepave/open-falcon-backend/common/gin/mvc"
	apiModel "github.com/Cepave/open-falcon-backend/common/model/mysqlapi"
	"github.com/Cepave/open-falcon-backend/modules/mysqlapi/rdb"
	"github.com/Cepave/open-falcon-backend/modules/mysqlapi/service"
)

func health() mvc.OutputBody {
	portalRdbDiag := rdb.GlobalDbHolder.Diagnose(rdb.DB_PORTAL)
	graphRdbDiag := rdb.GlobalDbHolder.Diagnose(rdb.DB_GRAPH)

	health := &apiModel.HealthView{
		Rdb: &apiModel.AllRdbHealth{
			Dsn:             portalRdbDiag.Dsn,
			OpenConnections: portalRdbDiag.OpenConnections,
			PingResult:      portalRdbDiag.PingResult,
			PingMessage:     portalRdbDiag.PingMessage,
			Portal:          portalRdbDiag,
			Graph:           graphRdbDiag,
		},
		Http: &apiModel.Http{
			Listening: GinConfig.GetAddress(),
		},
		Nqm: &apiModel.Nqm{
			Heartbeat: &apiModel.Heartbeat{
				Count: service.NqmQueue.ConsumedCount(),
			},
		},
	}

	return mvc.JsonOutputBody(health)
}