logger.go 2.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798
  1. package jobs
  2. import (
  3. "IotAdmin/core/logger"
  4. "IotAdmin/core/sdk/config"
  5. cLog "IotAdmin/core/sdk/pkg/logger"
  6. "IotAdmin/core/tools/writer"
  7. "fmt"
  8. "log"
  9. "sync"
  10. )
  11. var jobLogger *JobLogger
  12. var once sync.Once
  13. type JobLogger struct {
  14. *logger.Helper
  15. }
  16. func init() {
  17. GetJobLogger()
  18. }
  19. func GetJobLogger() *JobLogger {
  20. once.Do(func() {
  21. loggerHelper := setupJobLogger()
  22. jobLogger = &JobLogger{
  23. loggerHelper,
  24. }
  25. })
  26. return jobLogger
  27. }
  28. // setupJobLogger 设置设备日志组件
  29. func setupJobLogger() *logger.Helper {
  30. path := getJobLogPath()
  31. logCap := config.LoggerConfig.Cap
  32. if logCap == 0 {
  33. logCap = 1024
  34. }
  35. output, err := writer.NewFileWriter(
  36. writer.WithPath(path),
  37. writer.WithCap(logCap<<10),
  38. writer.WithSuffix("log"),
  39. )
  40. if err != nil {
  41. log.Printf("job logger setup error: %s \r\n", err.Error())
  42. }
  43. level := logger.DebugLevel
  44. if config.ApplicationConfig.Mode == "prod" {
  45. level = logger.InfoLevel
  46. }
  47. return logger.NewHelper(logger.NewLogger(logger.WithLevel(level), logger.WithOutput(output), logger.WithName("JOB")))
  48. }
  49. func getJobLogPath() string {
  50. path := config.LoggerConfig.Path
  51. if path == "" {
  52. path = "_logs/"
  53. } else if path[len(path)-1] != '/' {
  54. path += "/"
  55. }
  56. path += "jobs"
  57. return path
  58. }
  59. func CleanJobLog(days int) error {
  60. path := getJobLogPath()
  61. return cLog.DeleteLogsOlderThan(path, "log", days)
  62. }
  63. func (j *JobLogger) Info(msg string, args ...interface{}) {
  64. msg = "[CRON]" + msg
  65. if len(args) > 0 {
  66. for _, v := range args {
  67. msg += fmt.Sprintf(" %v", v)
  68. }
  69. }
  70. j.Helper.Info(msg)
  71. }
  72. func (j *JobLogger) Infof(msg string, args ...interface{}) {
  73. msg = "[JobCore]" + msg
  74. j.Helper.Infof(msg, args...)
  75. }
  76. func (j *JobLogger) Error(err error, msg string, args ...interface{}) {
  77. msg = "[CRON]" + msg
  78. if len(args) > 0 {
  79. for _, v := range args {
  80. msg += fmt.Sprintf("%v ", v)
  81. }
  82. }
  83. j.Helper.Errorf("%s; ERROR: %s", msg, err.Error())
  84. }
  85. func (j *JobLogger) Errorf(msg string, args ...interface{}) {
  86. msg = "[JobCore]" + msg
  87. j.Helper.Errorf(msg, args...)
  88. }