Function TestConcurrentLoadByCVAndRevOnCache

db/revision_cache_test.go:1944–1996

TestConcurrentLoadByCVAndRevOnCache: - Create cache - Now perform two concurrent Gets, one by CV and one by revid on a document that doesn't exist in the cache - This will trigger two concurrent loads from bucket in the CV code path and revid code path - In doing so we will have two processes trying

(t *testing.T)

//     element
//   - Grab the single element in the list and assert that both maps point to that element in the cache list
func TestConcurrentLoadByCVAndRevOnCache(t *testing.T) {
	cacheHitCounter, cacheMissCounter, cacheNumItems, memoryBytesCounted, getDocumentCounter, getRevisionCounter := base.SgwIntStat{}, base.SgwIntStat{}, base.SgwIntStat{}, base.SgwIntStat{}, base.SgwIntStat{}, base.SgwIntStat{}
	cacheOptions := &RevisionCacheOptions{
		MaxItemCount: 10,
		MaxBytes:     0,
	}
	cache := NewLRURevisionCache(cacheOptions, CreateTestSingleBackingStoreMap(&testBackingStore{[]string{"test_doc"}, &getDocumentCounter, &getRevisionCounter}, testCollectionID), &cacheHitCounter, &cacheMissCounter, &cacheNumItems, &memoryBytesCounted)

	ctx := base.TestCtx(t)

	wg := sync.WaitGroup{}
	wg.Add(2)

	cv := Version{SourceID: "test", Value: 123}
	go func() {
		_, err := cache.GetWithRev(ctx, "doc1", "1-abc", testCollectionID, RevCacheIncludeDelta)
		require.NoError(t, err)
		wg.Done()
	}()

	go func() {
		_, err := cache.GetWithCV(ctx, "doc1", &cv, testCollectionID, RevCacheIncludeDelta, false)
		require.NoError(t, err)
		wg.Done()
	}()

	wg.Wait()

	revElement := cache.cache[IDAndRev{RevID: "1-abc", DocID: "doc1"}].Value.(*revCacheValue)
	cvElement := cache.hlvCache[IDandCV{DocID: "doc1", Source: "test", Version: 123}].Value.(*revCacheValue)
	// we may have concurrent loads that lead to the same doc being loaded into two
	// rev cache items but only one reference to one or the other in the maps
	//	┌──────────┐   ┌──────────┐
	//	│          │   │          │
	//	│  Doc1    │   │   Doc1   │
	//	│          │   │          │
	//	└──────────┘   └──────────┘
	//	    ▲               ▲
	//	    │               │
	//	    │               │
	//	┌──────┐       ┌──────┐
	//	│ HLV  │       │ REV  │
	//	│ MAP  │       │ MAP  │
	//	└──────┘       └──────┘

	assert.LessOrEqual(t, cache.lruList.Len(), 2)
	assert.Equal(t, 1, len(cache.cache))
	assert.Equal(t, 1, len(cache.hlvCache))
	assert.Equal(t, revElement.revID, cvElement.revID)
	assert.Equal(t, revElement.id, cvElement.id)
	assert.Equal(t, revElement.cv.String(), cvElement.cv.String())
	require.JSONEq(t, string(revElement.bodyBytes), string(cvElement.bodyBytes))
}

// TestGetActive:
//   - Create db, create a doc on the db