feat: support load batch

Signed-off-by: MEX7 <mex7.0828@gmail.com>

Author: kl7sn

container.go

@@ -1,6 +1,8 @@
 package prom2click
 
 import (
+	"fmt"
+
 	"github.com/gotomicro/ego/core/econf"
 	"github.com/gotomicro/ego/core/elog"
 )
@@ -32,6 +34,27 @@ func Load(key string) *Container {
 	return c
 }
 
+func LoadBatch(key string) []*Container {
+	containers := make([]*Container, 0)
+	configs := make([]*config, 0)
+	if err := econf.UnmarshalKey(key, &configs); err != nil {
+		elog.EgoLogger.With(elog.FieldComponent(PackageName)).Panic("parse config error", elog.FieldErr(err), elog.FieldKey(key))
+		return nil
+	}
+	for index := range configs {
+		c := DefaultContainer()
+		c.logger = c.logger.With(elog.FieldComponentName(key))
+		c.config.Host = configs[index].Host
+		c.config.Port = configs[index].Port
+		c.config.ClickhouseDSN = configs[index].ClickhouseDSN
+		c.config.ClickhouseDB = configs[index].ClickhouseDB
+		c.config.ClickhouseTable = configs[index].ClickhouseTable
+		c.name = fmt.Sprintf("%s_%d", key, index)
+		containers = append(containers, c)
+	}
+	return containers
+}
+
 // Build 构建组件
 func (c *Container) Build(options ...Option) *Component {
 	for _, option := range options {
@@ -45,6 +68,5 @@ func (c *Container) Build(options ...Option) *Component {
 	if c.config.EnableMetricInterceptor {
 		server.Use(metricServerInterceptor())
 	}
-
 	return server
 }

prom_writer.go

@@ -4,6 +4,7 @@ import (
 	"database/sql"
 	"fmt"
 	"sort"
+	"strconv"
 	"sync"
 	"time"
 
@@ -49,37 +50,42 @@ func NewWriter(conf *config) (*promWriter, error) {
 
 	w.tx = prometheus.NewCounter(
 		prometheus.CounterOpts{
-			Name: "sent_samples_total",
-			Help: "Total number of processed samples sent to remote storage.",
+			Name:        "sent_samples_total",
+			Help:        "Total number of processed samples sent to remote storage.",
+			ConstLabels: map[string]string{"host": conf.Host, "port": strconv.Itoa(conf.Port)},
 		},
 	)
 
 	w.ko = prometheus.NewCounter(
 		prometheus.CounterOpts{
-			Name: "failed_samples_total",
-			Help: "Total number of processed samples which failed on send to remote storage.",
+			Name:        "failed_samples_total",
+			Help:        "Total number of processed samples which failed on send to remote storage.",
+			ConstLabels: map[string]string{"host": conf.Host, "port": strconv.Itoa(conf.Port)},
 		},
 	)
 
 	w.test = prometheus.NewCounter(
 		prometheus.CounterOpts{
-			Name: "prometheus_remote_storage_sent_batch_duration_seconds_bucket_test",
-			Help: "Test metric to ensure backfilled metrics are readable via prometheus.",
+			Name:        "prometheus_remote_storage_sent_batch_duration_seconds_bucket_test",
+			Help:        "Test metric to ensure backfilled metrics are readable via prometheus.",
+			ConstLabels: map[string]string{"host": conf.Host, "port": strconv.Itoa(conf.Port)},
 		},
 	)
 
 	w.timings = prometheus.NewHistogram(
 		prometheus.HistogramOpts{
-			Name:    "sent_batch_duration_seconds",
-			Help:    "Duration of sample batch send calls to the remote storage.",
-			Buckets: prometheus.DefBuckets,
+			Name:        "sent_batch_duration_seconds",
+			Help:        "Duration of sample batch send calls to the remote storage.",
+			Buckets:     prometheus.DefBuckets,
+			ConstLabels: map[string]string{"host": conf.Host, "port": strconv.Itoa(conf.Port)},
 		},
 	)
 
 	w.rx = prometheus.NewCounter(
 		prometheus.CounterOpts{
-			Name: "received_samples_total",
-			Help: "Total number of received samples.",
+			Name:        "received_samples_total",
+			Help:        "Total number of received samples.",
+			ConstLabels: map[string]string{"host": conf.Host, "port": strconv.Itoa(conf.Port)},
 		},
 	)