Feature: Task Scheduling configuration during task creation (#337)

* introduce very basic task_scheduling configuration during task creation

* allow node_groups plugin to schedule tasks based on matching plugin config

Author: JannikSt

crates/orchestrator/src/api/routes/heartbeat.rs

@@ -139,9 +139,7 @@ mod tests {
         let task = TaskRequest {
             image: "test".to_string(),
             name: "test".to_string(),
-            command: None,
-            args: None,
-            env_vars: None,
+            ..Default::default()
         };
         app_state.store_context.task_store.add_task(task.into());
 
@@ -171,12 +169,7 @@ mod tests {
         let value = value.unwrap();
         let heartbeat = HeartbeatRequest {
             address: "0x0000000000000000000000000000000000000000".to_string(),
-            task_id: None,
-            task_state: None,
-            metrics: None,
-            version: None,
-            timestamp: None,
-            p2p_id: None,
+            ..Default::default()
         };
         assert_eq!(value, heartbeat);
     }

crates/orchestrator/src/api/routes/task.rs

@@ -16,6 +16,7 @@ async fn get_all_tasks(app_state: Data<AppState>) -> HttpResponse {
 async fn create_task(task: web::Json<TaskRequest>, app_state: Data<AppState>) -> HttpResponse {
     let task = Task::from(task.into_inner());
     let task_store = app_state.store_context.task_store.clone();
+
     task_store.add_task(task.clone());
     HttpResponse::Ok().json(json!({"success": true, "task": task}))
 }
@@ -57,9 +58,7 @@ mod tests {
         let payload = TaskRequest {
             image: "test".to_string(),
             name: "test".to_string(),
-            command: None,
-            args: None,
-            env_vars: None,
+            ..Default::default()
         };
         let req = test::TestRequest::post()
             .uri("/tasks")
@@ -101,9 +100,7 @@ mod tests {
         let task: Task = TaskRequest {
             image: "test".to_string(),
             name: "test".to_string(),
-            command: None,
-            args: None,
-            env_vars: None,
+            ..Default::default()
         }
         .into();
 
@@ -136,9 +133,7 @@ mod tests {
         let task: Task = TaskRequest {
             image: "test".to_string(),
             name: "test".to_string(),
-            command: None,
-            args: None,
-            env_vars: None,
+            ..Default::default()
         }
         .into();
 
@@ -178,9 +173,7 @@ mod tests {
         let task1: Task = TaskRequest {
             image: "test1".to_string(),
             name: "test1".to_string(),
-            command: None,
-            args: None,
-            env_vars: None,
+            ..Default::default()
         }
         .into();
         task_store.add_task(task1.clone());
@@ -192,9 +185,7 @@ mod tests {
         let task2: Task = TaskRequest {
             image: "test2".to_string(),
             name: "test2".to_string(),
-            command: None,
-            args: None,
-            env_vars: None,
+            ..Default::default()
         }
         .into();
         task_store.add_task(task2.clone());
@@ -223,9 +214,7 @@ mod tests {
             let task: Task = TaskRequest {
                 image: format!("test{}", i),
                 name: format!("test{}", i),
-                command: None,
-                args: None,
-                env_vars: None,
+                ..Default::default()
             }
             .into();
             task_store.add_task(task);

crates/orchestrator/src/scheduler/mod.rs

@@ -60,12 +60,9 @@ mod tests {
             id: Uuid::new_v4(),
             image: "image".to_string(),
             name: "name".to_string(),
-            env_vars: None,
-            command: None,
-            args: None,
             state: TaskState::PENDING,
             created_at: 1,
-            updated_at: None,
+            ..Default::default()
         };
 
         state.store_context.task_store.add_task(task.clone());

crates/orchestrator/src/scheduler/plugins/newest_task.rs

@@ -37,23 +37,17 @@ mod tests {
                 id: Uuid::new_v4(),
                 image: "image".to_string(),
                 name: "name".to_string(),
-                env_vars: None,
-                command: None,
-                args: None,
                 state: TaskState::PENDING,
                 created_at: 1,
-                updated_at: None,
+                ..Default::default()
             },
             Task {
                 id: Uuid::new_v4(),
                 image: "image".to_string(),
                 name: "name".to_string(),
-                env_vars: None,
-                command: None,
-                args: None,
                 state: TaskState::PENDING,
                 created_at: 2,
-                updated_at: None,
+                ..Default::default()
             },
         ];