Handle panics by deferring recover as an error (#345)

* Handle panics by deferring recover as an error * Check for type in recover handler
d5 · Nov 13, 2021 · a7666f0 · a7666f0
1 parent 3b65ddf
commit a7666f0
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/script.go b/script.go
@@ -219,6 +219,18 @@ func (c *Compiled) RunContext(ctx context.Context) (err error) {
 	v := NewVM(c.bytecode, c.globals, c.maxAllocs)
 	ch := make(chan error, 1)
 	go func() {
+		defer func() {
+			if r := recover(); r != nil {
+				switch e := r.(type) {
+				case string:
+					ch <- fmt.Errorf(e)
+				case error:
+					ch <- e
+				default:
+					ch <- fmt.Errorf("unknown panic: %v", e)
+				}
+			}
+		}()
 		ch <- v.Run()
 	}()