Add MustRegister func

revenkroz · May 18, 2024 · 19ab130 · 19ab130
1 parent 5ec29e8
commit 19ab130
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/runner/runner.go b/runner/runner.go
@@ -31,6 +31,12 @@ func (r *Runner) Use(opts ...Option) {
 	}
 }
 
+func (r *Runner) MustRegister(handler CommandHandler) {
+	if err := r.Register(handler); err != nil {
+		panic(err)
+	}
+}
+
 func (r *Runner) Register(handler CommandHandler) error {
 	r.lock.Lock()
 	defer r.lock.Unlock()