ScuroNeko/mtg
1
0
Fork 0
mirror of https://github.com/ScuroNeko/mtg.git synced 2026-09-01 00:44:02 +03:00
Code Issues Packages Projects Releases Wiki Activity

Set buffer sizes to sockets

Browse Source
This commit is contained in:
9seconds
2018-07-09 13:01:21 +03:00
parent 2a6cc7e195
commit fb00cd1121
3 changed files with 29 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
telegram/dialer.go
+15 -3
View File
@@ -10,7 +10,11 @@ import (
"github.com/9seconds/mtg/wrappers"
)
const telegramDialTimeout = 10 * time.Second
const (
telegramDialTimeout = 10 * time.Second
readBufferSize = 64 * 1024
writeBufferSize = 64 * 1024
)
type tgDialer struct {
net.Dialer
@@ -23,8 +27,16 @@ func (t *tgDialer) dial(addr string) (net.Conn, error) {
if err != nil {
return nil, errors.Annotate(err, "Cannot connect to Telegram")
}
if err = config.SetSocketOptions(conn); err != nil {
return nil, errors.Annotate(err, "Cannot set socket options")
tcpSocket := conn.(*net.TCPConn)
if err = tcpSocket.SetNoDelay(true); err != nil {
return nil, errors.Annotate(err, "Cannot set NO_DELAY to Telegram")
}
if err = tcpSocket.SetReadBuffer(readBufferSize); err != nil {
return nil, errors.Annotate(err, "Cannot set read buffer size on telegram socket")
}
if err = tcpSocket.SetWriteBuffer(writeBufferSize); err != nil {
return nil, errors.Annotate(err, "Cannot set write buffer size on telegram socket")
}
return conn, nil