diff --git a/interceptors/ban/options.go b/interceptors/ban/options.go index 31b3400..fa6c403 100644 --- a/interceptors/ban/options.go +++ b/interceptors/ban/options.go @@ -2,6 +2,7 @@ package ban import ( "context" + "net" "time" "google.golang.org/grpc/codes" @@ -46,6 +47,9 @@ func DefaultActorFunc(ctx context.Context) (string, bool, error) { if !ok { return "", false, nil } + if host, _, err := net.SplitHostPort(p.Addr.String()); err == nil { + return host, true, nil + } return p.Addr.String(), true, nil }