| 21 | } |
| 22 | |
| 23 | func (rl *ResponseLogWriter) logCode(code int, status string) { |
| 24 | var codestr string |
| 25 | switch { |
| 26 | case code >= 200 && code < 300: |
| 27 | codestr = color.GreenString("%d %s", code, status) |
| 28 | case code >= 300 && code < 400: |
| 29 | codestr = color.BlueString("%d %s", code, status) |
| 30 | case code >= 400 && code < 500: |
| 31 | codestr = color.YellowString("%d %s", code, status) |
| 32 | case code >= 500 && code < 600: |
| 33 | codestr = color.RedString("%d %s", code, status) |
| 34 | default: |
| 35 | codestr = fmt.Sprintf("%d %s", code, status) |
| 36 | } |
| 37 | cl := rl.Header().Get("content-length") |
| 38 | clstr := "" |
| 39 | if cl != "" { |
| 40 | cli, err := strconv.Atoi(cl) |
| 41 | if err != nil { |
| 42 | rl.Log.Warn("Invalid content-length header") |
| 43 | } else if cli > 0 { |
| 44 | clstr = fmt.Sprintf("%s", humanize.Bytes(uint64(cli))) |
| 45 | } |
| 46 | } |
| 47 | rl.Log.Say("<- %s %s", codestr, clstr) |
| 48 | } |
| 49 | |
| 50 | // Header returns the header map that will be sent by WriteHeader. |
| 51 | // Changing the header after a call to WriteHeader (or Write) has |