package metric

import (
	"gitee.com/go-kade/library/ioc/config/http"

	"gitee.com/go-kade/library/ioc"
	"gitee.com/go-kade/library/ioc/apps/metric"
	"gitee.com/go-kade/library/ioc/config/log"
	"github.com/gin-gonic/gin"

	"github.com/prometheus/client_golang/prometheus/promhttp"
	"github.com/rs/zerolog"
)

func init() {
	ioc.Api().Registry(&ginHandler{
		Metric: metric.NewDefaultMetric(),
	})
}

type ginHandler struct {
	log *zerolog.Logger
	ioc.ObjectImpl

	*metric.Metric
}

func (h *ginHandler) Init() error {
	h.log = log.Sub(metric.AppName)
	return nil
}

func (h *ginHandler) Name() string {
	return metric.AppName
}

func (h *ginHandler) Version() string {
	return "v1"
}

func (h *ginHandler) Meta() ioc.ObjectMeta {
	meta := ioc.DefaultObjectMeta()
	meta.CustomPathPrefix = h.Endpoint
	return meta
}

func (h *ginHandler) Registry(r gin.IRouter) {
	r.GET("/", func(ctx *gin.Context) {
		// 基于标准库 包装了一层
		promhttp.Handler().ServeHTTP(ctx.Writer, ctx.Request)
	})

	h.log.Info().Msgf("Get the Metric using http://%s%s", http.Get().Addr(), h.Endpoint)
}