Merge pull request #5 from Parker-Kasiewicz/hydrator_updates

Adding Error Caching
stanfordio · Oct 10, 2024 · 69e3c4f · 69e3c4f
2 parents 3d025fc + 96ef537
commit 69e3c4f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pkg/hydrator/hydrator.go b/pkg/hydrator/hydrator.go
@@ -78,6 +78,10 @@ func (h *Hydrator) LookupIdentity(identifier string) (identity *atpidentity.Iden
 	cachedValue, found := h.Cache.Get(key)
 
 	if found && cachedValue != nil {
+		if cachedError, isErr := cachedValue.(error); isErr {
+            log.Debugf("Cached error for %s: %v", identifier, cachedError)
+            return nil, cachedError // Return the cached error
+        }
 		identity = cachedValue.(*atpidentity.Identity)
 		return
 	}
@@ -93,6 +97,7 @@ func (h *Hydrator) LookupIdentity(identifier string) (identity *atpidentity.Iden
 
 	identity, err = h.IdentityDirectory.Lookup(h.Context, *resolvedIdentifier)
 	if err != nil {
+		h.Cache.SetWithTTL(key, err, 0, time.Duration(1)*time.Hour*24)
 		return
 	}