Merge pull request #102 from chaindexing/enforce-declarative-migrations

Enforce declarative state migrations
chaindexing · Apr 18, 2024 · 93e19ff · 93e19ff
2 parents 1f8dc78 + ce815fa
commit 93e19ff
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 19 deletions.
diff --git a/chaindexing-tests/src/tests/states.rs b/chaindexing-tests/src/tests/states.rs
@@ -99,12 +99,9 @@ impl ContractState for Nft {
 }
 struct NftMigrations;
 impl StateMigrations for NftMigrations {
-    fn migrations(&self) -> Vec<&'static str> {
-        vec![
-            "CREATE TABLE IF NOT EXISTS nfts (
-        token_id INTEGER NOT NULL,
-    )",
-        ]
+    fn migrations(&self) -> &'static [&'static str] {
+        &["CREATE TABLE IF NOT EXISTS nfts (
+            token_id INTEGER NOT NULL)"]
     }
 }
 

diff --git a/chaindexing/src/states/migrations.rs b/chaindexing/src/states/migrations.rs
@@ -6,7 +6,7 @@ use super::STATE_VERSIONS_TABLE_PREFIX;
 // easen the type strictness for consumer applications.
 // Trait/Callback? this way, consumer apps can statically visualize their migrations
 pub trait StateMigrations: Send + Sync {
-    fn migrations(&self) -> Vec<&'static str>;
+    fn migrations(&self) -> &'static [&'static str];
 
     fn get_table_names(&self) -> Vec<String> {
         self.migrations().iter().fold(vec![], |mut table_names, migration| {
@@ -411,8 +411,8 @@ mod contract_state_migrations_get_migration_test {
     struct TestState;
 
     impl StateMigrations for TestState {
-        fn migrations(&self) -> Vec<&'static str> {
-            vec![
+        fn migrations(&self) -> &'static [&'static str] {
+            &[
                 "CREATE TABLE IF NOT EXISTS nft_states (
                       token_id INTEGER NOT NULL,
                       contract_address VARCHAR NOT NULL,
@@ -429,29 +429,25 @@ mod contract_state_migrations_get_migration_test {
     struct TestStateWithPrimaryKey;
 
     impl StateMigrations for TestStateWithPrimaryKey {
-        fn migrations(&self) -> Vec<&'static str> {
-            vec![
-                "CREATE TABLE IF NOT EXISTS nft_states (
+        fn migrations(&self) -> &'static [&'static str] {
+            &["CREATE TABLE IF NOT EXISTS nft_states (
                       id SERIAL PRIMARY KEY,
                       token_id INTEGER NOT NULL,
                       contract_address VARCHAR NOT NULL,
                       owner_address VARCHAR NOT NULL
-                  )",
-            ]
+                  )"]
         }
     }
 
     struct TestStateWithJsonField;
 
     impl StateMigrations for TestStateWithJsonField {
-        fn migrations(&self) -> Vec<&'static str> {
-            vec![
-                "CREATE TABLE IF NOT EXISTS nft_states (
+        fn migrations(&self) -> &'static [&'static str] {
+            &["CREATE TABLE IF NOT EXISTS nft_states (
                       id SERIAL PRIMARY KEY,
                       token_id INTEGER NOT NULL,
                       json_field JSON DEFAULT '{}',
-                  )",
-            ]
+                  )"]
         }
     }
 }