refactor code

This commit is contained in:
naiba
2024-11-23 00:02:57 +08:00
parent 58cc8aa5df
commit 68f6da436d
2 changed files with 4 additions and 4 deletions

View File

@@ -51,11 +51,11 @@ func getRealIp(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo,
}
a := strings.Split(vals[0], ",")
h := strings.TrimSpace(a[len(a)-1])
ip, err := netip.ParseAddr(h)
ip, err := netip.ParseAddrPort(h)
if err != nil {
return nil, err
}
ctx = context.WithValue(ctx, model.CtxKeyRealIP{}, ip.String())
ctx = context.WithValue(ctx, model.CtxKeyRealIP{}, ip.Addr().String())
return handler(ctx, req)
}