add index

Author: evanh

pg_migrations/0001_create_inflight_activations.sql

@@ -17,4 +17,6 @@ CREATE TABLE IF NOT EXISTS inflight_taskactivations (
     namespace TEXT NOT NULL,
     taskname TEXT NOT NULL,
     on_attempts_exceeded INTEGER NOT NULL DEFAULT 1
-) PARTITION BY LIST (partition);
+);
+
+CREATE INDEX idx_activation_partition ON inflight_taskactivations (partition);

src/store/postgres_activation_store.rs

@@ -276,7 +276,7 @@ impl InflightActivationStore for PostgresActivationStore {
                 b.push_bind(row.taskname);
                 b.push_bind(row.on_attempts_exceeded as i32);
             })
-            .push(" ON CONFLICT(id) DO NOTHING")
+            .push(" ON CONFLICT(id, partition) DO NOTHING")
             .build();
         let mut conn = self.acquire_write_conn_metric("store").await?;
         Ok(query.execute(&mut *conn).await?.into())