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

Defer counters

Browse Source
This commit is contained in:
9seconds
2018-07-08 17:56:44 +03:00
parent 474fb8a294
commit a99c5b2442
3 changed files with 24 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
wrappers/mtproto_intermediate.go
+8 -2
View File
@@ -24,6 +24,10 @@ type MTProtoIntermediate struct {
}
func (m *MTProtoIntermediate) Read() ([]byte, error) {
defer func() {
m.readCounter++
}()
m.logger.Debugw("Read packet",
"simple_ack", m.opts.ReadHacks.SimpleAck,
"quick_ack", m.opts.ReadHacks.QuickAck,
@@ -59,18 +63,20 @@ func (m *MTProtoIntermediate) Read() ([]byte, error) {
if length%4 != 0 {
length -= length % 4
}
m.readCounter++
return buf.Bytes()[:length], nil
}
func (m *MTProtoIntermediate) Write(p []byte) (int, error) {
defer func() {
m.writeCounter++
}()
m.logger.Debugw("Write packet",
"simple_ack", m.opts.WriteHacks.SimpleAck,
"quick_ack", m.opts.WriteHacks.QuickAck,
"counter", m.writeCounter,
)
m.writeCounter++
if m.opts.ReadHacks.SimpleAck {
return m.conn.Write(p)