seang96/gmessages
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity Actions

Send extra GET_UPDATES call after disconnections

Browse source
This commit is contained in:
Tulir Asokan 2024-05-15 15:31:08 +03:00
parent c0ac0571f6
commit c3a477b704
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
libgm/longpoll.go
View file

@ -187,19 +187,26 @@ const DefaultBugleDefaultCheckInterval = 2*time.Hour + 55*time.Minute
func (dp *dittoPinger) Loop() { func (dp *dittoPinger) Loop() {
for { for {
var pingStart time.Time
select { select {
case <-dp.client.pingShortCircuit: case <-dp.client.pingShortCircuit:
pingID := pingIDCounter.Add(1) pingID := pingIDCounter.Add(1)
dp.log.Debug().Uint64("ping_id", pingID).Msg("Ditto ping wait short-circuited") dp.log.Debug().Uint64("ping_id", pingID).Msg("Ditto ping wait short-circuited")
pingStart = time.Now()
dp.Ping(pingID, shortPingTimeout, 0) dp.Ping(pingID, shortPingTimeout, 0)
case <-dp.ping: case <-dp.ping:
pingID := pingIDCounter.Add(1) pingID := pingIDCounter.Add(1)
dp.log.Trace().Uint64("ping_id", pingID).Msg("Doing normal ditto ping") dp.log.Trace().Uint64("ping_id", pingID).Msg("Doing normal ditto ping")
pingStart = time.Now()
dp.Ping(pingID, defaultPingTimeout, 0) dp.Ping(pingID, defaultPingTimeout, 0)
case <-dp.stop: case <-dp.stop:
return return
} }
if dp.client.shouldDoDataReceiveCheck() { if dp.client.shouldDoDataReceiveCheck() {
dp.log.Warn().Msg("No data received recently, sending extra GET_UPDATES call")
go dp.HandleNoRecentUpdates()
} else if time.Since(pingStart) > 5*time.Minute {
dp.log.Warn().Msg("Was disconnected for over 5 minutes, sending extra GET_UPDATES call")
go dp.HandleNoRecentUpdates() go dp.HandleNoRecentUpdates()
} }
} }
@ -207,7 +214,6 @@ func (dp *dittoPinger) Loop() {
func (dp *dittoPinger) HandleNoRecentUpdates() { func (dp *dittoPinger) HandleNoRecentUpdates() {
dp.client.triggerEvent(&events.NoDataReceived{}) dp.client.triggerEvent(&events.NoDataReceived{})
dp.log.Warn().Msg("No data received recently, sending extra GET_UPDATES call")
err := dp.client.sessionHandler.sendMessageNoResponse(SendMessageParams{ err := dp.client.sessionHandler.sendMessageNoResponse(SendMessageParams{
Action: gmproto.ActionType_GET_UPDATES, Action: gmproto.ActionType_GET_UPDATES,
OmitTTL: true, OmitTTL: true,