Keep in cache the Id of the last seen post for backlogging purposes

This commit is contained in:
Alex 2020-02-21 19:36:16 +01:00
parent b0644c3a17
commit a14bacc9c1
3 changed files with 31 additions and 2 deletions

View file

@ -315,3 +315,17 @@ func (a *Account) eventInternal(event *Event) error {
return nil
}
}
// ----
func (a *Account) CacheGet(key string) string {
cache_key := fmt.Sprintf("%s/account/%s/%s/%s",
a.Protocol, a.MatrixUser, a.AccountName, key)
return dbCacheGet(cache_key)
}
func (a *Account) CachePut(key string, value string) {
cache_key := fmt.Sprintf("%s/account/%s/%s/%s",
a.Protocol, a.MatrixUser, a.AccountName, key)
dbCachePut(cache_key, value)
}

View file

@ -84,6 +84,13 @@ type Handler interface {
// Called when an event occurs in a room
// This must not be called for events authored by the user of the connection
Event(event *Event)
// These two functions enable the connector to access a simple key/value
// database to cache some information in order not to generate useless events.
// The connector should function when they are not implemented,
// in which case CacheGet always returns ""
CachePut(key string, value string)
CacheGet(key string) string
}
type EventType int

View file

@ -337,6 +337,7 @@ func (mm *Mattermost) handleConnected() {
mm.handler.RoomInfoUpdated(id, UserID(""), room_info)
// Update member list
// TODO (when this will be slow, i.e. hundreds of members): do only a diff
members, resp := mm.conn.Client.GetChannelMembers(ch.Id, 0, 1000, "")
if resp.Error == nil {
for _, mem := range *members {
@ -356,8 +357,14 @@ func (mm *Mattermost) handleConnected() {
}
// Read backlog
// TODO: remember what was the last seen post in this channel
backlog, resp := mm.conn.Client.GetPostsForChannel(ch.Id, 0, 1000, "")
var backlog *model.PostList
last_seen_post := mm.handler.CacheGet(fmt.Sprintf("last_seen_%s", ch.Id))
if last_seen_post != "" {
backlog, resp = mm.conn.Client.GetPostsAfter(ch.Id, last_seen_post, 0, 1000, "")
// TODO: if there are more than 1000, loop around
} else {
backlog, resp = mm.conn.Client.GetPostsForChannel(ch.Id, 0, 1000, "")
}
if resp.Error == nil {
for i := 0; i < len(backlog.Order); i++ {
post_id := backlog.Order[len(backlog.Order)-i-1]
@ -519,6 +526,7 @@ func (mm *Mattermost) handlePost(channel_name string, post *model.Post, only_mes
} else if post.Type == "" || post.Type == "me" {
msg_ev.Room = roomId
mm.handler.Event(msg_ev)
mm.handler.CachePut(fmt.Sprintf("last_seen_%s", post.ChannelId), post.Id)
} else {
return fmt.Errorf("Unhandled post type: %s", post.Type)
}