(dialer proxyclient.Dial, timeout time.Duration)
| 124 | } |
| 125 | |
| 126 | func customDialFunc(dialer proxyclient.Dial, timeout time.Duration) fasthttp.DialFunc { |
| 127 | if dialer == nil { |
| 128 | return func(addr string) (net.Conn, error) { |
| 129 | return fasthttp.DialTimeout(addr, timeout) |
| 130 | } |
| 131 | } |
| 132 | return func(addr string) (net.Conn, error) { |
| 133 | ctx, cancel := context.WithTimeout(context.Background(), timeout) |
| 134 | defer cancel() |
| 135 | return dialer.DialContext(ctx, "tcp", addr) |
| 136 | } |
| 137 | } |