service: add eager jobsv0.3.4

1 files changed, 10 insertions, 3 deletions

diff --git a/service/app_runner.go b/service/app_runner.go
index e1eb98b..0240c27 100644
--- a/service/app_runner.go
+++ b/service/app_runner.go
@@ -19,9 +19,10 @@ type RunnerFunc func(context.Context, *sync.WaitGroup) error
 
 type AppRunner struct {
         Logger         AppRunnerLogger
-        initJobs       []SyncRunnerFunc
-        jobs           []RunnerFunc
-        backgroundJobs []RunnerFunc
+        initJobs       []SyncRunnerFunc // Jobs that run synchronously before other jobs
+        jobs           []RunnerFunc     // Normal jobs
+        backgroundJobs []RunnerFunc     // Jobs that can be disabled
+        eagerJobs      []RunnerFunc     // Jobs that must start immediately
         wg             *sync.WaitGroup
         sigs           chan os.Signal
         ctx            context.Context
@@ -33,6 +34,7 @@ func NewAppRunner(ctx context.Context, logger AppRunnerLogger) *AppRunner {
                 initJobs:       []SyncRunnerFunc{},
                 jobs:           []RunnerFunc{},
                 backgroundJobs: []RunnerFunc{},
+                eagerJobs:      []RunnerFunc{},
                 wg:             &sync.WaitGroup{},
                 sigs:           make(chan os.Signal),
                 Logger:         logger,
@@ -61,6 +63,11 @@ func (r *AppRunner) AddJob(f RunnerFunc) {
         r.jobs = append(r.jobs, f)
 }
 
+func (r *AppRunner) AddJobRunNow(f RunnerFunc) {
+        r.eagerJobs = append(r.eagerJobs, f)
+        r.goRunPanic(f)
+}
+
 func (r *AppRunner) AddJobs(jobs []RunnerFunc) {
         for _, j := range jobs {
                 r.AddJob(j)