astra/gotdlib
1
0
Fork 0
mirror of https://github.com/c0re100/gotdlib.git synced 2026-02-21 20:20:17 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

timeout configuration

Browse source
This commit is contained in:
Aleksandr Zelenin 2018-10-09 06:08:28 +03:00
parent d74e676cd7
commit 43d64cbc45
2 changed files with 29 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

28
client/client.go
View file

@ -12,6 +12,8 @@ type Client struct {
catcher chan *Response
listenerStore *listenerStore
catchersStore *sync.Map
updatesTimeout time.Duration
catchTimeout time.Duration
}
type Option func(*Client)
@ -22,6 +24,18 @@ func WithExtraGenerator(extraGenerator ExtraGenerator) Option {
}
}
func WithCatchTimeout(timeout time.Duration) Option {
return func(client *Client) {
client.catchTimeout = timeout
}
}
func WithUpdatesTimeout(timeout time.Duration) Option {
return func(client *Client) {
client.updatesTimeout = timeout
}
}
func NewClient(authorizationStateHandler AuthorizationStateHandler, options ...Option) (*Client, error) {
catchersListener := make(chan *Response, 1000)
@ -40,6 +54,14 @@ func NewClient(authorizationStateHandler AuthorizationStateHandler, options ...O
client.extraGenerator = UuidV4Generator()
}
if client.catchTimeout == 0 {
client.catchTimeout = 60 * time.Second
}
if client.updatesTimeout == 0 {
client.updatesTimeout = 60 * time.Second
}
go client.receive()
go client.catch(catchersListener)
@ -53,7 +75,7 @@ func NewClient(authorizationStateHandler AuthorizationStateHandler, options ...O
func (client *Client) receive() {
for {
resp, err := client.jsonClient.Receive(10)
resp, err := client.jsonClient.Receive(client.updatesTimeout)
if err != nil {
continue
}
@ -107,8 +129,8 @@ func (client *Client) Send(req Request) (*Response, error) {
case response := <-catcher:
return response, nil
case <-time.After(10 * time.Second):
return nil, errors.New("timeout")
case <-time.After(client.catchTimeout):
return nil, errors.New("response catching timeout")
}
}