Skip to content

Commit

Permalink
handle asynq logs with slog
Browse files Browse the repository at this point in the history
  • Loading branch information
aradwann committed Jan 7, 2024
1 parent 9a7b622 commit f4b1d7f
Show file tree
Hide file tree
Showing 2 changed files with 62 additions and 0 deletions.
52 changes: 52 additions & 0 deletions worker/logger.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,52 @@
package worker

import (
"context"
"fmt"
"log/slog"
)

type Logger struct{}

func NewLogger() *Logger {
return &Logger{}
}

func (logger *Logger) Print(level slog.Level, args ...interface{}) {
slog.LogAttrs(
context.Background(),
level,
fmt.Sprint(args...),
)

}

// Debug logs a message at Debug level.
func (logger *Logger) Debug(args ...interface{}) {
logger.Print(slog.LevelDebug, args...)
}

// Info logs a message at Info level.
func (logger *Logger) Info(args ...interface{}) {
logger.Print(slog.LevelInfo, args...)

}

// Warn logs a message at Warning level.
func (logger *Logger) Warn(args ...interface{}) {
logger.Print(slog.LevelWarn, args...)

}

// Error logs a message at Error level.
func (logger *Logger) Error(args ...interface{}) {
logger.Print(slog.LevelError, args...)

}

// Fatal logs a message at Fatal level
// and process will exit with status set to 1.
func (logger *Logger) Fatal(args ...interface{}) {
logger.Print(slog.LevelError, args...)

}
10 changes: 10 additions & 0 deletions worker/processor.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@ package worker

import (
"context"
"log/slog"

db "github.com/aradwann/eenergy/db/store"
"github.com/hibiken/asynq"
Expand Down Expand Up @@ -30,6 +31,15 @@ func NewRedisTaskProcessor(redisOpt asynq.RedisClientOpt, store db.Store) TaskPr
QueueDefault: 3,
QueueLow: 1,
},
ErrorHandler: asynq.ErrorHandlerFunc(func(ctx context.Context, task *asynq.Task, err error) {
slog.LogAttrs(ctx,
slog.LevelError,
"process task failed",
slog.String("err", err.Error()),
slog.String("type", task.Type()),
slog.String("payload", string(task.Payload())))
}),
Logger: NewLogger(),
})

return &RedisTaskProcessor{
Expand Down

0 comments on commit f4b1d7f

Please sign in to comment.