stop exposing LogFormat and LogLevel from lambdacontext

Author: anzheyazzz

lambdacontext/context.go

@@ -15,12 +15,6 @@ import (
 	"strconv"
 )
 
-// LogFormat is the log format, either TEXT or JSON (from AWS_LAMBDA_LOG_FORMAT)
-var LogFormat string
-
-// LogLevel is the log level for structured logging (from AWS_LAMBDA_LOG_LEVEL). Only available when LogFormat is JSON
-var LogLevel string
-
 // LogGroupName is the name of the log group that contains the log streams of the current Lambda Function
 var LogGroupName string
 
@@ -39,8 +33,6 @@ var FunctionVersion string
 var maxConcurrency int
 
 func init() {
-	LogFormat = os.Getenv("AWS_LAMBDA_LOG_FORMAT")
-	LogLevel = os.Getenv("AWS_LAMBDA_LOG_LEVEL")
 	LogGroupName = os.Getenv("AWS_LAMBDA_LOG_GROUP_NAME")
 	LogStreamName = os.Getenv("AWS_LAMBDA_LOG_STREAM_NAME")
 	FunctionName = os.Getenv("AWS_LAMBDA_FUNCTION_NAME")

lambdacontext/logger.go

@@ -11,6 +11,12 @@ import (
 	"os"
 )
 
+// logFormat is the log format from AWS_LAMBDA_LOG_FORMAT (TEXT or JSON)
+var logFormat = os.Getenv("AWS_LAMBDA_LOG_FORMAT")
+
+// logLevel is the log level from AWS_LAMBDA_LOG_LEVEL
+var logLevel = os.Getenv("AWS_LAMBDA_LOG_LEVEL")
+
 // field represents a Lambda context field to include in log records.
 type field struct {
 	key   string
@@ -58,7 +64,7 @@ func NewLogHandler(opts ...LogOption) slog.Handler {
 	}
 
 	var h slog.Handler
-	if LogFormat == "JSON" {
+	if logFormat == "JSON" {
 		h = slog.NewJSONHandler(os.Stdout, handlerOpts)
 	} else {
 		h = slog.NewTextHandler(os.Stdout, handlerOpts)
@@ -130,7 +136,7 @@ func (h *lambdaHandler) WithGroup(name string) slog.Handler {
 }
 
 func parseLogLevel() slog.Level {
-	switch LogLevel {
+	switch logLevel {
 	case "DEBUG":
 		return slog.LevelDebug
 	case "INFO":

lambdacontext/logger_test.go

@@ -81,7 +81,7 @@ func TestParseLogLevel(t *testing.T) {
 
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
-			LogLevel = tt.input
+			logLevel = tt.input
 			result := parseLogLevel()
 			assert.Equal(t, tt.expected, result)
 		})
@@ -386,17 +386,11 @@ func TestWithTenantID(t *testing.T) {
 }
 
 func TestNewLogger(t *testing.T) {
-	LogFormat = "JSON"
-	LogLevel = "INFO"
-
 	logger := NewLogger()
 	assert.NotNil(t, logger)
 }
 
 func TestNewLogHandler(t *testing.T) {
-	LogFormat = "JSON"
-	LogLevel = "INFO"
-
 	handler := NewLogHandler()
 	assert.NotNil(t, handler)