Maybe handle syncing after repeated dis/connections better

This commit is contained in:
Tulir Asokan 2021-02-10 12:13:18 +02:00
parent 394da69a69
commit db2793c5e1

18
user.go
View file

@ -176,8 +176,8 @@ func (bridge *Bridge) NewUser(dbUser *database.User) *User {
IsRelaybot: false, IsRelaybot: false,
chatListReceived: make(chan struct{}, 1), chatListReceived: make(chan struct{}),
syncPortalsDone: make(chan struct{}, 1), syncPortalsDone: make(chan struct{}),
syncStart: make(chan struct{}, 1), syncStart: make(chan struct{}, 1),
messageInput: make(chan PortalMessage), messageInput: make(chan PortalMessage),
messageOutput: make(chan PortalMessage, bridge.Config.Bridge.UserMessageBuffer), messageOutput: make(chan PortalMessage, bridge.Config.Bridge.UserMessageBuffer),
@ -451,6 +451,10 @@ func (user *User) PostLogin() {
user.bridge.Metrics.TrackLoginState(user.JID, true) user.bridge.Metrics.TrackLoginState(user.JID, true)
user.bridge.Metrics.TrackBufferLength(user.MXID, 0) user.bridge.Metrics.TrackBufferLength(user.MXID, 0)
user.log.Debugln("Locking processing of incoming messages and starting post-login sync") user.log.Debugln("Locking processing of incoming messages and starting post-login sync")
// TODO can an old sync still be ongoing when PostLogin is called again?
// TODO 2: can the new sync happen before this happens?
user.chatListReceived = make(chan struct{})
user.syncPortalsDone = make(chan struct{})
user.syncWait.Add(1) user.syncWait.Add(1)
user.syncStart <- struct{}{} user.syncStart <- struct{}{}
go user.intPostLogin(user.Conn) go user.intPostLogin(user.Conn)
@ -593,6 +597,8 @@ func (user *User) HandleChatList(chats []whatsapp.Chat) {
} }
func (user *User) syncPortals(chatMap map[string]whatsapp.Chat, createAll bool) { func (user *User) syncPortals(chatMap map[string]whatsapp.Chat, createAll bool) {
// TODO use contexts instead of checking if user.Conn is the same?
connAtStart := user.Conn
if chatMap == nil { if chatMap == nil {
chatMap = user.Conn.Store.Chats chatMap = user.Conn.Store.Chats
} }
@ -633,6 +639,10 @@ func (user *User) syncPortals(chatMap map[string]whatsapp.Chat, createAll bool)
if limit < 0 { if limit < 0 {
limit = len(chats) limit = len(chats)
} }
if user.Conn != connAtStart {
user.log.Debugln("Connection seems to have changed before sync, cancelling")
return
}
now := uint64(time.Now().Unix()) now := uint64(time.Now().Unix())
user.log.Infoln("Syncing portals") user.log.Infoln("Syncing portals")
for i, chat := range chats { for i, chat := range chats {
@ -651,6 +661,10 @@ func (user *User) syncPortals(chatMap map[string]whatsapp.Chat, createAll bool)
} }
} }
} }
if user.Conn != connAtStart {
user.log.Debugln("Connection seems to have changed during sync, cancelling")
return
}
user.UpdateDirectChats(nil) user.UpdateDirectChats(nil)
user.log.Infoln("Finished syncing portals") user.log.Infoln("Finished syncing portals")
select { select {