Move cluster, webhub and store out of Server (#20899)

store/sqlstore/store.go

@@ -440,6 +440,15 @@ func (ss *SqlStore) GetInternalReplicaDBs() []*sql.DB {
 	return dbs
 }
 
+func (ss *SqlStore) GetInternalReplicaDB() *sql.DB {
+	if len(ss.settings.DataSourceReplicas) == 0 || ss.lockedToMaster || !ss.hasLicense() {
+		return ss.GetMasterX().DB.DB
+	}
+
+	rrNum := atomic.AddInt64(&ss.rrCounter, 1) % int64(len(ss.ReplicaXs))
+	return ss.ReplicaXs[rrNum].DB.DB
+}
+
 func (ss *SqlStore) TotalMasterDbConnections() int {
 	return ss.GetMasterX().Stats().OpenConnections
 }

store/store.go

@@ -73,6 +73,7 @@ type Store interface {
 	GetInternalMasterDB() *sql.DB
 	// GetInternalReplicaDBs allows access to the raw replica DB
 	// handles for the multi-product architecture.
+	GetInternalReplicaDB() *sql.DB
 	GetInternalReplicaDBs() []*sql.DB
 	TotalMasterDbConnections() int
 	TotalReadDbConnections() int
@@ -974,7 +975,6 @@ type SharedChannelStore interface {
 // Paginate whether to paginate the results.
 // Page page requested, if results are paginated.
 // PerPage number of results per page, if paginated.
-//
 type ChannelSearchOpts struct {
 	Term                     string
 	NotAssociatedToGroup     string

store/storetest/mocks/Store.go

@@ -305,6 +305,22 @@ func (_m *Store) GetInternalMasterDB() *sql.DB {
 	return r0
 }
 
+// GetInternalReplicaDB provides a mock function with given fields:
+func (_m *Store) GetInternalReplicaDB() *sql.DB {
+	ret := _m.Called()
+
+	var r0 *sql.DB
+	if rf, ok := ret.Get(0).(func() *sql.DB); ok {
+		r0 = rf()
+	} else {
+		if ret.Get(0) != nil {
+			r0 = ret.Get(0).(*sql.DB)
+		}
+	}
+
+	return r0
+}
+
 // GetInternalReplicaDBs provides a mock function with given fields:
 func (_m *Store) GetInternalReplicaDBs() []*sql.DB {
 	ret := _m.Called()

store/storetest/store.go

@@ -107,6 +107,7 @@ func (s *Store) UnlockFromMaster()                       { /* do nothing */ }
 func (s *Store) DropAllTables()                          { /* do nothing */ }
 func (s *Store) GetDbVersion(bool) (string, error)       { return "", nil }
 func (s *Store) GetInternalMasterDB() *sql.DB            { return nil }
+func (s *Store) GetInternalReplicaDB() *sql.DB           { return nil }
 func (s *Store) GetInternalReplicaDBs() []*sql.DB        { return nil }
 func (s *Store) RecycleDBConnections(time.Duration)      {}
 func (s *Store) GetDBSchemaVersion() (int, error)        { return 1, nil }