diff options
| author | Adnan Maolood <[email protected]> | 2021-04-21 11:38:26 -0400 |
|---|---|---|
| committer | Adnan Maolood <[email protected]> | 2021-04-21 11:38:34 -0400 |
| commit | de93d44786b8b5298693c2e047a241c4094292c1 (patch) | |
| tree | b04c7ac77755a6df8847f2e8fb649b1c7b1f2091 | |
| parent | fs: Fix panic on indexing URL of zero length (diff) | |
| download | go-gemini-de93d44786b8b5298693c2e047a241c4094292c1.tar.xz go-gemini-de93d44786b8b5298693c2e047a241c4094292c1.zip | |
LoggingMiddleware: Prevent writing empty meta
| -rw-r--r-- | middleware.go | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/middleware.go b/middleware.go index fb8ca90..77c4d94 100644 --- a/middleware.go +++ b/middleware.go @@ -31,7 +31,12 @@ func (w *logResponseWriter) SetMediaType(mediatype string) { func (w *logResponseWriter) Write(b []byte) (int, error) { if !w.wroteHeader { - w.WriteHeader(StatusSuccess, w.mediatype) + meta := w.mediatype + if meta == "" { + // Use default media type + meta = defaultMediaType + } + w.WriteHeader(StatusSuccess, meta) } n, err := w.rw.Write(b) w.Wrote += n |