Merge pull request #10 from logzio/aws-dev

v1.0.1 - flush buffered logs if exists, before the function run ends

Author: yotamloe

handler/handler.go

@@ -101,6 +101,7 @@ func initLogger(ctx context.Context, request events.APIGatewayProxyRequest) zap.
 	firehoseRequestId := request.Headers["X-Amz-Firehose-Request-Id"]
 	config := zap.NewProductionConfig()
 	config.EncoderConfig.StacktraceKey = "" // to hide stacktrace info
+	config.OutputPaths = []string{"stdout"} // write to stdout
 	config.InitialFields = map[string]interface{}{
 		"aws_account":          account,
 		"lambda_invocation_id": awsRequestId,
@@ -250,6 +251,8 @@ func summaryValuesToMetrics(metricsToSendSlice pdata.InstrumentationLibraryMetri
 }
 func HandleRequest(ctx context.Context, request events.APIGatewayProxyRequest) (events.APIGatewayProxyResponse, error) {
 	log := initLogger(ctx, request)
+	// flush buffered logs if exists, before the function run ends
+	defer log.Sync()
 	metricCount := 0
 	dataPointCount := 0
 	shippingErrors := new(ErrorCollector)