Add DB replica support

server/plugin.go

@@ -518,8 +518,14 @@ func (p *Plugin) onActivate() error {
 			return dbErr
 		}
 
+		replica, repErr := p.apiClient.Store.GetReplicaDB()
+		if repErr != nil {
+			return repErr
+		}
+
 		store := sqlstore.New(
 			db,
+			replica,
 			p.API,
 			func() []string { return strings.Split(p.configuration.EnabledTeams, ",") },
 			func() []byte { return []byte(p.configuration.EncryptionKey) },

server/store/sqlstore/helper_test.go

@@ -82,6 +82,7 @@ func setupTestStore(t *testing.T) (*SQLStore, *plugintest.API) {
 	store := &SQLStore{}
 	store.api = api
 	store.db = db
+	store.replica = db
 
 	err := store.createTable("Teams", "Id VARCHAR(255), DisplayName VARCHAR(255)")
 	require.NoError(t, err)

server/store/sqlstore/migrations.go

@@ -9,7 +9,7 @@ import (
 )
 
 func (s *SQLStore) runMigrationRemoteID(remoteID string) error {
-	_, err := s.getQueryBuilder().Update("Users").Set("RemoteID", remoteID).Where(sq.And{
+	_, err := s.getMasterQueryBuilder().Update("Users").Set("RemoteID", remoteID).Where(sq.And{
 		sq.NotEq{"RemoteID": nil},
 		sq.NotEq{"RemoteID": ""},
 		sq.Expr("RemoteID NOT IN (SELECT remoteid FROM remoteclusters)"),
@@ -19,7 +19,7 @@ func (s *SQLStore) runMigrationRemoteID(remoteID string) error {
 }
 
 func (s *SQLStore) runSetEmailVerifiedToTrueForRemoteUsers(remoteID string) error {
-	_, err := s.getQueryBuilder().
+	_, err := s.getMasterQueryBuilder().
 		Update("Users").
 		Set("EmailVerified", true).
 		Where(sq.And{
@@ -37,7 +37,7 @@ const (
 
 func (s *SQLStore) runMSTeamUserIDDedup() error {
 	// get all users with duplicate msteamsuserid
-	rows, err := s.getQueryBuilder().Select(
+	rows, err := s.getReplicaQueryBuilder().Select(
 		"mmuserid",
 		"msteamsuserid",
 		"remoteid",
@@ -103,7 +103,7 @@ func (s *SQLStore) runMSTeamUserIDDedup() error {
 	}
 
 	s.api.LogInfo("Deleting duplicates")
-	_, err = s.getQueryBuilder().Delete(usersTableName).
+	_, err = s.getMasterQueryBuilder().Delete(usersTableName).
 		Where(orCond).
 		Exec()
 
@@ -129,7 +129,7 @@ func (s *SQLStore) ensureMigrationWhitelistedUsers() error {
 	// as being added to the old whitelist only happened after successful connection.
 
 	// has-connected users (presently and previously)
-	_, err = s.getQueryBuilder().
+	_, err = s.getMasterQueryBuilder().
 		Update(usersTableName).
 		Set("lastConnectAt", now.UnixMicro()).
 		Where(sq.Or{
@@ -142,7 +142,7 @@ func (s *SQLStore) ensureMigrationWhitelistedUsers() error {
 	}
 
 	// only previously-connected
-	_, err = s.getQueryBuilder().
+	_, err = s.getMasterQueryBuilder().
 		Update(usersTableName).
 		Set("lastDisconnectAt", now.UnixMicro()).
 		Where(sq.And{

server/store/sqlstore/migrations_test.go

@@ -48,7 +48,7 @@ func TestRunMSTeamUserIDDedup(t *testing.T) {
 	assert.NoError(err)
 	defer func() { _ = store.createMSTeamsUserIDUniqueIndex() }()
 
-	res, err := store.getQueryBuilder().Insert("users").
+	res, err := store.getMasterQueryBuilder().Insert("users").
 		Columns("id", "createat", "remoteid").
 		Values(userID1Remote, 100, "remote-id").
 		Values(userID2, 200, "").
@@ -62,7 +62,7 @@ func TestRunMSTeamUserIDDedup(t *testing.T) {
 
 	// ms teams user 1 will have all 3 users;
 	// user1 and 3 should be removed after the dedup because user 2 is a real user
-	res, err = store.getQueryBuilder().Insert(usersTableName).Columns("mmuserid", "msteamsuserid").
+	res, err = store.getMasterQueryBuilder().Insert(usersTableName).Columns("mmuserid", "msteamsuserid").
 		Values(userID1Remote, teamsUserA).
 		Values(userID2, teamsUserA).
 		Values(userID3Remote, teamsUserA).
@@ -82,7 +82,7 @@ func TestRunMSTeamUserIDDedup(t *testing.T) {
 	err = store.runMSTeamUserIDDedup()
 	assert.NoError(err)
 
-	rows, err := store.getQueryBuilder().Select("mmuserid", "msteamsuserid").From(usersTableName).Query()
+	rows, err := store.getMasterQueryBuilder().Select("mmuserid", "msteamsuserid").From(usersTableName).Query()
 	assert.NoError(err)
 	var found int
 	for rows.Next() {