Change expiration to expire_at_s

xmtp · Mar 11, 2024 · 85112a0 · 85112a0
1 parent 8345016
commit 85112a0
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 8 deletions.
diff --git a/xmtp_mls/migrations/2024-02-23-182644_add_expiration/down.sql b/xmtp_mls/migrations/2024-02-23-182644_add_expiration/down.sql
@@ -1,4 +1,4 @@
 DROP INDEX IF EXISTS key_store_expiration;
 
 ALTER TABLE openmls_key_store
-DROP COLUMN expiration;
+DROP COLUMN expire_at_s;
diff --git a/xmtp_mls/migrations/2024-02-23-182644_add_expiration/up.sql b/xmtp_mls/migrations/2024-02-23-182644_add_expiration/up.sql
@@ -1,4 +1,4 @@
 ALTER TABLE openmls_key_store
-ADD COLUMN expiration BIGINT;
+ADD COLUMN expire_at_s BIGINT;
 
-CREATE INDEX IF NOT EXISTS key_store_expiration ON openmls_key_store(expiration);
+CREATE INDEX IF NOT EXISTS key_store_expiration ON openmls_key_store(expire_at_s);
diff --git a/xmtp_mls/src/storage/encrypted_store/key_store_entry.rs b/xmtp_mls/src/storage/encrypted_store/key_store_entry.rs
@@ -10,7 +10,7 @@ use crate::{impl_fetch, impl_store, Delete};
 pub struct StoredKeyStoreEntry {
     pub key_bytes: Vec<u8>,
     pub value_bytes: Vec<u8>,
-    pub expiration: Option<i64>,
+    pub expire_at_s: Option<i64>,
 }
 
 impl_fetch!(StoredKeyStoreEntry, openmls_key_store, Vec<u8>);
@@ -37,7 +37,7 @@ impl DbConnection<'_> {
         let entry = StoredKeyStoreEntry {
             key_bytes: key,
             value_bytes: value,
-            expiration: if let Some(e) = exp {
+            expire_at_s: if let Some(e) = exp {
                 e.try_into().ok()
             } else {
                 None
@@ -54,7 +54,7 @@ impl DbConnection<'_> {
                 .values(entry)
                 .execute(conn)?;
             // Delete expired entries.
-            diesel::delete(openmls_key_store.filter(expiration.lt(current_time))).execute(conn)
+            diesel::delete(openmls_key_store.filter(expire_at_s.lt(current_time))).execute(conn)
         })?;
 
         Ok(())

diff --git a/xmtp_mls/src/storage/encrypted_store/schema.rs b/xmtp_mls/src/storage/encrypted_store/schema.rs
@@ -47,7 +47,7 @@ diesel::table! {
     openmls_key_store (key_bytes) {
         key_bytes -> Binary,
         value_bytes -> Binary,
-        expiration -> Nullable<BigInt>,
+        expire_at_s -> Nullable<BigInt>,
     }
 }
 

diff --git a/xmtp_mls/src/subscriptions.rs b/xmtp_mls/src/subscriptions.rs
@@ -93,7 +93,7 @@ where
                             log::error!("Error rotating key package: {:?}", err);
                         }
                         Some(group)
-                    },
+                    }
                     Err(err) => {
                         log::error!("Error processing stream entry: {:?}", err);
                         None