Merge pull request PaesslerAG#85 from Yangruipis/fix/function_panic_r…

…ecover fix(PaesslerAG#79): function panic recover
tw1nk · Sep 21, 2022 · 02d92bf · 02d92bf
2 parents 8fe7bad + 6ca2339
commit 02d92bf
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/functions.go b/functions.go
@@ -14,6 +14,11 @@ func toFunc(f interface{}) function {
  var v interface{}
  errCh := make(chan error)
  go func() {
+ defer func() {
+ if recovered := recover(); recovered != nil {
+ errCh <- fmt.Errorf("%v", recovered)
+ }
+ }()
  result, err := f(arguments...)
  v = result
  errCh <- err
@@ -37,6 +42,11 @@ func toFunc(f interface{}) function {
  var v interface{}
  errCh := make(chan error)
  go func() {
+ defer func() {
+ if recovered := recover(); recovered != nil {
+ errCh <- fmt.Errorf("%v", recovered)
+ }
+ }()
  in, err := createCallArguments(ctx, t, args)
  if err != nil {
  errCh <- err