stats: try to interpret error with statuser interface

upfluence · Jun 22, 2021 · b68db6c · b68db6c
1 parent cf71630
commit b68db6c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 7 deletions.
diff --git a/stats/statuser.go b/stats/statuser.go
@@ -34,6 +34,10 @@ func (defaultStatuser) Status(err error) string {
 }
 
 func GetStatus(err error, opts ...ExtractStatusOption) string {
+	type statuser interface {
+		Status() string
+	}
+
 	var o = defaultStatusOptions
 
 	for _, opt := range opts {
@@ -45,10 +49,8 @@ func GetStatus(err error, opts ...ExtractStatusOption) string {
 	}
 
 	for {
-		ws, ok := err.(*withStatus)
-
-		if ok {
-			return ws.status
+		if st, ok := err.(statuser); ok {
+			return st.Status()
 		}
 
 		cause := base.UnwrapOnce(err)

diff --git a/stats/with_status.go b/stats/with_status.go
@@ -5,9 +5,10 @@ type withStatus struct {
 	status string
 }
 
-func (ws *withStatus) Error() string { return ws.cause.Error() }
-func (ws *withStatus) Unwrap() error { return ws.cause }
-func (ws *withStatus) Cause() error  { return ws.cause }
+func (ws *withStatus) Error() string  { return ws.cause.Error() }
+func (ws *withStatus) Unwrap() error  { return ws.cause }
+func (ws *withStatus) Cause() error   { return ws.cause }
+func (ws *withStatus) Status() string { return ws.status }
 
 func (ws *withStatus) Tags() map[string]interface{} {
 	return map[string]interface{}{"status": ws.status}