more multi-tenancy

backend/src/workers/ocr.rs

@@ -25,6 +25,7 @@ use crate::{
     },
     schema::{document_asset_objects, document_assets, document_versions, documents},
     state::AppState,
+    storage::TenantStorage,
     utils::storage_paths::document_asset_object_prefix,
 };
 
@@ -55,7 +56,12 @@ impl JobHandler for GenerateOcrTextJob {
         JOB_GENERATE_OCR_TEXT
     }
 
-    async fn handle(&self, state: Arc<AppState>, job: crate::models::Job) -> JobExecution {
+    async fn handle(
+        &self,
+        state: Arc<AppState>,
+        job: crate::models::Job,
+        storage: TenantStorage,
+    ) -> JobExecution {
         let payload: OcrPayload = match serde_json::from_value(job.payload.clone()) {
             Ok(payload) => payload,
             Err(err) => {
@@ -92,7 +98,7 @@ impl JobHandler for GenerateOcrTextJob {
             return JobExecution::Success;
         }
 
-        let bytes = match state.storage.get_object(&context.version.s3_key).await {
+        let bytes = match storage.get_object(&context.version.s3_key).await {
             Ok(bytes) => bytes,
             Err(err) => {
                 warn!(job_id = %job.id, error = %err, "failed to fetch document for ocr");
@@ -129,7 +135,7 @@ impl JobHandler for GenerateOcrTextJob {
 
         if context.existing_asset.is_some() {
             for object in &context.existing_objects {
-                if let Err(err) = state.storage.delete_object(&object.s3_key).await {
+                if let Err(err) = storage.delete_object(&object.s3_key).await {
                     warn!(job_id = %job.id, error = %err, s3_key = %object.s3_key, "failed to delete existing ocr asset object");
                 }
             }
@@ -144,8 +150,7 @@ impl JobHandler for GenerateOcrTextJob {
             asset_id,
         );
 
-        if let Err(err) = state
-            .storage
+        if let Err(err) = storage
             .put_object(
                 &s3_key,
                 generation.text.into_bytes(),
@@ -168,11 +173,8 @@ impl JobHandler for GenerateOcrTextJob {
         .await
         {
             Ok(Ok(())) => {
-                if state.config.quickwit_endpoint.is_some() && state.config.quickwit_index.is_some()
-                {
-                    if let Err(err) = enqueue_index_job(&state, &payload) {
-                        warn!(job_id = %job.id, error = %err, "failed to enqueue index job");
-                    }
+                if let Err(err) = enqueue_index_job(&state, &payload) {
+                    warn!(job_id = %job.id, error = %err, "failed to enqueue index job");
                 }
                 JobExecution::Success
             }