make allowance for blocking or non-blocking executor calls to Start

Author: gbenroscience

main/main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"fmt"
+	"sync"
 	"time"
 
 	"github.com/gbenroscience/scheduled-executor/utils"
@@ -13,6 +14,9 @@ func timeStampMillis() int {
 
 func main() {
 
+	var wg sync.WaitGroup
+	wg.Add(2)
+
 	totalCount := 0
 	const MAX_CYCLES = 10
 
@@ -24,16 +28,21 @@ func main() {
 		fmt.Printf("sc:---%d.%4stime is %d\n", totalCount, " ", timeStampMillis())
 		if totalCount > MAX_CYCLES {
 			sc.Close()
+			wg.Done()
 		}
-	}, true)
+	}, true, true)
 
 	totalCount1 := 0
 	sc1.Start(func() {
 		totalCount1++
 		fmt.Printf("sc1:---%d.%4stime is %d\n", totalCount1, " ", timeStampMillis())
 		if totalCount1 > MAX_CYCLES {
 			sc1.Close()
+			wg.Done()
 		}
-	}, true)
+	}, true, false)
+
+	wg.Wait()
+	fmt.Println("All tasks completed.")
 
 }

utils/timedexecutor.go

@@ -24,7 +24,13 @@ func NewTimedExecutor(initialDelay time.Duration, delay time.Duration) *Schedule
 	}
 }
 
-func (se *ScheduledExecutor) Start(task func(), runAsync bool) {
+// Start begins the execution of the task at the specified intervals.
+// If runAsync is true, the task will be executed in a separate goroutine.
+// If runAsync is false, the task will block the goroutine until it completes.
+// If startAsync is false, the executor will start asynchronously. So the caller can continue without waiting for the task to start.
+// If startAsync is false, the caller(of Start) will wait until the task starts executing.
+// The task will be stopped when the context is canceled.
+func (se *ScheduledExecutor) Start(task func(), runAsync bool, startAsync bool) {
 	se.ctx, se.cancel = context.WithCancel(context.Background())
 	se.sigs = make(chan os.Signal, 1)
 	signal.Notify(se.sigs, syscall.SIGINT, syscall.SIGTERM)
@@ -61,7 +67,9 @@ func (se *ScheduledExecutor) Start(task func(), runAsync bool) {
 		}
 	}()
 
-	<-se.ctx.Done()
+	if !startAsync {
+		<-se.ctx.Done()
+	}
 }
 
 func (se *ScheduledExecutor) Close() {