(ctx context.Context, config *websocket.Config)
| 146 | } |
| 147 | |
| 148 | func wsDialContext(ctx context.Context, config *websocket.Config) (*websocket.Conn, error) { |
| 149 | var conn net.Conn |
| 150 | var err error |
| 151 | switch config.Location.Scheme { |
| 152 | case "ws": |
| 153 | conn, err = dialContext(ctx, "tcp", wsDialAddress(config.Location)) |
| 154 | case "wss": |
| 155 | dialer := contextDialer(ctx) |
| 156 | conn, err = tls.DialWithDialer(dialer, "tcp", wsDialAddress(config.Location), config.TlsConfig) |
| 157 | default: |
| 158 | err = websocket.ErrBadScheme |
| 159 | } |
| 160 | if err != nil { |
| 161 | return nil, err |
| 162 | } |
| 163 | ws, err := websocket.NewClient(config, conn) |
| 164 | if err != nil { |
| 165 | conn.Close() |
| 166 | return nil, err |
| 167 | } |
| 168 | return ws, err |
| 169 | } |
| 170 | |
| 171 | var wsPortMap = map[string]string{"ws": "80", "wss": "443"} |
| 172 |
no test coverage detected