narayana/go-tdlib
3
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

fix: time.After will cause memory leak

Browse source
This commit is contained in:
divinerapier 2021-09-05 09:36:38 +08:00
parent 98e8cb003f
commit 278b1ae10d
No known key found for this signature in database
GPG key ID: 102F344DF3581AED
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
client/client.go
View file

@ -1,6 +1,7 @@
package client package client
import ( import (
"context"
"errors" "errors"
"sync" "sync"
"time" "time"
@ -123,17 +124,20 @@ func (client *Client) Send(req Request) (*Response, error) {
client.catchersStore.Store(req.Extra, catcher) client.catchersStore.Store(req.Extra, catcher)
defer func() { defer func() {
close(catcher)
client.catchersStore.Delete(req.Extra) client.catchersStore.Delete(req.Extra)
close(catcher)
}() }()
client.jsonClient.Send(req) client.jsonClient.Send(req)
ctx, cancel := context.WithTimeout(context.Background(), client.catchTimeout)
defer cancel()
select { select {
case response := <-catcher: case response := <-catcher:
return response, nil return response, nil
case <-time.After(client.catchTimeout): case <-ctx.Done():
return nil, errors.New("response catching timeout") return nil, errors.New("response catching timeout")
} }
} }