add mongodb tokens update after restart

codex-team · Dec 3, 2024 · 86bf099 · 86bf099
1 parent 2df0bb1
commit 86bf099
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/cmd/collector/main.go b/cmd/collector/main.go
@@ -90,6 +90,12 @@ func (x *RunCommand) Execute(args []string) error {
 	// connect to accounts MongoDB
 	doneAccountsContext := make(chan struct{})
 	accountsClient := accounts.New(cfg.AccountsMongoDBURI)
+
+	err = accountsClient.UpdateTokenCache()
+	if err != nil {
+		log.Errorf("failed to update token cache: %s", err)
+	}
+
 	go periodic.RunPeriodically(accountsClient.UpdateTokenCache, cfg.TokenUpdatePeriod, doneAccountsContext)
 	defer close(doneAccountsContext)