collector: receive context

Signed-off-by: Alexey Ivanov <[email protected]>

Author: SaveTheRbtz

collector/collector.go

@@ -208,8 +208,8 @@ type collector struct {
 	logger log.Logger
 }
 
-func New(target string, module *config.Module, logger log.Logger) *collector {
-	return &collector{target: target, module: module, logger: logger}
+func New(ctx context.Context, target string, module *config.Module, logger log.Logger) *collector {
+	return &collector{ctx: ctx, target: target, module: module, logger: logger}
 }
 
 // Describe implements Prometheus.Collector.

main.go

@@ -99,7 +99,7 @@ func handler(w http.ResponseWriter, r *http.Request, logger log.Logger) {
 
 	start := time.Now()
 	registry := prometheus.NewRegistry()
-	c := collector.New(target, module, logger)
+	c := collector.New(r.Context(), target, module, logger)
 	registry.MustRegister(c)
 	// Delegate http serving to Prometheus client library, which will call collector.Collect.
 	h := promhttp.HandlerFor(registry, promhttp.HandlerOpts{})