feat(tasks): tasks can have IDs

handler.go

@@ -2,8 +2,18 @@ package yotei
 
 import "context"
 
+// Handler determines an interface of something
+// that can be handled.
+type Handler interface {
+	// Handle is the callback to execute the task action.
+	Handle(context.Context) Action
+}
+
+// HandlerFunc is a simple type to quickly transform a func
+// into a [Handler].
 type HandlerFunc func(context.Context) Action
 
+// Hande runs the action
 func (handler HandlerFunc) Handle(ctx context.Context) Action {
 	return handler(ctx)
 }

scheduler.go

@@ -249,6 +249,13 @@ func (scheduler *Scheduler) IsRunning() bool {
 	return scheduler.ctx != nil
 }
 
+func (scheduler *Scheduler) FindTaskByID(id string) (*Task, bool) {
+	scheduler.mutex.Lock()
+	defer scheduler.mutex.Unlock()
+
+	return scheduler.tasks.FindByID(id)
+}
+
 // String returns a string representation of a scheduler.
 func (scheduler *Scheduler) String() string {
 	scheduler.mutex.Lock()

task.go

@@ -7,20 +7,14 @@ import (
 	"time"
 )
 
-// Handler determines an interface of something
-// that can be handled.
-type Handler interface {
-	// Handle is the callback to execute the task action.
-	Handle(context.Context) Action
-}
-
 // Task is the the executionable action in the [Scheduler].
 //
 // A task must:
 //   - Be handlable
 //   - Have a weight
 //   - Have a duration
 //   - Be sequential or concurrent
+//   - Have an id
 //
 // Use [NewTask] to create a new task.
 type Task struct {
@@ -29,6 +23,7 @@ type Task struct {
 	duration   atomic.Int64
 	locked     atomic.Bool
 	concurrent atomic.Bool
+	id         atomic.Value
 }
 
 // A list of actionable tasks
@@ -59,6 +54,16 @@ func NewTask(handler Handler) *Task {
 	return task
 }
 
+func (task *Task) Idenfified(id string) *Task {
+	task.id.Store(id)
+
+	return task
+}
+
+func (task *Task) ID() string {
+	return task.id.Load().(string)
+}
+
 func (task *Task) Lock() {
 	task.locked.Store(true)
 }
@@ -102,6 +107,10 @@ func (task *Task) Weight() uint64 {
 }
 
 func (task *Task) Handle(ctx context.Context) Action {
+	if task.handler == nil {
+		panic("no task handler defined. please ensure the task handler is not nil")
+	}
+
 	return task.handler.Handle(ctx)
 }
 
@@ -153,3 +162,13 @@ func (tasks Tasks) Locked() Tasks {
 
 	return locked
 }
+
+func (tasks Tasks) FindByID(id string) (*Task, bool) {
+	for _, task := range tasks {
+		if task.ID() == id {
+			return task, true
+		}
+	}
+
+	return nil, false
+}