From 5a4c5034a3f8e9e3a8a2d3671ae8207528ec8b3c Mon Sep 17 00:00:00 2001
From: Kinesin Data Technologies Incorporated
 <93931750+kinesintech@users.noreply.github.com>
Date: Wed, 5 Oct 2022 09:25:11 -0300
Subject: [PATCH] Agent compiling

---
 src/bin/agent/config.rs         | 8 ++++----
 src/bin/agent/main.rs           | 6 +++---
 src/executors/agent_executor.rs | 6 +++---
 3 files changed, 10 insertions(+), 10 deletions(-)

diff --git a/src/bin/agent/config.rs b/src/bin/agent/config.rs
index 7857451..876b8c9 100644
--- a/src/bin/agent/config.rs
+++ b/src/bin/agent/config.rs
@@ -51,7 +51,7 @@ pub struct GlobalConfig {
     pub ip: String,
     pub port: u32,
     pub resources: TaskResources,
-    pub tracker: mpsc::UnboundedSender<TrackerMessage>,
+    pub storage: mpsc::UnboundedSender<StorageMessage>,
     pub executor: mpsc::UnboundedSender<ExecutorMessage>,
 }
 
@@ -66,14 +66,14 @@ impl GlobalConfig {
         local_executor::start(*workers as usize, exe_rx);
 
         // Tracker
-        let (tracker, trx) = mpsc::unbounded_channel();
-        noop_tracker::start(trx);
+        let (storage, trx) = mpsc::unbounded_channel();
+        waterfall::storage::noop::start(trx);
 
         GlobalConfig {
             ip: spec.ip.clone(),
             port: spec.port,
             resources: spec.resources.clone(),
-            tracker,
+            storage,
             executor,
         }
     }
diff --git a/src/bin/agent/main.rs b/src/bin/agent/main.rs
index d1b5589..5fdc378 100644
--- a/src/bin/agent/main.rs
+++ b/src/bin/agent/main.rs
@@ -27,16 +27,16 @@ async fn submit_task(
 ) -> impl Responder {
     let (response, rx) = oneshot::channel();
 
-    let submission = data.into_inner();
-
-    let trx = data.tracker.clone();
+    let submission = details.into_inner();
 
+    let (_, kill) = oneshot::channel();
     data.executor
         .send(ExecutorMessage::ExecuteTask {
             details: submission.details,
             output_options: submission.output_options,
             varmap: submission.varmap,
             response,
+            kill,
         })
         .unwrap();
 
diff --git a/src/executors/agent_executor.rs b/src/executors/agent_executor.rs
index 07dacb2..a9a4b04 100644
--- a/src/executors/agent_executor.rs
+++ b/src/executors/agent_executor.rs
@@ -105,9 +105,9 @@ fn validate_task(details: &TaskDetails, max_capacities: &[TaskResources]) -> Res
 /// Contains specifics on how to run a local task
 #[derive(Serialize, Deserialize, Clone, Debug)]
 pub struct TaskSubmission {
-    details: TaskDetails,
-    varmap: VarMap,
-    output_options: TaskOutputOptions,
+    pub details: TaskDetails,
+    pub varmap: VarMap,
+    pub output_options: TaskOutputOptions,
 }
 
 async fn submit_task(