Go httpClient утечки памяти - PullRequest
0 голосов
/ 25 мая 2018

ОБНОВЛЕННЫЙ КОД

Привет! У меня утечки памяти в httpClient, я добавил sync.WaitGroup, и теперь я вижу, что подпрограмма с httpClient не закрылась.Как это исправить?

func checkProxySOCKS(prox string, c chan QR, wg *sync.WaitGroup) (err error) {

defer wg.Done()
dialer, _ := proxy.SOCKS5("tcp", prox, nil, proxy.Direct)

timeout := time.Duration(2 * time.Second)

httpClient := &http.Client{
    Timeout: timeout,
    Transport: &http.Transport{
        DisableKeepAlives: true,
        Dial:              dialer.Dial,
    },
}

res, err := httpClient.Get("https://telegram.org/")
if err != nil {

    c <- QR{Addr: prox, Res: false}
    return
}

defer res.Body.Close()
io.Copy(ioutil.Discard, res.Body)

c <- QR{Addr: prox, Res: true}

return nil

} Здесь я создаю goroutines

for _, proxy := range splitedProxies {
    wg.Add(1)
    go checkProxySOCKS(proxy, respChan, &wg)
}

for range splitedProxies {
    wg.Add(1)
    r := <-respChan
    if r.Res {
        checkedProxiesArray = append(checkedProxiesArray, r.Addr)
    }
    wg.Done()
}

wg.Wait()

1 Ответ

0 голосов
/ 27 мая 2018

Я прочитал комментарии, так что это должно решить вашу проблему

const (
    timeout = time.Duration(1000 * time.Millisecond)
    tt      = time.Duration(100 * time.Millisecond)
)

func checkProxySOCKS(prox string, c chan QR, wg *sync.WaitGroup) (err error) {

    defer wg.Done()

    d := net.Dialer{
        Timeout:   tt,
        KeepAlive: tt,
    }

    dialer, _ := proxy.SOCKS5("tcp", prox, nil, &d)

    httpClient := &http.Client{
        Timeout: timeout,
        Transport: &http.Transport{
            DisableKeepAlives: true,
            Dial:              dialer.Dial,
        },
    }

    res, err := httpClient.Get("https://telegram.org/")
    if err != nil {

        c <- QR{Addr: prox, Res: false}
        return
    }

    defer res.Body.Close()
    io.Copy(ioutil.Discard, res.Body)

    c <- QR{Addr: prox, Res: true}

    return nil
}
...