add servecontent to caddy

protomaps · Feb 10, 2024 · f8d2c37 · f8d2c37
1 parent bac8bf4
commit f8d2c37
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/caddy/pmtiles_proxy.go b/caddy/pmtiles_proxy.go
@@ -1,6 +1,7 @@
 package caddy
 
 import (
+	"bytes"
 	"fmt"
 	"io"
 	"log"
@@ -72,8 +73,12 @@ func (m Middleware) ServeHTTP(w http.ResponseWriter, r *http.Request, next caddy
 	for k, v := range headers {
 		w.Header().Set(k, v)
 	}
-	w.WriteHeader(statusCode)
-	w.Write(body)
+	if statusCode == 200 {
+		http.ServeContent(w, r, "", time.UnixMilli(0), bytes.NewReader(body))
+	} else {
+		w.WriteHeader(statusCode)
+		w.Write(body)
+	}
 	m.logger.Info("response", zap.Int("status", statusCode), zap.String("path", r.URL.Path), zap.Duration("duration", time.Since(start)))
 
 	return next.ServeHTTP(w, r)