From cb4a56b54dc12d99a6c1ef121d8a6844f5d0d066 Mon Sep 17 00:00:00 2001
From: Caleb Doxsey <cdoxsey@pomerium.com>
Date: Thu, 14 Sep 2023 15:37:26 -0600
Subject: [PATCH] core/identity: fix slow restart

---
 internal/identity/manager/manager.go | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/internal/identity/manager/manager.go b/internal/identity/manager/manager.go
index 5f13d5967..02ddb635a 100644
--- a/internal/identity/manager/manager.go
+++ b/internal/identity/manager/manager.go
@@ -117,6 +117,7 @@ func (mgr *Manager) refreshLoop(ctx context.Context, update <-chan updateRecords
 	}
 	select {
 	case <-ctx.Done():
+		return ctx.Err()
 	case msg := <-update:
 		mgr.onUpdateRecords(ctx, msg)
 	}
@@ -128,9 +129,9 @@ func (mgr *Manager) refreshLoop(ctx context.Context, update <-chan updateRecords
 
 	// start refreshing
 	maxWait := time.Minute * 10
-	nextTime := time.Now().Add(maxWait)
+	var nextTime time.Time
 
-	timer := time.NewTimer(time.Until(nextTime))
+	timer := time.NewTimer(0)
 	defer timer.Stop()
 
 	for {