Add ResponseBuilder to rest_api

2026-03-14 10:22:38 +00:00 · 2019-09-02 14:29:36 +10:00
parent d05c2d4110
commit 6c50758bdf
4 changed files with 61 additions and 13 deletions
--- a/beacon_node/rest_api/src/response_builder.rs
+++ b/beacon_node/rest_api/src/response_builder.rs
@@ -0,0 +1,50 @@
+use super::{ApiError, ApiResult};
+use http::header;
+use hyper::{Body, Request, Response, StatusCode};
+use serde::Serialize;
+use ssz::Encode;
+
+pub enum Encoding {
+    JSON,
+    SSZ,
+    YAML,
+}
+
+pub struct ResponseBuilder {
+    encoding: Encoding,
+}
+
+impl ResponseBuilder {
+    pub fn new(req: &Request<Body>) -> Self {
+        let encoding = match req.headers().get(header::CONTENT_TYPE) {
+            Some(h) if h == "application/ssz" => Encoding::SSZ,
+            Some(h) if h == "application/yaml" => Encoding::YAML,
+            _ => Encoding::JSON,
+        };
+
+        Self { encoding }
+    }
+
+    pub fn body<T: Serialize + Encode>(self, item: &T) -> ApiResult {
+        let body: Body = match self.encoding {
+            Encoding::JSON => Body::from(serde_json::to_string(&item).map_err(|e| {
+                ApiError::ServerError(format!(
+                    "Unable to serialize response body as JSON: {:?}",
+                    e
+                ))
+            })?),
+            Encoding::SSZ => Body::from(item.as_ssz_bytes()),
+            Encoding::YAML => Body::from(serde_yaml::to_string(&item).map_err(|e| {
+                ApiError::ServerError(format!(
+                    "Unable to serialize response body as YAML: {:?}",
+                    e
+                ))
+            })?),
+        };
+
+        Response::builder()
+            .status(StatusCode::OK)
+            .body(Body::from(body))
+            .map_err(|e| ApiError::ServerError(format!("Failed to build response: {:?}", e)))
+    }
+}