Remove TaskWrapper in favor of directly using PhantomData (#21)

* Remove TaskWrapper in favor of directly using PhantomData

* Remove Arc

Author: Aaron1011

src/client.rs

@@ -2,11 +2,12 @@ use serde::Serialize;
 use serde_json::Value as JsonValue;
 use sqlx::{Executor, PgPool, Postgres};
 use std::collections::HashMap;
+use std::marker::PhantomData;
 use std::sync::Arc;
 use tokio::sync::RwLock;
 use uuid::Uuid;
 
-use crate::task::{Task, TaskRegistry, TaskWrapper};
+use crate::task::{Task, TaskRegistry};
 use crate::types::{
     CancellationPolicy, RetryStrategy, SpawnOptions, SpawnResult, SpawnResultRow, WorkerOptions,
 };
@@ -261,10 +262,7 @@ where
     /// Register a task type. Required before spawning or processing.
     pub async fn register<T: Task<State>>(&self) -> &Self {
         let mut registry = self.registry.write().await;
-        registry.insert(
-            T::NAME.to_string(),
-            Arc::new(TaskWrapper::<T, State>::new()),
-        );
+        registry.insert(T::NAME.to_string(), &PhantomData::<T>);
         self
     }
 

src/task.rs

@@ -1,7 +1,7 @@
 use async_trait::async_trait;
 use serde::{Serialize, de::DeserializeOwned};
 use serde_json::Value as JsonValue;
-use std::sync::Arc;
+use std::marker::PhantomData;
 
 use crate::context::TaskContext;
 use crate::error::{TaskError, TaskResult};
@@ -110,34 +110,8 @@ where
     ) -> Result<JsonValue, TaskError>;
 }
 
-/// Wrapper that implements ErasedTask for any Task type
-pub struct TaskWrapper<T, State>(std::marker::PhantomData<(T, State)>)
-where
-    T: Task<State>,
-    State: Clone + Send + Sync + 'static;
-
-impl<T, State> TaskWrapper<T, State>
-where
-    T: Task<State>,
-    State: Clone + Send + Sync + 'static,
-{
-    pub fn new() -> Self {
-        Self(std::marker::PhantomData)
-    }
-}
-
-impl<T, State> Default for TaskWrapper<T, State>
-where
-    T: Task<State>,
-    State: Clone + Send + Sync + 'static,
-{
-    fn default() -> Self {
-        Self::new()
-    }
-}
-
 #[async_trait]
-impl<T, State> ErasedTask<State> for TaskWrapper<T, State>
+impl<T, State> ErasedTask<State> for PhantomData<T>
 where
     T: Task<State>,
     State: Clone + Send + Sync + 'static,
@@ -159,4 +133,4 @@ where
 }
 
 /// Type alias for the task registry
-pub type TaskRegistry<State> = std::collections::HashMap<String, Arc<dyn ErasedTask<State>>>;
+pub type TaskRegistry<State> = std::collections::HashMap<String, &'static dyn ErasedTask<State>>;

src/worker.rs

@@ -261,7 +261,7 @@ impl Worker {
         // Look up handler
         let registry = registry.read().await;
         let handler = match registry.get(&task.task_name) {
-            Some(h) => h.clone(),
+            Some(h) => *h,
             None => {
                 tracing::error!("Unknown task: {}", task.task_name);
                 Self::fail_run(
@@ -319,6 +319,7 @@ impl Worker {
                                 deadline = Instant::now();
                                 warn_fired = false;
                             }
+
                         }
 
                         _ = sleep_until(warn_at), if !warn_fired => {