使用您的数据为Prometheus类型收

你在学习Golang相关的知识吗？本文《使用您的数据为Prometheus类型收》，主要介绍的内容就涉及到，如果你想提升自己的开发能力，就不要错过这篇文章，大家要知道编程理论基础和实战操作都是不可或缺的哦！

免责声明：我是 golang 新手，之前没有用任何其他语言进行过太多编程。 我仍然希望有人能给我指出正确的方向。

目标是： 根据 prometheus golang 模块下的“示例”部分 (https://godoc.org/github.com/prometheus/client_golang/prometheus#collector) 以及提到“// just example fake data”的部分。当然是使用我自己的真实数据。

我的数据来自 rabbitmq 端点，采用 json 格式。我解析 json，然后可以使用正确的 key:value 创建自己的映射，我需要将其作为 func main() 范围下 goroutine 的一部分。

假设我的地图如下所示： [ “设备1”：754， “设备2”：765， ]

对于代码，让我们按照原始示例进行操作。

package main

import (
    "log"
    "net/http"
    "fmt"

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

type ClusterManager struct {
    Zone string
    // Contains many more fields not listed in this example.
}

func (c *ClusterManager) ReallyExpensiveAssessmentOfTheSystemState() (
    oomCountByHost map[string]int, ramUsageByHost map[string]float64,
) {
        // Just example fake data.
        oomCountByHost = map[string]int{
            "foo.example.org": 42,
            "bar.example.org": 2001,
        }
        ramUsageByHost = map[string]float64{
            "foo.example.org": 6.023e23,
            "bar.example.org": 3.14,
        }
        return
}

type ClusterManagerCollector struct {
    ClusterManager *ClusterManager
}

var (
    oomCountDesc = prometheus.NewDesc(
        "clustermanager_oom_crashes_total",
        "Number of OOM crashes.",
        []string{"host"}, nil,
    )
    ramUsageDesc = prometheus.NewDesc(
        "clustermanager_ram_usage_bytes",
        "RAM usage as reported to the cluster manager.",
        []string{"host"}, nil,
    )
)

func (cc ClusterManagerCollector) Describe(ch chan<- *prometheus.Desc) {
    prometheus.DescribeByCollect(cc, ch)
}

func (cc ClusterManagerCollector) Collect(ch chan<- prometheus.Metric) {
    oomCountByHost, ramUsageByHost := cc.ClusterManager.ReallyExpensiveAssessmentOfTheSystemState()
    for host, oomCount := range oomCountByHost {
        ch <- prometheus.MustNewConstMetric(
            oomCountDesc,
            prometheus.CounterValue,
            float64(oomCount),
            host,
        )
    }
    for host, ramUsage := range ramUsageByHost {
        ch <- prometheus.MustNewConstMetric(
            ramUsageDesc,
            prometheus.GaugeValue,
            ramUsage,
            host,
        )
    }
}

func NewClusterManager(zone string, reg prometheus.Registerer) *ClusterManager {
    c := &ClusterManager{
        Zone: zone,
    }
    cc := ClusterManagerCollector{ClusterManager: c}
    prometheus.WrapRegistererWith(prometheus.Labels{"zone": zone}, reg).MustRegister(cc)
    return c
}

func main() {
    conn, err := amqp.Dial("amqp://guest:guest@localhost:5672/")
    failOnError(err, "Failed to connect to RabbitMQ")
    defer conn.Close()

    ch, err := conn.Channel()
    failOnError(err, "Failed to open a channel")
    defer ch.Close()

    q, err := ch.QueueDeclare("hello", false, false, false, false, nil)
    failOnError(err, "Failed to declare a queue")

    msgs, err := ch.Consume(q.Name, "", true, false, false, false, nil)
    failOnError(err, "Failed to register a consumer")

    forever := make(chan bool)

    go func() {
        for d := range msgs {
            var streams []byte
            streams = d.Body

            var metrics sStreamingMetrics
            err := json.Unmarshal(streams, &metrics)
            if err != nil {
                fmt.Println(err)
            }

            var category string
            category = metrics.Resource.Category

            if category == "server" {
               myMap := make(map[string]float64)
               MyMap [metrics.Resource.ResourceDataList[0].ResourceId] = metrics.Resource.ResourceDataList[0].MetricSampleList[0].ValueArray[0]
             }
    }()

    reg := prometheus.NewPedanticRegistry()

    NewClusterManager("zone", reg)

    reg.MustRegister(
        prometheus.NewProcessCollector(prometheus.ProcessCollectorOpts{}),
        prometheus.NewGoCollector(),
    )

    http.Handle("/metrics", promhttp.HandlerFor(reg, promhttp.HandlerOpts{}))
    log.Fatal(http.ListenAndServe(":8080", nil))
    log.Printf(" [*] Waiting for logs. To exit press CTRL+C")
    <-forever
}

//relevant structs go here for parsing JSON

添加了 goroutine 和 main 以获取更大的图片。 由于我会时不时地接收数据，所以存在并发性。 我认为我在这里缺少的技能是如何调用该函数，以便 mymap 中的 key:values 进入 func (c *clustermanager)reallyexpenseassessmentofthesystemstate() ( oomcountbyhost 地图[字符串]int, ramusagebyhost 地图[字符串]float64,){}

解决方案

reallyexpectiveassessmentofthesystemstate 在 collect 方法中调用。您的地图不会“进入”该方法，它是由 reallyexppressiveassessmentofthesystemstate 返回的。

只需将代码从 goroutine 移至 reallyexpectiveassessmentofthesystemstate 即可：

func (c *ClusterManager) ReallyExpensiveAssessmentOfTheSystemState() map[string]float64 {
    myMap := make(map[string]float64)
    myMap["device1"] = 754
    myMap["device2"] = 765

    return myMap
}

func (cc ClusterManagerCollector) Collect(ch chan<- prometheus.Metric) {
    values := cc.ClusterManager.ReallyExpensiveAssessmentOfTheSystemState()

    for key, value := range values {
        ch <- prometheus.MustNewConstMetric(
            valueDesc,
            prometheus.CounterValue,
            value,
            key,
        )
    }
}

好了，本文到此结束，带大家了解了《使用您的数据为Prometheus类型收》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！