fix comment

bitfinity-network · Mar 14, 2024 · ce067e2 · ce067e2
1 parent ce3c6de
commit ce067e2
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 42 deletions.
diff --git a/ic-task-scheduler/src/scheduler.rs b/ic-task-scheduler/src/scheduler.rs
@@ -239,7 +239,16 @@ impl<T: 'static + Task, P: 'static + IterableUnboundedMapStructure<u32, InnerSch
         let time_secs = time_secs();
         let mut lock = self.pending_tasks.lock();
         let key = lock.last_key().map(|val| val + 1).unwrap_or_default();
-        lock.insert(&key, &InnerScheduledTask::waiting(key, task, time_secs));
+        lock.insert(
+            &key,
+            &InnerScheduledTask::with_status(
+                key,
+                task,
+                TaskStatus::Waiting {
+                    timestamp_secs: time_secs,
+                },
+            ),
+        );
         key
     }
 
@@ -254,7 +263,16 @@ impl<T: 'static + Task, P: 'static + IterableUnboundedMapStructure<u32, InnerSch
 
         let mut keys = Vec::with_capacity(tasks.len());
         for task in tasks {
-            lock.insert(&key, &InnerScheduledTask::waiting(key, task, time_secs));
+            lock.insert(
+                &key,
+                &InnerScheduledTask::with_status(
+                    key,
+                    task,
+                    TaskStatus::Waiting {
+                        timestamp_secs: time_secs,
+                    },
+                ),
+            );
             keys.push(key);
             key += 1;
         }

diff --git a/ic-task-scheduler/src/task.rs b/ic-task-scheduler/src/task.rs
@@ -70,46 +70,6 @@ impl<T: Task> InnerScheduledTask<T> {
         }
     }
 
-    /// Creates a new InnerScheduledTask with Waiting status
-    pub fn waiting(id: u32, task: ScheduledTask<T>, timestamp_secs: u64) -> Self {
-        Self {
-            id,
-            task: task.task,
-            options: task.options,
-            status: TaskStatus::Waiting { timestamp_secs },
-        }
-    }
-
-    /// Creates a new InnerScheduledTask with Complete status
-    pub fn completed(id: u32, task: ScheduledTask<T>, timestamp_secs: u64) -> Self {
-        Self {
-            id,
-            task: task.task,
-            options: task.options,
-            status: TaskStatus::Completed { timestamp_secs },
-        }
-    }
-
-    /// Creates a new InnerScheduledTask with TimeoutOrPanic status
-    pub fn timeout_or_panic(id: u32, task: ScheduledTask<T>, timestamp_secs: u64) -> Self {
-        Self {
-            id,
-            task: task.task,
-            options: task.options,
-            status: TaskStatus::TimeoutOrPanic { timestamp_secs },
-        }
-    }
-
-    /// Creates a new InnerScheduledTask with Running status
-    pub fn running(id: u32, task: ScheduledTask<T>, timestamp_secs: u64) -> Self {
-        Self {
-            id,
-            task: task.task,
-            options: task.options,
-            status: TaskStatus::Running { timestamp_secs },
-        }
-    }
-
     /// Returs the status of the task
     pub fn status(&self) -> &TaskStatus {
         &self.status