This commit is contained in:
Kevin Wan
2021-08-04 18:45:05 +08:00
committed by GitHub
parent 16cc990fdd
commit 0ee7654407
5 changed files with 80 additions and 1 deletions

View File

@@ -1,6 +1,8 @@
package codes
import (
"context"
"google.golang.org/grpc/codes"
"google.golang.org/grpc/status"
)
@@ -10,6 +12,17 @@ func Acceptable(err error) bool {
switch status.Code(err) {
case codes.DeadlineExceeded, codes.Internal, codes.Unavailable, codes.DataLoss:
return false
case codes.Unknown:
return acceptableUnknown(err)
default:
return true
}
}
func acceptableUnknown(err error) bool {
switch err {
case context.DeadlineExceeded:
return false
default:
return true
}