fix(chore): Content-Length set before store

darkweak · Aug 17, 2023 · 60f2d29 · 60f2d29
1 parent bca1957
commit 60f2d29
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/pkg/middleware/middleware.go b/pkg/middleware/middleware.go
@@ -226,6 +226,9 @@ func (s *SouinBaseHandler) Store(
 		if res.Header.Get("Date") == "" {
 			res.Header.Set("Date", now.Format(http.TimeFormat))
 		}
+		if res.Header.Get("Content-Length") == "" {
+			res.Header.Set("Content-Length", fmt.Sprint(customWriter.Buf.Len()))
+		}
 		res.Header.Set(rfc.StoredLengthHeader, res.Header.Get("Content-Length"))
 		response, err := httputil.DumpResponse(&res, true)
 		if err == nil {

diff --git a/plugins/caddy/httpcache_test.go b/plugins/caddy/httpcache_test.go
@@ -56,7 +56,6 @@ func TestHead(t *testing.T) {
 	localhost:9080 {
 		route /cache-head {
 			cache
-			header Content-Length 12
 			respond "Hello, HEAD!"
 		}
 	}`, "caddyfile")