Prometheus自定义注册表无法正常运行

时间:2018-08-16 11:17:59

标签: go prometheus

我仍在学习Prometheus,所以也许我不确定该问题是否正确。

我需要的是一个自定义注册表,我只能在其中收集指标。因为我正在学习Prometheus,所以我对Prometheus提供的默认度量标准(例如go_gc_duration_secondsgo_gc_duration_seconds_countgo_threadspromhttp_metric_handler_requests_in_flight

)并不真正感兴趣。
package main

import (
    "fmt"
    "log"
    "math/rand"
    "net/http"
    "sync"
    "time"

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

var Types = [2]string{"Random", "Simple"}

type Queue struct {
    mutex sync.Mutex
    jobs  []Job
}

func (q *Queue) Add(job Job) {
    q.mutex.Lock()
    q.jobs = append(q.jobs, job)
    q.mutex.Unlock()
}

func (q *Queue) Dequeue() Job {
    q.mutex.Lock()
    job := q.jobs[0]
    q.jobs = q.jobs[1:]
    q.mutex.Unlock()
    return job
}

type Job struct {
    message string
    Type    string
}

func (j *Job) Run() {
    fmt.Println(j.message)
}

var jobsInQueue = prometheus.NewGaugeVec(
    prometheus.GaugeOpts{
        Name: "jobs_in_queue",
        Help: "Current number of jobs in the queue",
    },
    []string{"job_type"},
)
var register = prometheus.NewRegistry()
var queue = &Queue{}

func init() {
    rand.Seed(2)
    // prometheus.MustRegister(jobsInQueue)
    // register the collector.. 
    register.MustRegister(jobsInQueue)
    queue.jobs = make([]Job, 0)
}

func main() {

    go func() {
        i := 0
        for {
            job := Job{}
            num := rand.Intn(2)
            type_d := Types[num]
            job.Type = type_d
            job.message = fmt.Sprintf("[%s] job %d", type_d, i)
            enqueueJob(job)
            fmt.Println(i)
            i++
            time.Sleep(1 * time.Second)
        }
    }()

    // sleep so that we do not read from a empty queue
    time.Sleep(2 * time.Millisecond)

    go func() {
        for {
            runNextJob()
            time.Sleep(2 * time.Second)
        }
    }()

    http.Handle("/metrics", promhttp.Handler())
    log.Fatal(http.ListenAndServe(":8080", nil))

}

func enqueueJob(job Job) {
    queue.Add(job)
    jobsInQueue.WithLabelValues(job.Type).Inc()
}

func runNextJob() {
    job := queue.Dequeue()
    jobsInQueue.WithLabelValues(job.Type).Dec()
    job.Run()
}

但是当我运行以下代码时,在jobs_in_queue的{​​{1}}端点中看不到我的/metrics指标。

我应该如何进行这项工作。

1 个答案:

答案 0 :(得分:0)

promhttp.Handler()为默认注册表创建一个处理程序。您需要使用promhttp.HandlerFor(registry,HandlerOpts {})