feat: expose Database#CreatePartitionIfNotExists method

forbole · Dec 27, 2023 · 67e8053 · 67e8053
1 parent a274ebd
commit 67e8053
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/database/postgresql/postgresql.go b/database/postgresql/postgresql.go
@@ -63,8 +63,8 @@ type Database struct {
 	Logger         logging.Logger
 }
 
-// createPartitionIfNotExists creates a new partition having the given partition id if not existing
-func (db *Database) createPartitionIfNotExists(table string, partitionID int64) error {
+// CreatePartitionIfNotExists creates a new partition having the given partition id if not existing
+func (db *Database) CreatePartitionIfNotExists(table string, partitionID int64) error {
 	partitionTable := fmt.Sprintf("%s_%d", table, partitionID)
 
 	stmt := fmt.Sprintf(
@@ -155,7 +155,7 @@ func (db *Database) SaveTx(tx *types.Tx) error {
 	partitionSize := config.Cfg.Database.PartitionSize
 	if partitionSize > 0 {
 		partitionID = tx.Height / partitionSize
-		err := db.createPartitionIfNotExists("transaction", partitionID)
+		err := db.CreatePartitionIfNotExists("transaction", partitionID)
 		if err != nil {
 			return err
 		}
@@ -286,7 +286,7 @@ func (db *Database) SaveMessage(msg *types.Message) error {
 	partitionSize := config.Cfg.Database.PartitionSize
 	if partitionSize > 0 {
 		partitionID = msg.Height / partitionSize
-		err := db.createPartitionIfNotExists("message", partitionID)
+		err := db.CreatePartitionIfNotExists("message", partitionID)
 		if err != nil {
 			return err
 		}