extract loop logic into handleKinesisRecord

Author: ndakkoune

aws/logs_monitoring_go/internal/parsing/kinesis.go

@@ -24,21 +24,25 @@ func HandleKinesis(ctx context.Context, event json.RawMessage, cfg *config.Confi
 	}
 
 	for _, record := range kinesisEvent.Records {
-		cwEvent := events.CloudwatchLogsEvent{
-			AWSLogs: events.CloudwatchLogsRawData{
-				Data: base64.StdEncoding.EncodeToString(record.Kinesis.Data),
-			},
-		}
-
-		cwRaw, err := json.Marshal(cwEvent)
-		if err != nil {
-			return fmt.Errorf("marshal cloudwatch event from kinesis: %w", err)
-		}
-
-		if err := HandleCloudwatchLogs(ctx, cwRaw, cfg, out); err != nil {
+		if err := handleKinesisRecord(ctx, record, cfg, out); err != nil {
 			slog.WarnContext(ctx, "skipping kinesis record", "error", err)
 			continue
 		}
 	}
 	return nil
 }
+
+func handleKinesisRecord(ctx context.Context, record events.KinesisEventRecord, cfg *config.Config, out chan<- model.CloudwatchLogEntry) error {
+	cwEvent := events.CloudwatchLogsEvent{
+		AWSLogs: events.CloudwatchLogsRawData{
+			Data: base64.StdEncoding.EncodeToString(record.Kinesis.Data),
+		},
+	}
+
+	cwRaw, err := json.Marshal(cwEvent)
+	if err != nil {
+		return fmt.Errorf("marshal cloudwatch event from kinesis: %w", err)
+	}
+
+	return HandleCloudwatchLogs(ctx, cwRaw, cfg, out)
+}