worker details

This commit is contained in:
Brandon Thomas committed 2023-09-15 04:43:23 -04:00
1 parent 063760663f
commit 4e935a1c2b
5 files changed
+29 -17

No files matched your search

@@ -0,0 +1,7 @@
-- noinspection SqlDialectInspectionForFile
-- noinspection SqlNoDataSourceInspectionForFile
-- noinspection SqlResolveForFile
ALTER TABLE media_files DROP COLUMN generated_by_cluster;
ALTER TABLE media_files DROP COLUMN generated_by_worker;
ALTER TABLE media_files DROP COLUMN is_generated_on_prem;
@@ -0,0 +1,15 @@
-- noinspection SqlDialectInspectionForFile
-- noinspection SqlNoDataSourceInspectionForFile
-- noinspection SqlResolveForFile
ALTER TABLE media_files
ADD COLUMN is_generated_on_prem BOOLEAN NOT NULL DEFAULT FALSE
AFTER maybe_mod_user_token;
ALTER TABLE media_files
ADD COLUMN generated_by_worker VARCHAR(255) DEFAULT NULL
AFTER is_generated_on_prem;
ALTER TABLE media_files
ADD COLUMN generated_by_cluster VARCHAR(255) DEFAULT NULL
AFTER generated_by_worker;
+1 -4
View File
@@ -142,10 +142,6 @@ CREATE TABLE media_files (
-- ========== INFERENCE METADATA, DC, ROUTING, DEBUGGING ==========
-- If true, the request was routed to a special "debug" worker.
-- Or perhaps uploaded with a debug flag.
is_debug BOOLEAN NOT NULL DEFAULT FALSE,
-- For inference and processing, whether we generated this result from
-- our own data center (vs. cloud).
is_generated_on_prem BOOLEAN NOT NULL DEFAULT FALSE,
@@ -154,6 +150,7 @@ CREATE TABLE media_files (
-- Worker hostname (linux hostname, k8s pod name)
generated_by_worker VARCHAR(255) DEFAULT NULL,
-- For inference or processing, which cluster handled the processing.
-- Cluster name (k8s)
generated_by_cluster VARCHAR(255) DEFAULT NULL,
@@ -26,7 +26,6 @@ pub struct InsertArgs<'a> {
pub is_on_prem: bool,
pub worker_hostname: &'a str,
pub worker_cluster: &'a str,
pub is_debug_worker: bool,
}
pub async fn insert_media_file_from_face_animation(
@@ -91,8 +90,11 @@ SET
creator_set_visibility = ?,
maybe_creator_file_synthetic_id = ?,
maybe_creator_category_synthetic_id = ?
maybe_creator_category_synthetic_id = ?,
is_generated_on_prem = ?,
generated_by_worker = ?,
generated_by_cluster = ?
"#,
result_token.as_str(),
@@ -116,21 +118,13 @@ SET
maybe_creator_file_synthetic_id,
maybe_creator_category_synthetic_id,
/*args.is_on_prem,
args.is_on_prem,
args.worker_hostname,
args.worker_cluster,
args.is_debug_worker,*/
args.worker_cluster
)
.execute(&mut transaction)
.await;
/*is_generated_on_prem = ?,
generated_by_worker = ?,
generated_by_cluster = ?,
is_debug_request = ?*/
let record_id = match query_result {
Ok(res) => {
res.last_insert_id()
@@ -211,7 +211,6 @@ pub async fn process_job(args: SadTalkerProcessJobArgs<'_>) -> Result<JobSuccess
is_on_prem: args.job_dependencies.container.is_on_prem,
worker_hostname: &args.job_dependencies.container.hostname,
worker_cluster: &args.job_dependencies.container.cluster_name,
is_debug_worker: args.job_dependencies.worker_details.is_debug_worker
})
.await
.map_err(|e| ProcessSingleJobError::Other(e))?;