pass logger to kolide client

nais · Jun 7, 2024 · d0581f0 · d0581f0
1 parent 26a681f
commit d0581f0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/cmd/apiserver/main.go b/cmd/apiserver/main.go
@@ -200,7 +200,7 @@ func run(log *logrus.Entry, cfg config.Config) error {
 			return fmt.Errorf("kolide integration enabled but no kolide-api-token provided")
 		}
 
-		kolideClient = kolide.New(cfg.KolideApiToken)
+		kolideClient = kolide.New(cfg.KolideApiToken, log.WithField("component", "kolide-client"))
 		err := kolideClient.RefreshCache(ctx)
 		if err != nil {
 			return fmt.Errorf("initial kolide cache warmup: %w", err)

diff --git a/internal/apiserver/kolide/client.go b/internal/apiserver/kolide/client.go
@@ -26,14 +26,15 @@ type client struct {
 	log logrus.FieldLogger
 }
 
-func New(token string) Client {
+func New(token string, log logrus.FieldLogger) Client {
 	return &client{
 		baseUrl: "https://k2.kolide.com/api/v0",
 		client: &http.Client{
 			Transport: NewTransport(token),
 		},
 		checks:  &Cache[uint64, Check]{},
 		devices: &Cache[string, Device]{},
+		log:     log,
 	}
 }