feat(log): Add set logger to context function and refactor

pkg/s3-proxy/log/contextkey.go

@@ -1,8 +1,22 @@
 package log
 
+import "context"
+
 // contextKey is a value for use with context.WithValue. It's used as
 // a pointer so it fits in an interface{} without allocation. This technique
 // for defining context keys was copied from Go 1.7's new use of context in net/http.
 type contextKey struct {
 	name string
 }
+
+var loggerContextKey = &contextKey{name: "LOGGER_CONTEXT_KEY"}
+
+func GetLoggerFromContext(ctx context.Context) Logger {
+	res, _ := ctx.Value(loggerContextKey).(Logger)
+
+	return res
+}
+
+func SetLoggerInContext(ctx context.Context, logger Logger) context.Context {
+	return context.WithValue(ctx, loggerContextKey, logger)
+}

pkg/s3-proxy/log/log.go

@@ -1,13 +1,9 @@
 package log
 
 import (
-	"context"
-
 	"github.com/sirupsen/logrus"
 )
 
-var loggerContextKey = &contextKey{name: "LOGGER_CONTEXT_KEY"}
-
 type Logger interface {
 	Configure(level string, format string, filePath string) error
 
@@ -61,9 +57,3 @@ func NewLogger() Logger {
 		FieldLogger: logrus.New(),
 	}
 }
-
-func GetLoggerFromContext(ctx context.Context) Logger {
-	res, _ := ctx.Value(loggerContextKey).(Logger)
-
-	return res
-}