From 69efa1fb92ef1cd8f17158f70115cc787bd21d4e Mon Sep 17 00:00:00 2001 From: Antoni Sawicki Date: Mon, 26 Oct 2020 01:27:21 -0700 Subject: [PATCH] if to switch to improve readability --- wrp.go | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/wrp.go b/wrp.go index d776108..369fbe5 100644 --- a/wrp.go +++ b/wrp.go @@ -224,9 +224,10 @@ func imgServer(out http.ResponseWriter, req *http.Request) { if !nodel { defer delete(img, req.URL.Path) } - if strings.HasPrefix(req.URL.Path, ".gif") { + switch { + case strings.HasPrefix(req.URL.Path, ".gif"): out.Header().Set("Content-Type", "image/gif") - } else if strings.HasPrefix(req.URL.Path, ".png") { + case strings.HasPrefix(req.URL.Path, ".png"): out.Header().Set("Content-Type", "image/png") } out.Header().Set("Content-Length", strconv.Itoa(len(imgbuf.Bytes())))