Feat: batching of RPC requests.

mempool · Jul 27, 2024 · c7c873c · c7c873c
1 parent d070272
commit c7c873c
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 12 deletions.
diff --git a/rust-toolchain b/rust-toolchain
@@ -1 +1 @@
-1.70
+1.80
diff --git a/src/electrum/server.rs b/src/electrum/server.rs
@@ -521,18 +521,25 @@ impl Connection {
                     trace!("RPC {:?}", msg);
                     match msg {
                         Message::Request(line) => {
-                            let cmd: Value = from_str(&line).chain_err(|| "invalid JSON format")?;
-                            let reply = match (
-                                cmd.get("method"),
-                                cmd.get("params").unwrap_or(&empty_params),
-                                cmd.get("id"),
-                            ) {
-                                (Some(Value::String(method)), Value::Array(params), Some(id)) => {
-                                    self.handle_command(method, params, id)?
-                                }
-                                _ => bail!("invalid command: {}", cmd),
+                            let cmd: [Value; 1] = [from_str(&line).chain_err(|| "invalid JSON format")?];
+                            let cmds = match &cmd[..] {
+                                [Value::Array(arr)] => arr,
+                                x => x,
                             };
-                            self.send_values(&[reply])?
+                            let mut replies = Vec::with_capacity(cmds.len());
+                            for cmd in cmds {
+                                replies.push(match (
+                                    cmd.get("method"),
+                                    cmd.get("params").unwrap_or(&empty_params),
+                                    cmd.get("id"),
+                                ) {
+                                    (Some(Value::String(method)), Value::Array(params), Some(id)) => {
+                                        self.handle_command(method, params, id)?
+                                    }
+                                    _ => bail!("invalid command: {}", cmd),
+                                });
+                            }
+                            self.send_values(&replies)?
                         }
                         Message::PeriodicUpdate => {
                             let values = self