add submit txn rpc

fleek-network · Jul 14, 2023 · 286f03e · 286f03e
1 parent ea30248
commit 286f03e
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 4 deletions.
diff --git a/core/rpc/src/handlers.rs b/core/rpc/src/handlers.rs
@@ -2,7 +2,7 @@ use std::sync::Arc;
 
 use axum::{Extension, Json};
 use draco_interfaces::{
-    types::{CommodityServed, EpochInfo, NodeInfo, ProtocolParams, TotalServed},
+    types::{CommodityServed, EpochInfo, NodeInfo, ProtocolParams, TotalServed, UpdateRequest},
     SyncQueryRunnerInterface,
 };
 use fleek_crypto::{EthAddress, NodePublicKey};
@@ -73,7 +73,8 @@ impl RpcServer {
             )
             .with_method("flk_is_valid_node", is_valid_node_handler::<Q>)
             .with_method("flk_get_node_registry", get_node_registry_handler::<Q>)
-            .with_method("flk_get_reputation", get_reputation_handler::<Q>);
+            .with_method("flk_get_reputation", get_reputation_handler::<Q>)
+            .with_method("flk_send_txn", send_txn::<Q>);
 
         RpcServer(server.finish())
     }
@@ -224,3 +225,14 @@ pub async fn get_node_registry_handler<Q: SyncQueryRunnerInterface>(
 ) -> Result<Vec<NodeInfo>> {
     Ok(data.0.query_runner.get_node_registry())
 }
+
+pub async fn send_txn<Q: SyncQueryRunnerInterface>(
+    data: Data<Arc<RpcData<Q>>>,
+    Params(param): Params<UpdateRequest>,
+) -> Result<()> {
+    data.0
+        .mempool_socket
+        .run(param)
+        .await
+        .map_err(Error::internal)
+}
diff --git a/core/rpc/src/server.rs b/core/rpc/src/server.rs
@@ -25,7 +25,7 @@ pub struct Rpc<Q: SyncQueryRunnerInterface> {
 
 pub struct RpcData<Q: SyncQueryRunnerInterface> {
     pub query_runner: Q,
-    pub _mempool_address: MempoolSocket,
+    pub mempool_socket: MempoolSocket,
 }
 
 impl<Q: SyncQueryRunnerInterface> Rpc<Q> {
@@ -84,7 +84,7 @@ impl<Q: SyncQueryRunnerInterface + Send + Sync + 'static> RpcInterface<Q> for Rp
     ) -> anyhow::Result<Self> {
         Ok(Self {
             data: Arc::new(RpcData {
-                _mempool_address: mempool,
+                mempool_socket: mempool,
                 query_runner,
             }),
             config,