Add solution for 30_task_scheduler exercise

Author: sidharth-chauhan

internal/exercises/templates/30_task_scheduler/task_scheduler.go

@@ -0,0 +1,17 @@
+package taskscheduler
+
+import "time"
+
+// RunAfter runs a task once after the given delay.
+func RunAfter(delay time.Duration, task func()) {
+	time.Sleep(delay)
+	task()
+}
+
+// RunEvery runs a task multiple times with the given delay between each run.
+func RunEvery(delay time.Duration, count int, task func()) {
+	for i := 0; i < count; i++ {
+		time.Sleep(delay)
+		task()
+	}
+}

internal/exercises/templates/30_task_scheduler/task_scheduler_test.go

@@ -0,0 +1,32 @@
+package taskscheduler
+
+import (
+	"testing"
+	"time"
+)
+
+// TestRunAfter checks if the task runs exactly once after a short delay.
+func TestRunAfter(t *testing.T) {
+	counter := 0
+
+	RunAfter(10*time.Millisecond, func() {
+		counter++
+	})
+
+	if counter != 1 {
+		t.Errorf("expected task to run once, got %d", counter)
+	}
+}
+
+// TestRunEvery checks if the task runs the correct number of times.
+func TestRunEvery(t *testing.T) {
+	counter := 0
+
+	RunEvery(5*time.Millisecond, 3, func() {
+		counter++
+	})
+
+	if counter != 3 {
+		t.Errorf("expected task to run 3 times, got %d", counter)
+	}
+}