Add constant value to make the code more readable (#568)

* add memery size constant values to make the code more readable

* change the unit to kib, mib etc.

* change the unit from Tib to TiB

---------

Co-authored-by: Matthias Loibl <[email protected]>

db.go

@@ -36,6 +36,14 @@ import (
 	"github.com/polarsignals/frostdb/wal"
 )
 
+const (
+	B   = 1
+	KiB = 1024 * B
+	MiB = 1024 * KiB
+	GiB = 1024 * MiB
+	TiB = 1024 * GiB
+)
+
 type ColumnStore struct {
 	mtx                 *sync.RWMutex
 	dbs                 map[string]*DB
@@ -89,8 +97,8 @@ func New(
 		indexConfig:      DefaultIndexConfig(),
 		indexDegree:      2,
 		splitSize:        2,
-		granuleSizeBytes: 1 * 1024 * 1024,   // 1MB granule size before splitting
-		activeMemorySize: 512 * 1024 * 1024, // 512MB
+		granuleSizeBytes: 1 * MiB,
+		activeMemorySize: 512 * MiB,
 	}
 
 	for _, option := range options {

db_test.go

@@ -46,7 +46,7 @@ func TestDBWithWALAndBucket(t *testing.T) {
 		WithWAL(),
 		WithStoragePath(dir),
 		WithReadWriteStorage(sinksource),
-		WithActiveMemorySize(100*1024),
+		WithActiveMemorySize(100*KiB),
 	)
 	require.NoError(t, err)
 	db, err := c.DB(context.Background(), "test")
@@ -71,7 +71,7 @@ func TestDBWithWALAndBucket(t *testing.T) {
 		WithWAL(),
 		WithStoragePath(dir),
 		WithReadWriteStorage(sinksource),
-		WithActiveMemorySize(100*1024),
+		WithActiveMemorySize(100*KiB),
 	)
 	require.NoError(t, err)
 	defer c.Close()
@@ -1358,7 +1358,7 @@ func Test_DB_ReadOnlyQuery(t *testing.T) {
 		WithWAL(),
 		WithStoragePath(dir),
 		WithReadWriteStorage(sinksource),
-		WithActiveMemorySize(100*1024),
+		WithActiveMemorySize(100*KiB),
 	)
 	require.NoError(t, err)
 	db, err := c.DB(context.Background(), "test")
@@ -1383,7 +1383,7 @@ func Test_DB_ReadOnlyQuery(t *testing.T) {
 		WithWAL(),
 		WithStoragePath(dir),
 		WithReadWriteStorage(sinksource),
-		WithActiveMemorySize(100*1024),
+		WithActiveMemorySize(100*KiB),
 	)
 	require.NoError(t, err)
 	defer c.Close()
@@ -1867,7 +1867,7 @@ func Test_DB_DropStorage(t *testing.T) {
 		WithLogger(logger),
 		WithWAL(),
 		WithStoragePath(dir),
-		WithActiveMemorySize(1024*1024),
+		WithActiveMemorySize(1*MiB),
 	)
 	defer func() {
 		_ = c.Close()
@@ -1909,7 +1909,7 @@ func Test_DB_DropStorage(t *testing.T) {
 		WithLogger(logger),
 		WithWAL(),
 		WithStoragePath(dir),
-		WithActiveMemorySize(1024*1024),
+		WithActiveMemorySize(1*MiB),
 	)
 	defer func() {
 		_ = c.Close()
@@ -2009,8 +2009,8 @@ func Test_DB_SnapshotOnClose(t *testing.T) {
 		WithLogger(logger),
 		WithStoragePath(dir),
 		WithWAL(),
-		WithActiveMemorySize(1024*1024*1024),
-		WithSnapshotTriggerSize(1024*1024*1024),
+		WithActiveMemorySize(1*GiB),
+		WithSnapshotTriggerSize(1*GiB),
 		WithManualBlockRotation(),
 	)
 	require.NoError(t, err)

store.go

@@ -175,7 +175,7 @@ func (b *DefaultObjstoreBucket) openBlockFile(ctx context.Context, blockName str
 	file, err := parquet.OpenFile(
 		r,
 		size,
-		parquet.ReadBufferSize(5*1024*1024), // 5MB read buffers
+		parquet.ReadBufferSize(5*MiB), // 5MB read buffers
 		parquet.SkipBloomFilters(true),
 		parquet.FileReadMode(parquet.ReadModeAsync),
 	)

table_test.go

@@ -1135,7 +1135,7 @@ func Test_Table_DynamicColumnNotDefined(t *testing.T) {
 func TestTableUniquePrimaryIndex(t *testing.T) {
 	c, err := New(WithGranuleSizeBytes(1), WithIndexConfig([]*index.LevelConfig{
 		{Level: index.L0, MaxSize: 180},
-		{Level: index.L1, MaxSize: 1024 * 1024 * 1024 * 1024},
+		{Level: index.L1, MaxSize: 1 * TiB},
 	}))
 	require.NoError(t, err)
 	defer c.Close()