log.go 1.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697
  1. package main
  2. import (
  3. "net/http"
  4. logrus "github.com/sirupsen/logrus"
  5. )
  6. func initLog() {
  7. logFormat := "pretty"
  8. strEnvConfig(&logFormat, "IMGPROXY_LOG_FORMAT")
  9. switch logFormat {
  10. case "structured":
  11. logrus.SetFormatter(&logStructuredFormatter{})
  12. case "json":
  13. logrus.SetFormatter(&logrus.JSONFormatter{})
  14. default:
  15. logrus.SetFormatter(newLogPrettyFormatter())
  16. }
  17. logrus.SetLevel(logrus.DebugLevel)
  18. if isSyslogEnabled() {
  19. slHook, err := newSyslogHook()
  20. if err != nil {
  21. logFatal("Unable to connect to local syslog daemon")
  22. }
  23. logrus.AddHook(slHook)
  24. }
  25. }
  26. func logRequest(reqID string, r *http.Request) {
  27. path := r.URL.RequestURI()
  28. logrus.WithFields(logrus.Fields{
  29. "request_id": reqID,
  30. "method": r.Method,
  31. }).Infof("Started %s", path)
  32. }
  33. func logResponse(reqID string, r *http.Request, status int, err *imgproxyError, imageURL *string, po *processingOptions) {
  34. var level logrus.Level
  35. switch {
  36. case status >= 500:
  37. level = logrus.ErrorLevel
  38. case status >= 400:
  39. level = logrus.WarnLevel
  40. default:
  41. level = logrus.InfoLevel
  42. }
  43. fields := logrus.Fields{
  44. "request_id": reqID,
  45. "method": r.Method,
  46. "status": status,
  47. }
  48. if err != nil {
  49. fields["error"] = err
  50. if stack := err.FormatStack(); len(stack) > 0 {
  51. fields["stack"] = stack
  52. }
  53. }
  54. if imageURL != nil {
  55. fields["image_url"] = *imageURL
  56. }
  57. if po != nil {
  58. fields["processing_options"] = po
  59. }
  60. logrus.WithFields(fields).Logf(
  61. level,
  62. "Completed in %s %s", getTimerSince(r.Context()), r.URL.RequestURI(),
  63. )
  64. }
  65. func logNotice(f string, args ...interface{}) {
  66. logrus.Infof(f, args...)
  67. }
  68. func logWarning(f string, args ...interface{}) {
  69. logrus.Warnf(f, args...)
  70. }
  71. func logFatal(f string, args ...interface{}) {
  72. logrus.Fatalf(f, args...)
  73. }
  74. func logDebug(f string, args ...interface{}) {
  75. logrus.Debugf(f, args...)
  76. }