feat: workerize external generation

2026-06-05 17:29:08 +08:00
parent 5150925947
commit 8d54ea3374
60 changed files with 5285 additions and 700 deletions
--- a/server-rs/crates/spacetime-client/src/external_generation.rs
+++ b/server-rs/crates/spacetime-client/src/external_generation.rs
@@ -0,0 +1,129 @@
+use super::*;
+use crate::mapper::*;
+
+impl SpacetimeClient {
+    pub async fn enqueue_external_generation_job(
+        &self,
+        input: ExternalGenerationJobEnqueueRecordInput,
+    ) -> Result<ExternalGenerationJobRecord, SpacetimeClientError> {
+        let procedure_input = input.into();
+
+        self.call_after_connect(
+            "enqueue_external_generation_job_and_return",
+            move |connection, sender| {
+                connection
+                    .procedures()
+                    .enqueue_external_generation_job_and_return_then(
+                        procedure_input,
+                        move |_, result| {
+                            let mapped = result
+                                .map_err(SpacetimeClientError::from_sdk_error)
+                                .and_then(map_external_generation_job_procedure_result);
+                            send_once(&sender, mapped);
+                        },
+                    );
+            },
+        )
+        .await
+    }
+
+    pub async fn claim_external_generation_jobs(
+        &self,
+        input: ExternalGenerationJobClaimRecordInput,
+    ) -> Result<Vec<ExternalGenerationJobRecord>, SpacetimeClientError> {
+        let procedure_input = input.into();
+
+        self.call_after_connect(
+            "claim_external_generation_jobs_and_return",
+            move |connection, sender| {
+                connection
+                    .procedures()
+                    .claim_external_generation_jobs_and_return_then(
+                        procedure_input,
+                        move |_, result| {
+                            let mapped = result
+                                .map_err(SpacetimeClientError::from_sdk_error)
+                                .and_then(map_external_generation_job_claim_result);
+                            send_once(&sender, mapped);
+                        },
+                    );
+            },
+        )
+        .await
+    }
+
+    pub async fn complete_external_generation_job(
+        &self,
+        input: ExternalGenerationJobCompleteRecordInput,
+    ) -> Result<ExternalGenerationJobRecord, SpacetimeClientError> {
+        let procedure_input = input.into();
+
+        self.call_after_connect(
+            "complete_external_generation_job_and_return",
+            move |connection, sender| {
+                connection
+                    .procedures()
+                    .complete_external_generation_job_and_return_then(
+                        procedure_input,
+                        move |_, result| {
+                            let mapped = result
+                                .map_err(SpacetimeClientError::from_sdk_error)
+                                .and_then(map_external_generation_job_procedure_result);
+                            send_once(&sender, mapped);
+                        },
+                    );
+            },
+        )
+        .await
+    }
+
+    pub async fn renew_external_generation_job_lease(
+        &self,
+        input: ExternalGenerationJobRenewLeaseRecordInput,
+    ) -> Result<ExternalGenerationJobRecord, SpacetimeClientError> {
+        let procedure_input = input.into();
+
+        self.call_after_connect(
+            "renew_external_generation_job_lease_and_return",
+            move |connection, sender| {
+                connection
+                    .procedures()
+                    .renew_external_generation_job_lease_and_return_then(
+                        procedure_input,
+                        move |_, result| {
+                            let mapped = result
+                                .map_err(SpacetimeClientError::from_sdk_error)
+                                .and_then(map_external_generation_job_procedure_result);
+                            send_once(&sender, mapped);
+                        },
+                    );
+            },
+        )
+        .await
+    }
+
+    pub async fn fail_external_generation_job(
+        &self,
+        input: ExternalGenerationJobFailRecordInput,
+    ) -> Result<ExternalGenerationJobRecord, SpacetimeClientError> {
+        let procedure_input = input.into();
+
+        self.call_after_connect(
+            "fail_external_generation_job_and_return",
+            move |connection, sender| {
+                connection
+                    .procedures()
+                    .fail_external_generation_job_and_return_then(
+                        procedure_input,
+                        move |_, result| {
+                            let mapped = result
+                                .map_err(SpacetimeClientError::from_sdk_error)
+                                .and_then(map_external_generation_job_procedure_result);
+                            send_once(&sender, mapped);
+                        },
+                    );
+            },
+        )
+        .await
+    }
+}