Implement on-demand updating of room & user pictures

1 files changed, 34 insertions, 0 deletions

diff --git a/appservice/db.go b/appservice/db.go
index 5cefd30..646dae3 100644
--- a/appservice/db.go
+++ b/appservice/db.go
@@ -23,6 +23,8 @@ func InitDb() error {
 		return err
 	}
 
+	db.AutoMigrate(&DbCache{})
+
 	db.AutoMigrate(&DbUserMap{})
 	db.Model(&DbUserMap{}).AddIndex("idx_protocol_user", "protocol", "user_id")
 
@@ -35,6 +37,14 @@ func InitDb() error {
 	return nil
 }
 
+// Long-term cache entries
+type DbCache struct {
+	gorm.Model
+
+	Key string `gorm:"unique_index"`
+	Value string
+}
+
 // User mapping between protocol user IDs and puppeted matrix ids
 type DbUserMap struct {
 	gorm.Model
@@ -76,6 +86,30 @@ type DbPmRoomMap struct {
 
 // ----
 
+func dbCacheGet(key string) string {
+	var entry DbCache
+	if db.Where(&DbCache{Key: key}).First(&entry).RecordNotFound() {
+		return ""
+	} else {
+		return entry.Value
+	}
+}
+
+func dbCachePut(key string, value string) {
+	var entry DbCache
+	db.Where(&DbCache{Key: key}).Assign(&DbCache{Value: value}).FirstOrCreate(&entry)
+}
+
+func dbCacheTestAndSet(key string, value string) bool {
+	// TODO make this really an atomic operation
+	// True if value was changed, false if was already set
+	if dbCacheGet(key) != value {
+		dbCachePut(key, value)
+		return true
+	}
+	return false
+}
+
 func dbGetMxRoom(protocol string, roomId connector.RoomID) (string, error) {
 	var room DbRoomMap