fix: durable state store schema definition

Tochemey · Dec 17, 2024 · c43fc33 · c43fc33
1 parent 9c75be3
commit c43fc33
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 10 deletions.
diff --git a/plugins/statestore/postgres/schema_utils.go b/plugins/statestore/postgres/schema_utils.go
@@ -46,14 +46,12 @@ func (d SchemaUtils) CreateTable(ctx context.Context) error {
 	DROP TABLE IF EXISTS states_store;
 	CREATE TABLE IF NOT EXISTS states_store
 	(
-	    persistence_id  VARCHAR(255)          NOT NULL,
-	    version_number BIGINT                NOT NULL,
+	    persistence_id  VARCHAR(255)          PRIMARY KEY,
+	    version_number BIGINT                 NOT NULL,
 	    state_payload   BYTEA                 NOT NULL,
 	    state_manifest  VARCHAR(255)          NOT NULL,
 	    timestamp       BIGINT                NOT NULL,
-	    shard_number BIGINT NOT NULL ,
-	
-	    PRIMARY KEY (persistence_id, version_number)
+	    shard_number BIGINT NOT NULL
 	);
 	`
 	_, err := d.db.Exec(ctx, schemaDDL)

diff --git a/resources/durablestore_postgres.sql b/resources/durablestore_postgres.sql
@@ -24,16 +24,13 @@
 
 CREATE TABLE IF NOT EXISTS states_store
 (
-    persistence_id  VARCHAR(255)          NOT NULL,
+    persistence_id  VARCHAR(255)          PRIMARY KEY,
     version_number BIGINT                 NOT NULL,
     state_payload   BYTEA                 NOT NULL,
     state_manifest  VARCHAR(255)          NOT NULL,
     timestamp       BIGINT                NOT NULL,
-    shard_number    BIGINT                NOT NULL,
-
-    PRIMARY KEY (persistence_id, version_number)
+    shard_number    BIGINT                NOT NULL
 );
 
 --- create an indexes
-CREATE INDEX IF NOT EXISTS idx_states_store_persistence_id ON states_store(persistence_id);
 CREATE INDEX IF NOT EXISTS idx_states_store_version_number ON states_store(version_number);