Function TestDB_MergeHintFileDifferentDataStructures

merge_test.go:843–974

(t *testing.T)

}

func TestDB_MergeHintFileDifferentDataStructures(t *testing.T) {
	runForMergeModes(t, func(t *testing.T, mode mergeTestMode) {
		opts := DefaultOptions
		opts.SegmentSize = KB
		opts.Dir = fmt.Sprintf("/tmp/test-merge-hintfile-ds-%s/", mode.name)
		opts.EnableHintFile = true
		opts.EnableMergeV2 = mode.enableMergeV2

		removeDir(opts.Dir)

		db, err := Open(opts)
		require.NoError(t, err)

		bucketBTree := "bucket_btree"
		txCreateBucket(t, db, DataStructureBTree, bucketBTree, nil)
		for i := 0; i < 100; i++ {
			txPut(t, db, bucketBTree, testutils.GetTestBytes(i), testutils.GetTestBytes(i), Persistent, nil, nil)
		}

		bucketSet := "bucket_set"
		txCreateBucket(t, db, DataStructureSet, bucketSet, nil)
		key := testutils.GetTestBytes(0)
		for i := 0; i < 50; i++ {
			txSAdd(t, db, bucketSet, key, testutils.GetTestBytes(i), nil, nil)
		}

		bucketList := "bucket_list"
		txCreateBucket(t, db, DataStructureList, bucketList, nil)
		listKey := testutils.GetTestBytes(0)
		for i := 0; i < 30; i++ {
			txPush(t, db, bucketList, listKey, testutils.GetTestBytes(i), true, nil, nil)
		}

		bucketZSet := "bucket_zset"
		txCreateBucket(t, db, DataStructureSortedSet, bucketZSet, nil)
		zsetKey := testutils.GetTestBytes(0)
		for i := 0; i < 20; i++ {
			txZAdd(t, db, bucketZSet, zsetKey, testutils.GetTestBytes(i), float64(i), nil, nil)
		}

		for i := 0; i < 25; i++ {
			txDel(t, db, bucketBTree, testutils.GetTestBytes(i), nil)
		}
		for i := 0; i < 10; i++ {
			txSRem(t, db, bucketSet, key, testutils.GetTestBytes(i), nil)
		}
		for i := 0; i < 5; i++ {
			txPop(t, db, bucketList, listKey, testutils.GetTestBytes(i), nil, false)
		}
		for i := 0; i < 5; i++ {
			txZRem(t, db, bucketZSet, zsetKey, testutils.GetTestBytes(i), nil)
		}

		require.NoError(t, db.Merge())

		sets := collectMergeFileSets(t, opts.Dir)
		idsToCheck := dataFileIDsForMode(mode, sets)
		require.Greater(t, len(idsToCheck), 0)