You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
47 lines
823 B
47 lines
823 B
3 years ago
|
package mtproto
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"github.com/go-faster/errors"
|
||
|
|
"go.uber.org/zap"
|
||
|
|
|
||
|
|
"github.com/gotd/td/internal/mt"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (c *Conn) ackLoop(ctx context.Context) error {
|
||
|
|
log := c.log.Named("ack")
|
||
|
|
|
||
|
|
var buf []int64
|
||
|
|
send := func() {
|
||
|
|
defer func() { buf = buf[:0] }()
|
||
|
|
|
||
|
|
if err := c.writeServiceMessage(ctx, &mt.MsgsAck{MsgIDs: buf}); err != nil {
|
||
|
|
c.log.Error("Failed to ACK", zap.Error(err))
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
log.Debug("Ack", zap.Int64s("msg_ids", buf))
|
||
|
|
}
|
||
|
|
|
||
|
|
ticker := c.clock.Ticker(c.ackInterval)
|
||
|
|
defer ticker.Stop()
|
||
|
|
|
||
|
|
for {
|
||
|
|
select {
|
||
|
|
case <-ctx.Done():
|
||
|
|
return errors.Wrap(ctx.Err(), "acl")
|
||
|
|
case <-ticker.C():
|
||
|
|
if len(buf) > 0 {
|
||
|
|
send()
|
||
|
|
}
|
||
|
|
case msgID := <-c.ackSendChan:
|
||
|
|
buf = append(buf, msgID)
|
||
|
|
if len(buf) >= c.ackBatchSize {
|
||
|
|
send()
|
||
|
|
ticker.Reset(c.ackInterval)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|