From 956e44a4526f91b498f1fbe7876eb8409ecfc5de Mon Sep 17 00:00:00 2001 From: Brandon Thomas Date: Tue, 14 Jun 2022 00:23:05 -0400 Subject: [PATCH] report worker hostname for tts results --- .../src/queries/tts/tts_results/query_tts_result.rs | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/crates/lib/database_queries/src/queries/tts/tts_results/query_tts_result.rs b/crates/lib/database_queries/src/queries/tts/tts_results/query_tts_result.rs index 958145448b..af49aeb064 100644 --- a/crates/lib/database_queries/src/queries/tts/tts_results/query_tts_result.rs +++ b/crates/lib/database_queries/src/queries/tts/tts_results/query_tts_result.rs @@ -36,6 +36,9 @@ pub struct TtsResultRecordForResponse { pub creator_set_visibility: RecordVisibility, + // Worker hostname that generated the audio. Has the value "unknown" if null. + pub generated_by_worker: String, + pub file_size_bytes: u32, pub duration_millis: u32, @@ -86,6 +89,8 @@ pub struct TtsResultRecordRaw { pub creator_set_visibility: String, + pub generated_by_worker: Option, + pub file_size_bytes: i32, pub duration_millis: i32, @@ -163,6 +168,8 @@ pub async fn select_tts_result_by_token( creator_set_visibility: RecordVisibility::from_str(&ir.creator_set_visibility) .unwrap_or(RecordVisibility::Public), + generated_by_worker: ir.generated_by_worker.unwrap_or("unknown".to_string()), + file_size_bytes: if ir.file_size_bytes > 0 { ir.file_size_bytes as u32 } else { 0 }, duration_millis: if ir.duration_millis > 0 { ir.duration_millis as u32 } else { 0 }, @@ -217,6 +224,8 @@ SELECT tts_results.creator_set_visibility, + tts_results.generated_by_worker, + tts_results.file_size_bytes, tts_results.duration_millis, tts_results.created_at, @@ -276,6 +285,8 @@ SELECT tts_results.creator_set_visibility, + tts_results.generated_by_worker, + tts_results.file_size_bytes, tts_results.duration_millis, tts_results.created_at,