Merge pull request #81 from holaplex/abdul/use-nonblocking

use non blocking solana rpc client
holaplex · Oct 12, 2023 · e39b3ed · e39b3ed
2 parents 40412c4 + a69cd38
commit e39b3ed
Show file tree

Hide file tree

Showing 3 changed files with 60 additions and 36 deletions.
diff --git a/consumer/src/backend.rs b/consumer/src/backend.rs
@@ -103,40 +103,43 @@ impl<A> From<TransactionResponse<A>> for SolanaPendingTransaction {
     }
 }
 
+#[async_trait]
 pub trait CollectionBackend {
-    fn create(
+    async fn create(
         &self,
         txn: MetaplexMasterEditionTransaction,
     ) -> Result<TransactionResponse<MasterEditionAddresses>>;
 
-    fn update(
+    async fn update(
         &self,
         collection: &collections::Model,
         txn: MetaplexMasterEditionTransaction,
     ) -> Result<TransactionResponse<UpdateMasterEditionAddresses>>;
 
-    fn update_mint(
+    async fn update_mint(
         &self,
         collection: &collections::Model,
         mint: &collection_mints::Model,
         txn: UpdateSolanaMintPayload,
     ) -> Result<TransactionResponse<UpdateCollectionMintAddresses>>;
 
-    fn retry_update_mint(
+    async fn retry_update_mint(
         &self,
         revision: &update_revisions::Model,
     ) -> Result<TransactionResponse<UpdateCollectionMintAddresses>>;
 
-    fn switch(
+    async fn switch(
         &self,
         mint: &collection_mints::Model,
         collection: &collections::Model,
         new_collection: &collections::Model,
     ) -> Result<TransactionResponse<SwitchCollectionAddresses>>;
 }
 
+#[async_trait]
 pub trait MintBackend<T, R> {
-    fn mint(&self, collection: &collections::Model, txn: T) -> Result<TransactionResponse<R>>;
+    async fn mint(&self, collection: &collections::Model, txn: T)
+    -> Result<TransactionResponse<R>>;
 }
 
 #[async_trait]

diff --git a/consumer/src/events.rs b/consumer/src/events.rs
@@ -251,7 +251,7 @@ impl EventKind {
 
                 let address = if let Some(compression_leaf) = compression_leafs {
                     let signature = Signature::from_str(&signature)?;
-                    let nonce = solana.extract_compression_nonce(&signature)?;
+                    let nonce = solana.extract_compression_nonce(&signature).await?;
 
                     let asset_id = mpl_bubblegum::utils::get_asset_id(
                         &Pubkey::from_str(&compression_leaf.merkle_tree)?,
@@ -765,7 +765,7 @@ impl Processor {
                 .map_err(|k| ProcessorError::new(k, kind, ErrorSource::TreasuryStatus));
         }
 
-        let res = match self.solana().submit_transaction(&res) {
+        let res = match self.solana().submit_transaction(&res).await {
             Ok(sig) => self
                 .event_submitted(kind, &key, sig)
                 .await
@@ -834,6 +834,7 @@ impl Processor {
         let conn = self.db.get();
         let tx = backend
             .create(payload.clone())
+            .await
             .map_err(ProcessorErrorKind::Solana)?;
 
         let MasterEditionAddresses {
@@ -880,6 +881,7 @@ impl Processor {
 
             let tx = backend
                 .mint(&collection, payload)
+                .await
                 .map_err(ProcessorErrorKind::Solana)?;
 
             let compression_leaf = compression_leafs::Model {
@@ -910,6 +912,7 @@ impl Processor {
 
         let tx = backend
             .mint(&collection, payload)
+            .await
             .map_err(ProcessorErrorKind::Solana)?;
 
         let collection_mint = collection_mints::Model {
@@ -949,6 +952,7 @@ impl Processor {
 
         let tx = backend
             .mint(&collection, payload)
+            .await
             .map_err(ProcessorErrorKind::Solana)?;
 
         let collection_mint = collection_mints::Model {
@@ -979,6 +983,7 @@ impl Processor {
 
         let tx = backend
             .update(&collection, payload)
+            .await
             .map_err(ProcessorErrorKind::Solana)?;
 
         Ok(tx.into())
@@ -1000,6 +1005,7 @@ impl Processor {
 
         let tx = backend
             .update_mint(&collection, &mint, payload)
+            .await
             .map_err(ProcessorErrorKind::Solana)?;
 
         let UpdateCollectionMintAddresses {
@@ -1035,6 +1041,7 @@ impl Processor {
 
         let tx = backend
             .retry_update_mint(&revision)
+            .await
             .map_err(ProcessorErrorKind::Solana)?;
 
         Ok(tx.into())
@@ -1083,6 +1090,7 @@ impl Processor {
         let conn = self.db.get();
         let tx = backend
             .create(payload.clone())
+            .await
             .map_err(ProcessorErrorKind::Solana)?;
 
         let MasterEditionAddresses {
@@ -1133,6 +1141,7 @@ impl Processor {
 
         let tx = backend
             .switch(&mint, &collection, &new_collection)
+            .await
             .map_err(ProcessorErrorKind::Solana)?;
 
         Ok(tx.into())
@@ -1157,6 +1166,7 @@ impl Processor {
 
         let tx = backend
             .mint(&collection, payload)
+            .await
             .map_err(ProcessorErrorKind::Solana)?;
 
         let MintEditionAddresses {
@@ -1196,6 +1206,7 @@ impl Processor {
 
             let tx = backend
                 .mint(&collection, payload)
+                .await
                 .map_err(ProcessorErrorKind::Solana)?;
 
             let leaf_model = CompressionLeaf::find_by_id(conn, id)
@@ -1218,6 +1229,7 @@ impl Processor {
 
         let tx = backend
             .mint(&collection, payload)
+            .await
             .map_err(ProcessorErrorKind::Solana)?;
 
         let MintMetaplexAddresses {