test phase 23 verification script
This commit is contained in:
@@ -191,22 +191,28 @@ public sealed class DataLakeWriterService : BackgroundService
|
||||
}
|
||||
}
|
||||
|
||||
private string ExtractDatePath(string topic, string payload) =>
|
||||
DataLakeEventParser.ExtractDatePath(topic, payload, _kafkaOptions.Topics);
|
||||
|
||||
private async Task<byte[]> BuildParquetAsync(
|
||||
string topic, List<BufferedEvent> events, int partition)
|
||||
{
|
||||
if (topic == _kafkaOptions.Topics.ObservationRecorded)
|
||||
{
|
||||
var rows = events.Select(e => ParseObservationRow(e, partition)).ToList();
|
||||
var rows = events.Select(e => DataLakeEventParser.ParseObservationRow(
|
||||
e.Payload, e.Offset, partition)).ToList();
|
||||
return await ParquetFileBuilder.BuildObservationsAsync(rows);
|
||||
}
|
||||
if (topic == _kafkaOptions.Topics.AlertGenerated)
|
||||
{
|
||||
var rows = events.Select(e => ParseAlertRow(e, partition)).ToList();
|
||||
var rows = events.Select(e => DataLakeEventParser.ParseAlertRow(
|
||||
e.Payload, e.Offset, partition)).ToList();
|
||||
return await ParquetFileBuilder.BuildAlertsAsync(rows);
|
||||
}
|
||||
if (topic == _kafkaOptions.Topics.EncounterStatusChanged)
|
||||
{
|
||||
var rows = events.Select(e => ParseEncounterRow(e, partition)).ToList();
|
||||
var rows = events.Select(e => DataLakeEventParser.ParseEncounterRow(
|
||||
e.Payload, e.Offset, partition)).ToList();
|
||||
return await ParquetFileBuilder.BuildEncountersAsync(rows);
|
||||
}
|
||||
throw new InvalidOperationException($"Unknown topic: {topic}");
|
||||
@@ -233,80 +239,6 @@ public sealed class DataLakeWriterService : BackgroundService
|
||||
return $"{folder}/{key.DatePath}/partition-{key.Partition}-offset-{firstOffset:D10}.parquet";
|
||||
}
|
||||
|
||||
private string ExtractDatePath(string topic, string payload)
|
||||
{
|
||||
try
|
||||
{
|
||||
var doc = JsonDocument.Parse(payload);
|
||||
var ts = topic switch
|
||||
{
|
||||
var t when t == _kafkaOptions.Topics.ObservationRecorded => doc.RootElement.GetProperty("recordedAt").GetDateTimeOffset(),
|
||||
var t when t == _kafkaOptions.Topics.AlertGenerated => doc.RootElement.GetProperty("triggeredAt").GetDateTimeOffset(),
|
||||
var t when t == _kafkaOptions.Topics.EncounterStatusChanged => doc.RootElement.GetProperty("changedAt").GetDateTimeOffset(),
|
||||
_ => DateTimeOffset.UtcNow,
|
||||
};
|
||||
return $"{ts.Year:D4}/{ts.Month:D2}/{ts.Day:D2}";
|
||||
}
|
||||
catch
|
||||
{
|
||||
// Malformed payload: use today so the event is not lost.
|
||||
var now = DateTimeOffset.UtcNow;
|
||||
return $"{now.Year:D4}/{now.Month:D2}/{now.Day:D2}";
|
||||
}
|
||||
}
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
// Payload deserializers — each reads only the fields needed for the Parquet row.
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
private static ObservationRow ParseObservationRow(BufferedEvent e, int partition)
|
||||
{
|
||||
var d = JsonDocument.Parse(e.Payload).RootElement;
|
||||
return new ObservationRow(
|
||||
ObservationId : d.GetProperty("observationId").GetString() ?? "",
|
||||
EncounterId : d.GetProperty("encounterId").GetString() ?? "",
|
||||
PatientId : d.GetProperty("patientId").GetString() ?? "",
|
||||
Mrn : d.TryGetProperty("mrn", out var mrn) ? mrn.GetString() ?? "" : "",
|
||||
ObservationCode : d.GetProperty("observationCode").GetString() ?? "",
|
||||
Value : d.GetProperty("value").GetDouble(),
|
||||
Unit : d.GetProperty("unit").GetString() ?? "",
|
||||
Source : d.GetProperty("source").GetString() ?? "",
|
||||
RecordedAt : d.GetProperty("recordedAt").GetString() ?? "",
|
||||
KafkaPartition : partition,
|
||||
KafkaOffset : e.Offset
|
||||
);
|
||||
}
|
||||
|
||||
private static AlertRow ParseAlertRow(BufferedEvent e, int partition)
|
||||
{
|
||||
var d = JsonDocument.Parse(e.Payload).RootElement;
|
||||
return new AlertRow(
|
||||
AlertId : d.GetProperty("alertId").GetString() ?? "",
|
||||
EncounterId : d.GetProperty("encounterId").GetString() ?? "",
|
||||
PatientId : d.GetProperty("patientId").GetString() ?? "",
|
||||
AlertType : d.GetProperty("alertType").GetString() ?? "",
|
||||
Severity : d.GetProperty("severity").GetString() ?? "",
|
||||
Details : d.TryGetProperty("details", out var det) ? det.GetString() ?? "" : "",
|
||||
TriggeredAt : d.GetProperty("triggeredAt").GetString() ?? "",
|
||||
KafkaPartition : partition,
|
||||
KafkaOffset : e.Offset
|
||||
);
|
||||
}
|
||||
|
||||
private static EncounterStatusRow ParseEncounterRow(BufferedEvent e, int partition)
|
||||
{
|
||||
var d = JsonDocument.Parse(e.Payload).RootElement;
|
||||
return new EncounterStatusRow(
|
||||
EncounterId : d.GetProperty("encounterId").GetString() ?? "",
|
||||
PatientId : d.GetProperty("patientId").GetString() ?? "",
|
||||
PreviousStatus : d.GetProperty("previousStatus").GetString() ?? "",
|
||||
NewStatus : d.GetProperty("newStatus").GetString() ?? "",
|
||||
ChangedAt : d.GetProperty("changedAt").GetString() ?? "",
|
||||
KafkaPartition : partition,
|
||||
KafkaOffset : e.Offset
|
||||
);
|
||||
}
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
// MinIO upload
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
Reference in New Issue
Block a user