@@ -34,6 +34,7 @@ type LambdaExtensionConfig struct {
34
34
MaxDataPayloadSize int
35
35
LambdaRegion string
36
36
SourceCategoryOverride string
37
+ EnhanceJsonLogs bool
37
38
}
38
39
39
40
var defaultLogTypes = []string{"platform", "function"}
@@ -73,6 +74,7 @@ func (cfg *LambdaExtensionConfig) setDefaults() {
73
74
maxConcurrentRequests := os.Getenv("SUMO_MAX_CONCURRENT_REQUESTS")
74
75
enableFailover := os.Getenv("SUMO_ENABLE_FAILOVER")
75
76
logTypes := os.Getenv("SUMO_LOG_TYPES")
77
+ enhanceJsonLogs := os.Getenv("SUMO_ENHANCE_JSON_LOGS")
76
78
77
79
if numRetry == "" {
78
80
cfg.NumRetry = 3
@@ -99,9 +101,11 @@ func (cfg *LambdaExtensionConfig) setDefaults() {
99
101
cfg.LogTypes = strings.Split(logTypes, ",")
100
102
}
101
103
if retrySleepTime == "" {
102
- cfg.RetrySleepTime = 300 * time.Millisecond
104
+ cfg.RetrySleepTime = 300 * time.Millisecond
105
+ }
106
+ if enhanceJsonLogs == "" {
107
+ cfg.EnhanceJsonLogs = true
103
108
}
104
-
105
109
}
106
110
107
111
func (cfg *LambdaExtensionConfig) validateConfig() error {
@@ -111,6 +115,7 @@ func (cfg *LambdaExtensionConfig) validateConfig() error {
111
115
maxConcurrentRequests := os.Getenv("SUMO_MAX_CONCURRENT_REQUESTS")
112
116
enableFailover := os.Getenv("SUMO_ENABLE_FAILOVER")
113
117
retrySleepTime := os.Getenv("SUMO_RETRY_SLEEP_TIME_MS")
118
+ enhanceJsonLogs := os.Getenv("SUMO_ENHANCE_JSON_LOGS")
114
119
115
120
var allErrors []string
116
121
var err error
@@ -189,6 +194,13 @@ func (cfg *LambdaExtensionConfig) validateConfig() error {
189
194
190
195
}
191
196
197
+ if enhanceJsonLogs != "" {
198
+ cfg.EnhanceJsonLogs, err = strconv.ParseBool(enhanceJsonLogs)
199
+ if err != nil {
200
+ allErrors = append(allErrors, fmt.Sprintf("Unable to parse SUMO_ENHANCE_JSON_LOGS: %v", err))
201
+ }
202
+ }
203
+
192
204
// test valid log format type
193
205
for _, logType := range cfg.LogTypes {
194
206
if !utils.StringInSlice(strings.TrimSpace(logType), validLogTypes) {
0 commit comments