Use serde_bytes for GsbHttpProxy Message and Response (#3339)

golemfactory · Oct 3, 2024 · 7f6a3ae · 7f6a3ae
1 parent 5f8be31
commit 7f6a3ae
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/exe-unit/components/gsb-http-proxy/Cargo.toml b/exe-unit/components/gsb-http-proxy/Cargo.toml
@@ -31,6 +31,7 @@ log = { version = "0.4", features = [] }
 rand = { workspace = true }
 reqwest = { version = "0.11", features = ["json", "stream"] }
 serde = { version = "1.0", features = ["derive"] }
+serde_bytes = "0.11.3"
 serde_derive = "1.0"
 serde_json = "1.0"
 thiserror = "1.0"

diff --git a/exe-unit/components/gsb-http-proxy/src/message.rs b/exe-unit/components/gsb-http-proxy/src/message.rs
@@ -13,6 +13,7 @@ use ya_service_bus::{RpcMessage, RpcStreamMessage};
 pub struct GsbHttpCallMessage {
     pub method: String,
     pub path: String,
+    #[serde(with = "serde_bytes")]
     pub body: Option<Vec<u8>>,
     pub headers: HashMap<String, Vec<String>>,
 }
@@ -28,6 +29,7 @@ impl RpcMessage for GsbHttpCallMessage {
 pub struct GsbHttpCallStreamingMessage {
     pub method: String,
     pub path: String,
+    #[serde(with = "serde_bytes")]
     pub body: Option<Vec<u8>>,
     pub headers: HashMap<String, Vec<String>>,
 }

diff --git a/exe-unit/components/gsb-http-proxy/src/response.rs b/exe-unit/components/gsb-http-proxy/src/response.rs
@@ -22,6 +22,7 @@ pub struct GsbHttpCallResponseHeader {
 
 #[derive(Clone, Debug, PartialEq, Serialize, Deserialize, Default)]
 pub struct GsbHttpCallResponseBody {
+    #[serde(with = "serde_bytes")]
     pub msg_bytes: Vec<u8>,
 }