mirror of
https://github.com/storytold/artcraft.git
synced 2026-11-05 17:58:55 +00:00
worker details
This commit is contained in:
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;
|
||||
@@ -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,
|
||||
|
||||
|
||||
+6
-12
@@ -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))?;
|
||||
|
||||
Reference in new issue
Block a user