diff --git a/fastdialer/context.go b/fastdialer/context.go index f7f39ba..45533aa 100644 --- a/fastdialer/context.go +++ b/fastdialer/context.go @@ -5,4 +5,5 @@ type ContextOption string const ( // SniName to use in tls connection SniName ContextOption = "sni-name" + IP ContextOption = "ip" ) diff --git a/fastdialer/dialer.go b/fastdialer/dialer.go index 0f8a5f8..2286333 100644 --- a/fastdialer/dialer.go +++ b/fastdialer/dialer.go @@ -180,7 +180,7 @@ func (d *Dialer) dial(ctx context.Context, network, address string, shouldUseTLS fixedIP = addressParts[2] } // check if the ip is within the context - if ctxIP := ctx.Value("ip"); ctxIP != nil { + if ctxIP := ctx.Value(IP); ctxIP != nil { fixedIP = fmt.Sprint(ctxIP) } } else {