Add a --crash-on-error flag to crash whenever an actor returns err

environment.go

@@ -42,6 +42,7 @@ type environment struct {
 		configFile   string
 		verbose      bool
 		crash        bool
+		crashOnError bool
 	}
 
 	// running stores whether the environment is currently running.
@@ -111,7 +112,12 @@ func (this *environment) Add(ctx context.Context, actors ...Actor) error {
 		}
 	}
 	err := this.initializeActors(ctx, initializable...)
-	if err != nil { return err }
+	if err != nil {
+		if this.flags.crashOnError {
+			panic(fmt.Sprint(err))
+		}
+		return err
+	}
 	for _, actor := range actors {
 		if actor, ok := actor.(Configurable); ok {
 			err := actor.Configure(this.conf)
@@ -137,11 +143,13 @@ func (this *environment) Del(ctx context.Context, actors ...Actor) error {
 	channels := []<- chan struct { } { }
 	for _, actor := range actors {
 		info := this.info(actor)
-		if info.done != nil {
+		if info.stopped != nil {
 			channels = append(channels, info.stopped)
 		}
+		info.done()
 	}
 	for _, channel := range channels {
+		if channel == nil { continue }
 		select {
 		case <- channel:
 		case <- ctx.Done():
@@ -244,13 +252,20 @@ func (this *environment) run(actor Actor) {
 			}
 		} else {
 			log.Printf("!!! [%s] stopped with error: %v", typ, stopErr)
+			if this.flags.crashOnError {
+				panic(fmt.Sprint(stopErr))
+			}
 		}
 	}()
 
 	// contains context information
 	info := this.info(actor)
 	ctx := info.ctx
-	defer close(info.stopped)
+	defer func() {
+		if info.stopped != nil {
+			close(info.stopped)
+		}
+	}()
 	
 	switch actor := actor.(type) {
 	case Runnable:
@@ -305,11 +320,17 @@ func (this *environment) runRunnable(ctx context.Context, actor Runnable) (stopE
 		} else {
 			// failure
 			log.Printf("XXX [%s] failed: %v", typ, err)
+			if this.flags.crashOnError {
+				panic(fmt.Sprint(err))
+			}
 		}
 
 		// restart logic
 		if time.Since(lastStart) < restartThreshold {
 			log.Printf("!!! [%s] failed too soon, restarting in %v", typ, restartInterval)
+			if this.flags.crashOnError {
+				panic("failed too soon")
+			}
 			timer := time.NewTimer(restartInterval)
 			select {
 			case <- timer.C:
@@ -335,6 +356,9 @@ func (this *environment) runRunnable(ctx context.Context, actor Runnable) (stopE
 				err := actor.Reset(ctx)
 				if err != nil {
 					log.Printf("XXX [%s] failed to reset", typ)
+					if this.flags.crashOnError {
+						panic("failed to reset")
+					}
 				}
 			}()
 			if this.Verb() { log.Printf(".// [%s] reset", typ) }

phases.go

@@ -6,6 +6,7 @@ import "log"
 import "io/fs"
 import "errors"
 import "context"
+import "runtime"
 import "strings"
 import "path/filepath"
 import "git.tebibyte.media/sashakoshka/go-cli"
@@ -18,13 +19,14 @@ func (this *environment) phase10FlagParsing() bool {
 		name: this.name,
 		description: this.description,
 	}
-	flagHelp          := set.Flag('h', "help",          "Display usage information and exit", nil)
+	flagHelp          := set.Flag('h', "help",           "Display usage information and exit", nil)
-	flagPidFile       := set.Flag('p', "pid-file",      "Write the PID to the specified file", cli.ValString)
+	flagPidFile       := set.Flag('p', "pid-file",       "Write the PID to the specified file", cli.ValString)
-	flagUser          := set.Flag('u', "user",          "The user:group to run as", cli.ValString)
+	flagUser          := set.Flag('u', "user",           "The user:group to run as", cli.ValString)
-	flagLogDirectory  := set.Flag('l', "log-directory", "Write logs to the specified directory", cli.ValString)
+	flagLogDirectory  := set.Flag('l', "log-directory",  "Write logs to the specified directory", cli.ValString)
-	flagConfigFile    := set.Flag('c', "config-file",   "Use this configuration file", cli.ValString)
+	flagConfigFile    := set.Flag('c', "config-file",    "Use this configuration file", cli.ValString)
-	flagVerbose       := set.Flag('v', "verbose",       "Enable verbose output/logging", nil)
+	flagVerbose       := set.Flag('v', "verbose",        "Enable verbose output/logging", nil)
-	flagCrash         := set.Flag(0,   "crash",         "Crash when an actor panics", nil)
+	flagCrash         := set.Flag(0,   "crash",          "Crash when an actor panics", nil)
+	flagCrashOnError  := set.Flag(0,   "crash-on-error", "Crash when an actor experiences any error", nil)
 
 	// ask actors to add flags
 	actors, done := this.actors.RBorrow()
@@ -64,6 +66,10 @@ func (this *environment) phase10FlagParsing() bool {
 	if _, ok := flagCrash.First(); ok {
 		this.flags.crash = true
 	}
+	if _, ok := flagCrashOnError.First(); ok {
+		this.flags.crash = true
+		this.flags.crashOnError = true
+	}
 	return true
 }
 
@@ -129,15 +135,15 @@ func (this *environment) phase30ConfigurationParsing() bool {
 		log.Println("!!! (30) could not determine location of file(s):", err)
 		return true
 	}
+	if this.flags.configFile != "" {
+		paths = append(paths, this.flags.configFile)
+	}
 	if this.Verb() {
 		log.Println("(i) (30) have configuration files:")
 		for _, paths := range paths {
 			log.Println("(i) (30) -", paths)
 		}
 	}
-	if this.flags.configFile != "" {
-		paths = append(paths, this.flags.configFile)
-	}
 	// parse every config and merge them all
 	configs := make([]iniConfig, 0, len(paths))
 	for _, path := range paths {
@@ -258,6 +264,9 @@ func (this *environment) phase70_5Trimming() bool {
 	}()
 	if err := this.trimActors(this.ctx, trimmable...); err != nil {
 		log.Println(".// (70.5) failed to trim:", err)
+		if this.flags.crashOnError {
+			panic(err)
+		}
 		return false
 	}
 	if this.Verb() { log.Println(".// (70.5) trimmed") }
@@ -273,6 +282,11 @@ func (this *environment) phase80Shutdown() bool {
 		<- ctx.Done()
 		if errors.Is(context.Cause(ctx), context.DeadlineExceeded) {
 			log.Println("XXX (80) shutdown timeout expired, performing emergency halt")
+			if Verb() || this.flags.crashOnError {
+				dumpBuffer := make([]byte, 8192)
+				runtime.Stack(dumpBuffer, true)
+				log.Printf("XXX (80) stack trace of all goroutines:\n%s", dumpBuffer)
+			}
 			log.Printf("====== [%s] END =======", this.name)
 			os.Exit(1)
 		}