mirror of https://github.com/rocboss/paopao-ce
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.
42 lines
1.3 KiB
42 lines
1.3 KiB
1 year ago
|
// Copyright 2023 ROC. All rights reserved.
|
||
|
|
// Use of this source code is governed by a MIT style
|
||
|
|
// license that can be found in the LICENSE file.
|
||
|
|
|
||
|
|
package prometheus
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net/http"
|
||
|
|
|
||
|
|
"github.com/prometheus/client_golang/prometheus"
|
||
|
|
"github.com/prometheus/client_golang/prometheus/collectors"
|
||
|
|
"github.com/prometheus/client_golang/prometheus/promhttp"
|
||
|
|
"github.com/robfig/cron/v3"
|
||
|
|
"github.com/rocboss/paopao-ce/internal/conf"
|
||
|
|
"github.com/rocboss/paopao-ce/internal/core"
|
||
|
|
"github.com/rocboss/paopao-ce/internal/events"
|
||
|
|
"github.com/sirupsen/logrus"
|
||
|
|
)
|
||
|
|
|
||
|
|
func scheduleJobs(metrics *metrics) {
|
||
|
|
spec := conf.JobManagerSetting.UpdateMetricsInterval
|
||
|
|
schedule, err := cron.ParseStandard(spec)
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
events.OnTask(schedule, metrics.onUpdate)
|
||
|
|
logrus.Debug("shedule prometheus metrics update jobs complete")
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewHandler(ds core.DataService, wc core.WebCache) http.Handler {
|
||
|
|
// Create non-global registry.
|
||
|
|
registry := prometheus.NewRegistry()
|
||
|
|
// Add go runtime metrics and process collectors.
|
||
|
|
registry.MustRegister(
|
||
|
|
collectors.NewGoCollector(),
|
||
|
|
collectors.NewProcessCollector(collectors.ProcessCollectorOpts{}),
|
||
|
|
)
|
||
|
|
metrics := newMetrics(registry, ds, wc)
|
||
|
|
scheduleJobs(metrics)
|
||
|
|
return promhttp.HandlerFor(registry, promhttp.HandlerOpts{EnableOpenMetrics: true})
|
||
|
|
}
|