fix: address review feedback on cache invalidation
Some checks failed
CI / build (pull_request) Successful in 19s
CI / integration (pull_request) Failing after 2m0s

- Fix cache not repopulated after invalidation: Always update cache with fresh data instead of just deleting on mismatch
- Fix race condition: Hold mutex lock during entire fetch operation to prevent SaveEvent from running between fetch and cache update
- Improve test: Add second GetLatestVersion call to verify cache was properly repopulated after invalidation

Co-Authored-By: Claude Code <noreply@anthropic.com>
This commit is contained in:
Claude Code
2026-01-13 01:31:03 +01:00
parent 6de897ef60
commit fd1938672e
2 changed files with 21 additions and 14 deletions

View File

@@ -303,17 +303,22 @@ func (jes *JetStreamEventStore) getEventsWithErrorsInternal(actorID string, from
return result, nil
}
// GetLatestVersion returns the latest version for an actor, invalidating cache
// if the actual version in JetStream is newer than cached version.
// This strategy ensures cache consistency even if external processes write to
// GetLatestVersion returns the latest version for an actor, repopulating cache
// with fresh data to ensure consistency even if external processes write to
// the same JetStream stream.
func (jes *JetStreamEventStore) GetLatestVersion(actorID string) (int64, error) {
// Hold lock during fetch to prevent race condition with SaveEvent
jes.mu.Lock()
defer jes.mu.Unlock()
events, err := jes.GetEvents(actorID, 0)
if err != nil {
return 0, err
}
if len(events) == 0 {
// No events for this actor - ensure cache is cleared
delete(jes.versions, actorID)
return 0, nil
}
@@ -324,17 +329,10 @@ func (jes *JetStreamEventStore) GetLatestVersion(actorID string) (int64, error)
}
}
// Invalidate cache if actual version differs from cached version
// This handles the case where external writers modify the stream
jes.mu.Lock()
if cachedVersion, ok := jes.versions[actorID]; ok && latestVersion > cachedVersion {
// Cache was stale - invalidate it by deleting
delete(jes.versions, actorID)
} else if !ok {
// Update cache for future calls
jes.versions[actorID] = latestVersion
}
jes.mu.Unlock()
// Always repopulate cache with the fresh data just fetched
// This ensures cache is in sync with actual state, whether from local writes
// or external writes detected by version comparison
jes.versions[actorID] = latestVersion
return latestVersion, nil
}

View File

@@ -1390,6 +1390,15 @@ func TestJetStreamEventStore_CacheInvalidationOnExternalWrite(t *testing.T) {
t.Errorf("store1 should see version 2 after external write, got %d", v2)
}
// Verify cache was repopulated - second GetLatestVersion should use cache efficiently
v2Again, err := store1.GetLatestVersion(actorID)
if err != nil {
t.Fatalf("Second GetLatestVersion from store1 failed: %v", err)
}
if v2Again != 2 {
t.Errorf("store1 cache should have version 2, got %d", v2Again)
}
// store2: Save event v3 (another external write)
event3 := &aether.Event{
ID: "evt-3",