Merge pull request #17 from Hsn723/fix-timeout

fix timeout on metrics collection

VERSION

@@ -1 +1 @@
-0.1.0
+0.1.1

cmd/root.go

@@ -7,7 +7,6 @@ import (
 	"net/http"
 	"os"
 	"strconv"
-	"time"
 
 	"github.com/hsn723/rdap-exporter/collector"
 	"github.com/hsn723/rdap-exporter/config"
@@ -45,7 +44,7 @@ func runRoot(cmd *cobra.Command, _ []string) error {
 	rdapExporter := collector.NewRdapExporter(*conf)
 	prometheus.MustRegister(rdapExporter)
 
-	ctx, cancelFunc := context.WithTimeout(cmd.Context(), time.Duration(conf.Timeout)*time.Second)
+	ctx, cancelFunc := context.WithCancel(cmd.Context())
 	defer cancelFunc()
 	go rdapExporter.StartMetricsCollection(ctx)
 

collector/collector.go

@@ -67,6 +67,7 @@ func collectRdapInfo(ctx context.Context, e *RdapExporter, domain string) {
 		Query: domain,
 	}
 	req = req.WithContext(ctx)
+	req.Timeout = time.Duration(e.config.Timeout) * time.Second
 	client := &rdap.Client{}
 	resp, err := client.Do(req)
 	if err != nil {