options_compaction_test.go

package grocksdb

import (
	"testing"

	"github.com/stretchr/testify/require"
)

func TestOptionCompactions(t *testing.T) {
	t.Parallel()

	co := NewCompactRangeOptions()
	defer co.Destroy()

	co.SetFullHistoryTsLow([]byte{1, 2, 3})

	require.EqualValues(t, false, co.GetExclusiveManualCompaction())
	co.SetExclusiveManualCompaction(true)
	require.EqualValues(t, true, co.GetExclusiveManualCompaction())

	require.EqualValues(t, KIfHaveCompactionFilter, co.BottommostLevelCompaction())
	co.SetBottommostLevelCompaction(KForce)
	require.EqualValues(t, KForce, co.BottommostLevelCompaction())

	require.EqualValues(t, false, co.ChangeLevel())
	co.SetChangeLevel(true)
	require.EqualValues(t, true, co.ChangeLevel())

	require.EqualValues(t, -1, co.TargetLevel())
	co.SetTargetLevel(2)
	require.EqualValues(t, 2, co.TargetLevel())
}

func TestFifoCompactOption(t *testing.T) {
	t.Parallel()

	fo := NewDefaultFIFOCompactionOptions()
	defer fo.Destroy()

	fo.SetMaxTableFilesSize(2 << 10)
	require.EqualValues(t, 2<<10, fo.GetMaxTableFilesSize())

	require.False(t, fo.AllowCompaction())
	fo.SetAllowCompaction(true)
	require.True(t, fo.AllowCompaction())
}

func TestUniversalCompactOption(t *testing.T) {
	t.Parallel()

	uo := NewDefaultUniversalCompactionOptions()
	defer uo.Destroy()

	uo.SetSizeRatio(2)
	require.EqualValues(t, 2, uo.GetSizeRatio())

	uo.SetMinMergeWidth(3)
	require.EqualValues(t, 3, uo.GetMinMergeWidth())

	uo.SetMaxMergeWidth(123)
	require.EqualValues(t, 123, uo.GetMaxMergeWidth())

	uo.SetMaxSizeAmplificationPercent(20)
	require.EqualValues(t, 20, uo.GetMaxSizeAmplificationPercent())

	uo.SetCompressionSizePercent(18)
	require.EqualValues(t, 18, uo.GetCompressionSizePercent())

	uo.SetStopStyle(CompactionStopStyleTotalSize)
	require.EqualValues(t, CompactionStopStyleTotalSize, uo.GetStopStyle())
}