feat: unwrap grpc status error before checking (#3)

Syncano · Jun 5, 2020 · f7338f8 · f7338f8
1 parent 45379dc
commit f7338f8
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/util/util.go b/util/util.go
@@ -128,9 +128,12 @@ func checkError(err, target error, code codes.Code) bool {
 		return true
 	}
 
-	s, ok := status.FromError(err)
+	var grpcstatus interface{ GRPCStatus() *status.Status }
+	if errors.As(err, &grpcstatus) {
+		return grpcstatus.GRPCStatus().Code() == code
+	}
 
-	return ok && s.Code() == code
+	return false
 }
 
 func IsContextError(err error) bool {