Function TestDB_MergeForString

merge_test.go:72–136

(t *testing.T)

}

func TestDB_MergeForString(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-string-merge-%s/", mode.name)

		removeDir(opts.Dir)

		for _, idxMode := range []EntryIdxMode{HintKeyValAndRAMIdxMode, HintKeyAndRAMIdxMode} {
			removeDir(opts.Dir)
			opts.EntryIdxMode = idxMode
			db, err := Open(opts)
			txCreateBucket(t, db, DataStructureBTree, bucket, nil)
			require.NoError(t, err)

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

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

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

			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++ {
				txGet(t, db, bucket, testutils.GetTestBytes(i), testutils.GetTestBytes(i), ErrKeyNotFound)
			}

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

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

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

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

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

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