perf: pass context to importer for better tracing

2 files changed, 11 insertions, 2 deletions

diff --git a/internal/importer/main.go b/internal/importer/main.go
index dfa2477..2af1666 100644
--- a/internal/importer/main.go
+++ b/internal/importer/main.go
@@ -144,6 +144,7 @@ func createSourceImporter(
 }
 
 func Start(
+	ctx context.Context,
 	cfg *config.Config,
 	log *log.Logger,
 	indexer *index.WriteIndex,
@@ -158,7 +159,7 @@ func Start(
 
 	log.Debug("starting importer", "timeout", cfg.Importer.Timeout.Duration)
 	importCtx, cancelImport := context.WithTimeout(
-		context.Background(),
+		ctx,
 		cfg.Importer.Timeout.Duration,
 	)
 	defer cancelImport()
diff --git a/internal/importer/main_test.go b/internal/importer/main_test.go
index 0bc438d..9e19b8c 100644
--- a/internal/importer/main_test.go
+++ b/internal/importer/main_test.go
@@ -1,6 +1,7 @@
 package importer
 
 import (
+	"context"
 	"testing"
 
 	"go.alanpearce.eu/searchix/internal/config"
@@ -21,7 +22,14 @@ func BenchmarkImporterLowMemory(b *testing.B) {
 		b.Fatal(err)
 	}
 
-	err = Start(&cfg, logger.Named("importer"), write, false, &[]string{"nixpkgs"})
+	err = Start(
+		context.Background(),
+		&cfg,
+		logger.Named("importer"),
+		write,
+		false,
+		&[]string{"nixpkgs"},
+	)
 	if err != nil {
 		b.Fatal(err)
 	}