changes following review

logzio · Sep 23, 2024 · 3645243 · 3645243
1 parent 055fc41
commit 3645243
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/logzio-lambda-extensions-logs/main.go b/logzio-lambda-extensions-logs/main.go
@@ -122,6 +122,7 @@ func main() {
 		select {
 		case <-ctx.Done():
 			logger.Info(printPrefix, "Received context done event")
+			logsApiLogger.Drain()
 			logsApiAgent.Shutdown()
 			logger.Info(printPrefix, "Exiting")
 			return