Function TestDB_MergeForZSet

merge_test.go:456–544

(t *testing.T)

}

func TestDB_MergeForZSet(t *testing.T) {
	runForMergeModes(t, func(t *testing.T, mode mergeTestMode) {
		bucket := "bucket"
		key := testutils.GetTestBytes(0)
		n := 1000
		opts := DefaultOptions
		opts.SegmentSize = KB
		opts.EnableMergeV2 = mode.enableMergeV2
		opts.Dir = fmt.Sprintf("/tmp/test-zset-merge-%s/", mode.name)

		for _, idxMode := range []EntryIdxMode{HintKeyValAndRAMIdxMode, HintKeyAndRAMIdxMode} {
			removeDir(opts.Dir)
			opts.EntryIdxMode = idxMode
			db, err := Open(opts)
			if exist := db.bucketMgr.ExistBucket(DataStructureSortedSet, bucket); !exist {
				txCreateBucket(t, db, DataStructureSortedSet, bucket, nil)
			}
			require.NoError(t, err)

			err = db.Merge()
			require.Equal(t, ErrDontNeedMerge, err)

			for i := 0; i < n; i++ {
				score, _ := strconv2.IntToFloat64(i)
				txZAdd(t, db, bucket, key, testutils.GetTestBytes(i), score, nil, nil)
			}

			for i := 0; i < n; i++ {
				score, _ := strconv2.IntToFloat64(i)
				txZScore(t, db, bucket, key, testutils.GetTestBytes(i), score, nil)
			}

			for i := 0; i < n/2; i++ {
				txZRem(t, db, bucket, key, testutils.GetTestBytes(i), nil)
			}

			for i := 0; i < n/2; i++ {
				score, _ := strconv2.IntToFloat64(i)
				txZScore(t, db, bucket, key, testutils.GetTestBytes(i), score, ErrSortedSetMemberNotExist)
			}

			for i := n / 2; i < n; i++ {
				score, _ := strconv2.IntToFloat64(i)
				txZScore(t, db, bucket, key, testutils.GetTestBytes(i), score, nil)
			}

			dbCnt, err := db.getRecordCount()
			require.NoError(t, err)
			require.Equal(t, int64(n/2), dbCnt)

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

			dbCnt, err = db.getRecordCount()
			require.NoError(t, err)
			require.Equal(t, int64(n/2), dbCnt)

			for i := 0; i < n/2; i++ {
				score, _ := strconv2.IntToFloat64(i)