README: include new middlewares

Peter Kieltyka · Peter Kieltyka · commit f4d4455cd3e7 · 2017-11-09T10:45:59.000-05:00
diff --git a/README.md b/README.md
@@ -320,6 +320,7 @@ with `net/http` can be used with chi's mux.
 -----------------------------------------------------------------------------------------------------------
 | chi/middlware Handler | description                                                                     |
 |:----------------------|:---------------------------------------------------------------------------------
+| AllowContentType      | Explicit whitelist of accepted request Content-Types                            |
 | Compress              | Gzip compression for clients that accept compressed responses                   |
 | GetHead               | Automatically route undefined HEAD requests to GET handlers                     |
 | Heartbeat             | Monitoring endpoint to check the servers pulse                                  |
@@ -330,6 +331,7 @@ with `net/http` can be used with chi's mux.
 | Recoverer             | Gracefully absorb panics and prints the stack trace                             |
 | RequestID             | Injects a request ID into the context of each request                           |
 | RedirectSlashes       | Redirect slashes on routing paths                                               |
+| SetHeader             | Short-hand middleware to set a response header key/value                        |
 | StripSlashes          | Strip slashes on routing paths                                                  |
 | Throttle              | Puts a ceiling on the number of concurrent requests                             |
 | Timeout               | Signals to the request context when the timeout deadline is reached             |
diff --git a/middleware/content_type.go b/middleware/content_type.go
@@ -26,22 +26,19 @@ func AllowContentType(contentTypes ...string) func(next http.Handler) http.Handl
 
 	return func(next http.Handler) http.Handler {
 		fn := func(w http.ResponseWriter, r *http.Request) {
-			s := strings.ToLower(strings.TrimSpace(strings.Split(r.Header.Get("Content-Type"), ";")[0]))
+			s := strings.ToLower(strings.TrimSpace(r.Header.Get("Content-Type")))
+			if i := strings.Index(s, ";"); i > -1 {
+				s = s[0:i]
+			}
 
-			f := false
 			for _, t := range cT {
 				if t == s {
-					f = true
-					break
+					next.ServeHTTP(w, r)
+					return
 				}
 			}
 
-			if !f {
-				w.WriteHeader(http.StatusUnsupportedMediaType)
-				return
-			}
-
-			next.ServeHTTP(w, r)
+			w.WriteHeader(http.StatusUnsupportedMediaType)
 		}
 		return http.HandlerFunc(fn)
 	}
