Skip to main content

forge_runtime/gateway/
request.rs

1use serde::{Deserialize, Serialize};
2
3/// RPC request body for function calls.
4#[derive(Debug, Clone, Serialize, Deserialize)]
5pub struct RpcRequest {
6    /// Function name to invoke.
7    pub function: String,
8    /// Function arguments as JSON.
9    #[serde(default)]
10    pub args: serde_json::Value,
11}
12
13impl RpcRequest {
14    /// Create a new RPC request.
15    pub fn new(function: impl Into<String>, args: serde_json::Value) -> Self {
16        Self {
17            function: function.into(),
18            args,
19        }
20    }
21}
22
23/// Batch RPC request containing multiple function calls.
24#[allow(dead_code)]
25#[derive(Debug, Clone, Serialize, Deserialize)]
26pub struct BatchRpcRequest {
27    /// List of requests to execute.
28    pub requests: Vec<RpcRequest>,
29}
30
31#[cfg(test)]
32#[allow(clippy::unwrap_used, clippy::indexing_slicing, clippy::panic)]
33mod tests {
34    use super::*;
35
36    #[test]
37    fn test_rpc_request_deserialization() {
38        let json = r#"{"function": "getUser", "args": {"id": "123"}}"#;
39        let req: RpcRequest = serde_json::from_str(json).unwrap();
40        assert_eq!(req.function, "getUser");
41        assert_eq!(req.args["id"], "123");
42    }
43
44    #[test]
45    fn test_rpc_request_default_args() {
46        let json = r#"{"function": "healthCheck"}"#;
47        let req: RpcRequest = serde_json::from_str(json).unwrap();
48        assert_eq!(req.function, "healthCheck");
49        assert!(req.args.is_null());
50    }
51
52    #[test]
53    fn test_batch_request() {
54        let json = r#"{"requests": [{"function": "a"}, {"function": "b"}]}"#;
55        let batch: BatchRpcRequest = serde_json::from_str(json).unwrap();
56        assert_eq!(batch.requests.len(), 2);
57    }
58}