x 初步确定

Author: FishGoddess

x/runtime.go

@@ -19,8 +19,8 @@ func Caller() string {
 }
 
 func Callers() []string {
-	pcs := make([]uintptr, 16)
-	n := runtime.Callers(2, pcs)
+	var pcs [16]uintptr
+	n := runtime.Callers(2, pcs[:])
 	frames := runtime.CallersFrames(pcs[:n])
 
 	var callers []string

x/runtime_test.go

@@ -8,6 +8,7 @@ import (
 	"testing"
 )
 
+// go test -v -run=^$ -bench=^BenchmarkCaller$ -benchtime=1s
 // BenchmarkCaller-2   	 1390318	       876.7 ns/op	     296 B/op	       3 allocs/op
 func BenchmarkCaller(b *testing.B) {
 	b.ReportAllocs()
@@ -18,6 +19,7 @@ func BenchmarkCaller(b *testing.B) {
 	}
 }
 
+// go test -v -run=^$ -bench=^BenchmarkCallers$ -benchtime=1s
 // BenchmarkCallers-2   	  443419	      2573 ns/op	     732 B/op	      12 allocs/op
 func BenchmarkCallers(b *testing.B) {
 	b.ReportAllocs()

x/unwrap.go

@@ -60,3 +60,24 @@ func CodeMessage(err error, defaultCode int32, defaultMessage string) (int32, st
 
 	return code, message
 }
+
+// Match unwraps error and check if its code equals to code.
+func Match(err error, code int32) bool {
+	if err == nil {
+		return code == 0
+	}
+
+	cerr, ok := err.(interface {
+		Code() int32
+	})
+	if ok {
+		return cerr.Code() == code
+	}
+
+	var xerr *Error
+	if As(err, &xerr) {
+		return xerr.Code() == code
+	}
+
+	return false
+}

x/unwrap_test.go

@@ -11,6 +11,8 @@ import (
 
 // go test -v -cover -count=1 -test.cpu=1 -run=^TestCode$
 func TestCode(t *testing.T) {
+	testErr := &testError{}
+
 	testCases := []struct {
 		err         error
 		defaultCode int32
@@ -26,6 +28,11 @@ func TestCode(t *testing.T) {
 			defaultCode: 999,
 			code:        999,
 		},
+		{
+			err:         testErr,
+			defaultCode: 999,
+			code:        testErr.Code(),
+		},
 		{
 			err:         Wrap(1000, "wow"),
 			defaultCode: 999,
@@ -82,3 +89,52 @@ func TestMessage(t *testing.T) {
 		}
 	}
 }
+
+// go test -v -cover -count=1 -test.cpu=1 -run=^TestMatch$
+func TestMatch(t *testing.T) {
+	testErr := &testError{}
+
+	testCases := []struct {
+		err   error
+		code  int32
+		match bool
+	}{
+		{
+			err:   nil,
+			code:  0,
+			match: true,
+		},
+		{
+			err:   nil,
+			code:  999,
+			match: false,
+		},
+		{
+			err:   io.EOF,
+			code:  999,
+			match: false,
+		},
+		{
+			err:   testErr,
+			code:  testErr.Code(),
+			match: true,
+		},
+		{
+			err:   Wrap(1000, "wow"),
+			code:  1000,
+			match: true,
+		},
+		{
+			err:   Wrap(1000, "eof").With(io.EOF),
+			code:  1000,
+			match: true,
+		},
+	}
+
+	for _, testCase := range testCases {
+		match := Match(testCase.err, testCase.code)
+		if match != testCase.match {
+			t.Errorf("match %+v != testCase.match %+v", match, testCase.match)
+		}
+	}
+}