From 1d3d5315a492966150851dd6879a9b4b8e31fc21 Mon Sep 17 00:00:00 2001 From: Adphi Date: Wed, 26 Oct 2022 12:02:53 +0200 Subject: [PATCH] ban: remove port from DefaultActorFunc Signed-off-by: Adphi --- interceptors/ban/options.go | 4 ++++ 1 file changed, 4 insertions(+) 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 }