Function TestDB_MergeForSet

merge_test.go:372–454

(t *testing.T)

}

func TestDB_MergeForSet(t *testing.T) {
	runForMergeModes(t, func(t *testing.T, mode mergeTestMode) {
		bucket := "bucket"
		opts := DefaultOptions
		opts.SegmentSize = KB
		opts.EnableMergeV2 = mode.enableMergeV2
		opts.Dir = fmt.Sprintf("/tmp/test-set-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(DataStructureSet, bucket); !exist {
				txCreateBucket(t, db, DataStructureSet, bucket, nil)
			}

			require.NoError(t, err)

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

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

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

			var spopValue []byte
			err = db.Update(func(tx *Tx) error {
				var err error
				spopValue, err = tx.SPop(bucket, key)
				AssertErr(t, err, nil)
				return nil
			})
			require.NoError(t, err)

			txSIsMember(t, db, bucket, key, spopValue, false)

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

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

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

			txSIsMember(t, db, bucket, key, spopValue, false)
			for i := n / 2; i < n; i++ {
				v := testutils.GetTestBytes(i)
				if bytes.Equal(v, spopValue) {
					continue
				}