add crash Handler

Author: leopoldxx

trace/handle_crash.go

@@ -0,0 +1,17 @@
+package trace
+
+import "context"
+
+// LogCrashStack for stack
+func LogCrashStack(ctx context.Context, r interface{}) {
+	GetTraceFromContext(ctx).Warnf("panic: %v, detail: %s", r, string(Stacks(false)))
+}
+
+// HandleCrash xx
+func HandleCrash(fn ...func(r interface{})) {
+	if r := recover(); r != nil {
+		for _, f := range fn {
+			f(r)
+		}
+	}
+}

trace/handle_crash_test.go

@@ -0,0 +1,47 @@
+package trace
+
+import (
+	"context"
+	"fmt"
+	"log"
+	"testing"
+)
+
+func TestHandleCrash(t *testing.T) {
+	crash()
+	t.Log("still ok")
+	f()
+}
+
+func crash() {
+	ctx := WithTraceForContext(context.TODO(), "crash-test")
+	defer HandleCrash(func(r interface{}) {
+		LogCrashStack(ctx, r)
+	})
+
+	func() {
+		func() {
+			panic("do it")
+		}()
+	}()
+}
+
+func f() {
+	defer HandleCrash(func(r interface{}) {
+		log.Printf("recover from: %v", r)
+	})
+
+	fmt.Println("Calling g.")
+	g(0)
+	fmt.Println("Returned normally from g.")
+}
+
+func g(i int) {
+	if i > 3 {
+		fmt.Println("Panicking!")
+		panic(fmt.Sprintf("%v", i))
+	}
+	defer fmt.Println("Defer in g", i)
+	fmt.Println("Printing in g", i)
+	g(i + 1)
+}

trace/trace.go

@@ -57,7 +57,7 @@ type Trace interface {
 	// Errorf will print the args with a format as the error level log
 	Errorf(format string, args ...interface{})
 	// Stack will return current stack
-	Stack() string
+	Stack(all ...bool) string
 	// String will return a string-serialized trace
 	String() string
 }
@@ -170,7 +170,7 @@ func (t *trace) Duration() time.Duration {
 }
 
 // copy this from glog
-func stacks(all bool) []byte {
+func Stacks(all bool) []byte {
 	n := 10000
 	if all {
 		n = 100000
@@ -191,8 +191,12 @@ func (t *trace) String() string {
 	return t.header()
 }
 
-func (t *trace) Stack() string {
-	return string(stacks(true))
+func (t *trace) Stack(all ...bool) string {
+	dumpAll := false
+	if len(all) > 0 {
+		dumpAll = all[0]
+	}
+	return string(Stacks(dumpAll))
 }
 
 func (t *trace) log(out func(depth int, args ...interface{}), args ...interface{}) {