log.go 2.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889
  1. /*
  2. * Copyright (c) 2014, Yawning Angel <yawning at torproject dot org>
  3. * All rights reserved.
  4. *
  5. * Redistribution and use in source and binary forms, with or without
  6. * modification, are permitted provided that the following conditions are met:
  7. *
  8. * * Redistributions of source code must retain the above copyright notice,
  9. * this list of conditions and the following disclaimer.
  10. *
  11. * * Redistributions in binary form must reproduce the above copyright notice,
  12. * this list of conditions and the following disclaimer in the documentation
  13. * and/or other materials provided with the distribution.
  14. *
  15. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  16. * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  17. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  18. * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
  19. * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  20. * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  21. * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  22. * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  23. * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  24. * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  25. * POSSIBILITY OF SUCH DAMAGE.
  26. */
  27. package main
  28. import (
  29. "fmt"
  30. "log"
  31. "strings"
  32. )
  33. const (
  34. levelError = iota
  35. levelWarn
  36. levelInfo
  37. )
  38. var logLevel = levelInfo
  39. func noticef(format string, a ...interface{}) {
  40. if enableLogging {
  41. msg := fmt.Sprintf(format, a...)
  42. log.Print("[NOTICE]: " + msg)
  43. }
  44. }
  45. func errorf(format string, a ...interface{}) {
  46. if enableLogging && logLevel >= levelError {
  47. msg := fmt.Sprintf(format, a...)
  48. log.Print("[ERROR]: " + msg)
  49. }
  50. }
  51. func warnf(format string, a ...interface{}) {
  52. if enableLogging && logLevel >= levelWarn {
  53. msg := fmt.Sprintf(format, a...)
  54. log.Print("[WARN]: " + msg)
  55. }
  56. }
  57. func infof(format string, a ...interface{}) {
  58. if enableLogging && logLevel >= levelInfo {
  59. msg := fmt.Sprintf(format, a...)
  60. log.Print("[INFO]: " + msg)
  61. }
  62. }
  63. func setLogLevel(logLevelStr string) error {
  64. switch strings.ToUpper(logLevelStr) {
  65. case "ERROR":
  66. logLevel = levelError
  67. case "WARN":
  68. logLevel = levelWarn
  69. case "INFO":
  70. logLevel = levelInfo
  71. default:
  72. return fmt.Errorf("invalid log level '%s'", logLevelStr)
  73. }
  74. return nil
  75. }