Print a stack trace on panic

nihei9 · Aug 5, 2021 · 3d417d5 · 3d417d5
1 parent 7b4ed66
commit 3d417d5
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/cmd/vartan/compile.go b/cmd/vartan/compile.go
@@ -6,6 +6,7 @@ import (
 	"io/ioutil"
 	"os"
 	"path/filepath"
+	"runtime/debug"
 	"strings"
 
 	verr "github.com/nihei9/vartan/error"
@@ -46,8 +47,8 @@ func runCompile(cmd *cobra.Command, args []string) (retErr error) {
 		if v != nil {
 			err, ok := v.(error)
 			if !ok {
-				retErr = fmt.Errorf("an unexpected error occurred: %v\n", v)
-				fmt.Fprintln(os.Stderr, retErr)
+				retErr = fmt.Errorf("an unexpected error occurred: %v", v)
+				fmt.Fprintf(os.Stderr, "%v:\n%v", retErr, string(debug.Stack()))
 				return
 			}
 
@@ -67,7 +68,8 @@ func runCompile(cmd *cobra.Command, args []string) (retErr error) {
 					}
 				}
 			}
-			fmt.Fprintln(os.Stderr, retErr)
+
+			fmt.Fprintf(os.Stderr, "%v:\n%v", retErr, string(debug.Stack()))
 		}
 	}()
 

diff --git a/cmd/vartan/parse.go b/cmd/vartan/parse.go
@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"io/ioutil"
 	"os"
+	"runtime/debug"
 
 	"github.com/nihei9/vartan/driver"
 	"github.com/nihei9/vartan/spec"
@@ -37,16 +38,16 @@ func runParse(cmd *cobra.Command, args []string) (retErr error) {
 		if v != nil {
 			err, ok := v.(error)
 			if !ok {
-				retErr = fmt.Errorf("an unexpected error occurred: %v\n", v)
-				fmt.Fprintln(os.Stderr, retErr)
+				retErr = fmt.Errorf("an unexpected error occurred: %v", v)
+				fmt.Fprintf(os.Stderr, "%v:\n%v", retErr, string(debug.Stack()))
 				return
 			}
 
 			retErr = err
 		}
 
 		if retErr != nil {
-			fmt.Fprintln(os.Stderr, retErr)
+			fmt.Fprintf(os.Stderr, "%v:\n%v", retErr, string(debug.Stack()))
 		}
 	}()