Update content-length header on setting response body

hummingbird-project · Apr 23, 2024 · 12f5696 · 12f5696
1 parent 32a724e
commit 12f5696
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/Sources/HummingbirdCore/Response/Response.swift b/Sources/HummingbirdCore/Response/Response.swift
@@ -17,7 +17,13 @@ import HTTPTypes
 /// Holds all the required to generate a HTTP Response
 public struct Response: Sendable {
     public var head: HTTPResponse
-    public var body: ResponseBody
+    public var body: ResponseBody {
+        didSet {
+            if let contentLength = body.contentLength {
+                self.head.headerFields[.contentLength] = String(describing: contentLength)
+            }
+        }
+    }
 
     public init(status: HTTPResponse.Status, headers: HTTPFields = .init(), body: ResponseBody = .init()) {
         self.head = .init(status: status, headerFields: headers)