From fd090ae0a4cac3f5feffbb9fe6dfe859bf342cd1 Mon Sep 17 00:00:00 2001
From: Josh Leverette <jleverette@onesignal.com>
Date: Tue, 8 Jan 2019 11:52:29 -0800
Subject: [PATCH] add support for github.com/pingcap/errors stacktraces

---
 stacktrace.go | 81 +++++++++++++++++++++++++++++++++++----------------
 1 file changed, 56 insertions(+), 25 deletions(-)

diff --git a/stacktrace.go b/stacktrace.go
index 1ec603f..9473f3c 100644
--- a/stacktrace.go
+++ b/stacktrace.go
@@ -14,6 +14,7 @@ import (
 	"strings"
 	"sync"
 
+	pingcap "github.com/pingcap/errors"
 	"github.com/pkg/errors"
 )
 
@@ -51,32 +52,62 @@ type StacktraceFrame struct {
 	InApp        bool     `json:"in_app"`
 }
 
-// Try to get stacktrace from err as an interface of github.com/pkg/errors, or else NewStacktrace()
-func GetOrNewStacktrace(err error, skip int, context int, appPackagePrefixes []string) *Stacktrace {
-	stacktracer, errHasStacktrace := err.(interface {
-		StackTrace() errors.StackTrace
-	})
-	if errHasStacktrace {
-		var frames []*StacktraceFrame
-		for f := range stacktracer.StackTrace() {
-			pc := uintptr(f) - 1
-			fn := runtime.FuncForPC(pc)
-			var fName string
-			var file string
-			var line int
-			if fn != nil {
-				file, line = fn.FileLine(pc)
-				fName = fn.Name()
-			} else {
-				file = "unknown"
-				fName = "unknown"
-			}
-			frame := NewStacktraceFrame(pc, fName, file, line, context, appPackagePrefixes)
-			if frame != nil {
-				frames = append([]*StacktraceFrame{frame}, frames...)
-			}
+type pkgStacktracer interface {
+	StackTrace() errors.StackTrace
+}
+
+func pkgStacktrace(stacktracer pkgStacktracer, context int, appPackagePrefixes []string) *Stacktrace {
+	var frames []*StacktraceFrame
+	for f := range stacktracer.StackTrace() {
+		pc := uintptr(f) - 1
+		fn := runtime.FuncForPC(pc)
+		var fName string
+		var file string
+		var line int
+		if fn != nil {
+			file, line = fn.FileLine(pc)
+			fName = fn.Name()
+		} else {
+			file = "unknown"
+			fName = "unknown"
+		}
+		frame := NewStacktraceFrame(pc, fName, file, line, context, appPackagePrefixes)
+		if frame != nil {
+			frames = append([]*StacktraceFrame{frame}, frames...)
+		}
+	}
+	return &Stacktrace{Frames: frames}
+}
+
+func pingcapStacktrace(stacktracer pingcap.StackTracer, context int, appPackagePrefixes []string) *Stacktrace {
+	var frames []*StacktraceFrame
+	for f := range stacktracer.StackTrace() {
+		pc := uintptr(f) - 1
+		fn := runtime.FuncForPC(pc)
+		var fName string
+		var file string
+		var line int
+		if fn != nil {
+			file, line = fn.FileLine(pc)
+			fName = fn.Name()
+		} else {
+			file = "unknown"
+			fName = "unknown"
+		}
+		frame := NewStacktraceFrame(pc, fName, file, line, context, appPackagePrefixes)
+		if frame != nil {
+			frames = append([]*StacktraceFrame{frame}, frames...)
 		}
-		return &Stacktrace{Frames: frames}
+	}
+	return &Stacktrace{Frames: frames}
+}
+
+// Try to get stacktrace from err as an interface of github.com/pkg/errors or github.com/pingcap/errors, or else NewStacktrace()
+func GetOrNewStacktrace(err error, skip int, context int, appPackagePrefixes []string) *Stacktrace {
+	if stacktracer, errHasStacktrace := err.(pkgStacktracer); errHasStacktrace {
+		return pkgStacktrace(stacktracer, context, appPackagePrefixes)
+	} else if stacktracer, errHasStacktrace := err.(pingcap.StackTracer); errHasStacktrace {
+		return pingcapStacktrace(stacktracer, context, appPackagePrefixes)
 	} else {
 		return NewStacktrace(skip+1, context, appPackagePrefixes)
 	}