Resolvers for Processing Jobs

Author: iamvigneshwars

.github/workflows/schema.yml

@@ -68,7 +68,7 @@ jobs:
           run: >
             rover subgraph check data-gateway-n63jcf@current
             --schema processed_data.graphql
-            --name processed_data
+            --name processed-data
           env:
             APOLLO_KEY: ${{ secrets.APOLLO_STUDIO }}
 

models/build.rs

@@ -33,6 +33,15 @@ const TABLES_SPECS: &[&Table] = &[
             "automatic",
         ],
     },
+    &Table {
+        name: "ProcessingJobParameter",
+        columns: &[
+            "processingJobParameterId",
+            "processingJobId",
+            "parameterKey",
+            "parameterValue",
+        ],
+    },
     &Table {
         name: "AutoProc",
         columns: &[

processed_data/src/graphql/entities.rs

@@ -2,7 +2,7 @@ use async_graphql::{Enum, SimpleObject};
 use models::{
     auto_proc, auto_proc_integration, auto_proc_program, auto_proc_scaling,
     auto_proc_scaling_statistics, data_collection_file_attachment, processing_job,
-    sea_orm_active_enums::ScalingStatisticsType,
+    processing_job_parameter, sea_orm_active_enums::ScalingStatisticsType,
 };
 
 /// Represents processed image file stored in s3 bucket
@@ -27,7 +27,7 @@ impl From<data_collection_file_attachment::Model> for DataProcessing {
 
 /// Represents a processing job
 #[derive(Clone, Debug, PartialEq, SimpleObject)]
-#[graphql(name = "ProcessingJob", unresolvable)]
+#[graphql(name = "ProcessingJob", unresolvable, complex)]
 pub struct ProcessingJob {
     /// An opaque unique identifier for the processing job
     pub processing_job_id: u32,
@@ -50,6 +50,28 @@ impl From<processing_job::Model> for ProcessingJob {
     }
 }
 
+/// Represents a processing job parameters
+#[derive(Clone, Debug, PartialEq, SimpleObject)]
+#[graphql(name = "ProcessingJobParameter", unresolvable)]
+#[allow(clippy::missing_docs_in_private_items)]
+pub struct ProcessingJobParameter {
+    pub processing_job_parameter_id: u32,
+    pub processing_job_id: Option<u32>,
+    pub parameter_key: Option<String>,
+    pub parameter_value: Option<String>,
+}
+
+impl From<processing_job_parameter::Model> for ProcessingJobParameter {
+    fn from(value: processing_job_parameter::Model) -> Self {
+        Self {
+            processing_job_id: value.processing_job_id,
+            processing_job_parameter_id: value.processing_job_parameter_id,
+            parameter_key: value.parameter_key,
+            parameter_value: value.parameter_value,
+        }
+    }
+}
+
 /// Represents an auto processed job
 #[derive(Clone, Debug, PartialEq, SimpleObject)]
 #[graphql(name = "AutoProc", unresolvable)]

processed_data/src/graphql/mod.rs

@@ -5,8 +5,8 @@ use async_graphql::{
     ComplexObject, Context, EmptyMutation, EmptySubscription, Object, Schema, SchemaBuilder,
 };
 use aws_sdk_s3::presigning::PresigningConfig;
-use entities::{DataCollection, DataProcessing};
-use models::data_collection_file_attachment;
+use entities::{DataCollection, DataProcessing, ProcessingJob, ProcessingJobParameter, AutoProcIntegration};
+use models::{data_collection_file_attachment, processing_job, processing_job_parameter, auto_proc_integration};
 use sea_orm::{ColumnTrait, DatabaseConnection, EntityTrait, QueryFilter};
 use std::time::Duration;
 use url::Url;
@@ -39,6 +39,21 @@ impl DataCollection {
             .map(DataProcessing::from)
             .collect())
     }
+
+    /// Fetched all the processing jobs
+    async fn processing_jobs(
+        &self,
+        ctx: &Context<'_>,
+    ) -> async_graphql::Result<Vec<ProcessingJob>, async_graphql::Error> {
+        let database = ctx.data::<DatabaseConnection>()?;
+        Ok(processing_job::Entity::find()
+            .filter(processing_job::Column::DataCollectionId.eq(self.id))
+            .all(database)
+            .await?
+            .into_iter()
+            .map(ProcessingJob::from)
+            .collect())
+    }
 }
 
 #[ComplexObject]
@@ -60,6 +75,23 @@ impl DataProcessing {
     }
 }
 
+#[ComplexObject]
+impl ProcessingJob {
+    async fn parameters(
+        &self,
+        ctx: &Context<'_>,
+    ) -> async_graphql::Result<Vec<ProcessingJobParameter>> {
+        let database = ctx.data::<DatabaseConnection>()?;
+        Ok(processing_job_parameter::Entity::find()
+            .filter(processing_job_parameter::Column::ProcessingJobId.eq(self.processing_job_id))
+            .all(database)
+            .await?
+            .into_iter()
+            .map(ProcessingJobParameter::from)
+            .collect())
+    }
+}
+
 #[Object]
 impl Query {
     /// Reference datasets resolver for the router