native client listen does not call the wg.Done if the context is cancelled (prior to a DialConfig success). Here is code to resolve the problem.

[daninmadison]

In the listen function, the for loop checks for a ctx closed. If it is, it exits the function. It needs to call the wg.Done.

If the Connect function is given a wrong details, the initial listen will never succeed the websocket.DialConfig. As a result, the only way to stop this function is to cancel the context. The problem is, the context closed does not call wg.Done prior to the return.

As a result, the Connect will be stuck at the wg.Wait and never return.

Here is code to correct the problem.

func (c Client) listen(ctx context.Context, wg sync.WaitGroup) { var signalUp sync.Once

for {
    // Exit if our context has been closed
    if ctx.Err() != nil {
        if wg != nil {
            wg.Done()
        }
        return
    }

[daninmadison]

I realized the code I pushed earlier this week had a bug. It needs to use the sync once signalUp for the wg.Done.

Pushed a new fix. Just in case that didn't come through correctly, here is the corrected code...

for {
    // Exit if our context has been closed
    if ctx.Err() != nil {
        if wg != nil {
            signalUp.Do(wg.Done)
        }
        return
    }