Prometheus的remote_write的样例

我会带着你远行 2022-10-12 12:29 182阅读 0赞

**[https://github.com/prometheus/prometheus/tree/release-2.28/documentation/examples/remote\_storage/example\_write\_adapter][https_github.com_prometheus_prometheus_tree_release-2.28_documentation_examples_remote_storage_example_write_adapter]**

**[prometheus][]/[documentation][]/[examples][]/[remote\_storage][remote_storage]/[example\_write\_adapter][https_github.com_prometheus_prometheus_tree_release-2.28_documentation_examples_remote_storage_example_write_adapter]/server.go：**

go get github.com/prometheus/prometheus@b0944590a1
    go get github.com/prometheus/common/model

## Remote Write Adapter Example ##

This is a simple example of how to write a server to receive samples from the remote storage output.

To use it:

go build
    ./example_write_adapter

...and then add the following to your `prometheus.yml`:

remote_write:
      - url: "http://localhost:1234/receive"

Then start Prometheus:

./prometheus

使用示例：

// Copyright 2016 The Prometheus Authors
    // Licensed under the Apache License, Version 2.0 (the "License");
    // you may not use this file except in compliance with the License.
    // You may obtain a copy of the License at
    //
    // http://www.apache.org/licenses/LICENSE-2.0
    //
    // Unless required by applicable law or agreed to in writing, software
    // distributed under the License is distributed on an "AS IS" BASIS,
    // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    // See the License for the specific language governing permissions and
    // limitations under the License.
    
    package main
    
    import (
    	"encoding/json"
    	"fmt"
    	"github.com/Shopify/sarama"
    	"github.com/prometheus/prometheus/storage/remote"
    	"log"
    	"net/http"
    	"time"
    )
    
    func main() {
    
    	config := sarama.NewConfig()  //实例化个sarama的Config
    	config.Producer.Return.Successes = true  //是否开启消息发送成功后通知 successes channel
    	config.Producer.Partitioner = sarama.NewRandomPartitioner //随机分区器
    	client,err := sarama.NewClient([]string{"aiops01:6667"}, config) //初始化客户端
    	if err != nil{
    		panic(err.Error())
    	}
    
    	defer func(client sarama.Client) {
    		err := client.Close()
    		if err != nil {
    			panic(err.Error())
    		}
    	}(client)
    
    	producer,err := sarama.NewSyncProducerFromClient(client)
    	if err != nil{
    		panic(err.Error())
    	}
    
    	http.HandleFunc("/receive", func(w http.ResponseWriter, r *http.Request) {
    		req, err := remote.DecodeWriteRequest(r.Body)
    		if err != nil {
    			http.Error(w, err.Error(), http.StatusBadRequest)
    			return
    		}
    
    		for _, ts := range req.Timeseries {
    			m := make(map[string]interface{}, len(ts.Labels))
    			for _, l := range ts.Labels {
    				m[l.Name] = l.Value
    			}
    
    			fmt.Println(m)
    
    			for _, s := range ts.Samples {
    				fmt.Printf("\tSample: %f %d\n", s.Value, s.Timestamp)
    				m["value"] = s.Value
    				m["Timestamp"] = s.Timestamp
    
    				var timestampStr = time.Unix(s.Timestamp,0).Format("2006-01-02 15:04:05")
    				bytes, _ := json.Marshal(&m)
    
    				log.Println(string(bytes))
    				partition, offset , err := producer.SendMessage(&sarama.ProducerMessage{Topic: "prometheus_topic", Key: sarama.StringEncoder(timestampStr), Value: sarama.StringEncoder(string(bytes))})
    				if err != nil {
    					log.Fatalf("unable to produce message: %q", err)
    				}
    				fmt.Println("partition",partition)
    				fmt.Println("offset",offset)
    			}
    
    			//for _, e := range ts.Exemplars {
    			//	m := make(model.Metric, len(e.Labels))
    			//	for _, l := range e.Labels {
    			//		m[model.LabelName(l.Name)] = model.LabelValue(l.Value)
    			//	}
    			//	fmt.Printf("\tExemplar:  %+v %f %d\n", m, e.Value, e.Timestamp)
    			//}
    		}
    	})
    
    	log.Fatal(http.ListenAndServe(":1234", nil))
    }

[https_github.com_prometheus_prometheus_tree_release-2.28_documentation_examples_remote_storage_example_write_adapter]: https://github.com/prometheus/prometheus/tree/release-2.28/documentation/examples/remote_storage/example_write_adapter
[prometheus]: https://github.com/prometheus/prometheus/tree/release-2.28
[documentation]: https://github.com/prometheus/prometheus/tree/release-2.28/documentation
[examples]: https://github.com/prometheus/prometheus/tree/release-2.28/documentation/examples
[remote_storage]: https://github.com/prometheus/prometheus/tree/release-2.28/documentation/examples/remote_storage