mirror of
https://github.com/hanzoai/team.git
synced 2026-09-30 01:57:35 +00:00
zeekay
8feb24ea1f
The server module named github.com/hanzoai/team/server/v8 — a path that resolves to the retired workspace's repository, not this one — while server/public still published github.com/mattermost/mattermost/server/public. Both are github.com/hanzoteam/server/... now, which is where this code actually lives. That second half is what made the binary unbootable. Upstream's public path stayed claimed by upstream, mattermost-plugin-ai imported it, and the build linked public/model twice; public/plugin's gob.Register then ran twice under one name and panicked at init, before main. Reverting the rename traded the brand for a working binary. Rebranding the plugin alongside it needs no such trade: the agents fork is github.com/hanzoteam/agents/v2 against our public module, so no module named mattermost/mattermost is in the graph at all. Measured: `go list -deps ./cmd/mattermost` names exactly one public/model, and it is ours.
128 lines
2.5 KiB
Go
128 lines
2.5 KiB
Go
package pluginapi
|
|
|
|
import (
|
|
"database/sql"
|
|
"sync"
|
|
|
|
// import sql drivers
|
|
_ "github.com/lib/pq"
|
|
|
|
"github.com/hanzoteam/server/server/public/plugin"
|
|
"github.com/hanzoteam/server/server/public/shared/driver"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// StoreService exposes the underlying database.
|
|
type StoreService struct {
|
|
initializedMaster bool
|
|
initializedReplica bool
|
|
api plugin.API
|
|
driver plugin.Driver
|
|
mutex sync.Mutex
|
|
|
|
masterDB *sql.DB
|
|
replicaDB *sql.DB
|
|
}
|
|
|
|
// GetMasterDB gets the master database handle.
|
|
//
|
|
// Minimum server version: 5.16
|
|
func (s *StoreService) GetMasterDB() (*sql.DB, error) {
|
|
s.mutex.Lock()
|
|
defer s.mutex.Unlock()
|
|
|
|
if err := s.initializeMaster(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return s.masterDB, nil
|
|
}
|
|
|
|
// GetReplicaDB gets the replica database handle.
|
|
// Returns masterDB if a replica is not configured.
|
|
//
|
|
// Minimum server version: 5.16
|
|
func (s *StoreService) GetReplicaDB() (*sql.DB, error) {
|
|
s.mutex.Lock()
|
|
defer s.mutex.Unlock()
|
|
|
|
if err := s.initializeReplica(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if s.replicaDB != nil {
|
|
return s.replicaDB, nil
|
|
}
|
|
|
|
if err := s.initializeMaster(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return s.masterDB, nil
|
|
}
|
|
|
|
// Close closes any open resources. This method is idempotent.
|
|
func (s *StoreService) Close() error {
|
|
s.mutex.Lock()
|
|
defer s.mutex.Unlock()
|
|
|
|
if s.replicaDB != nil {
|
|
if err := s.replicaDB.Close(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
if s.masterDB != nil {
|
|
if err := s.masterDB.Close(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// DriverName returns the driver name for the datasource.
|
|
func (s *StoreService) DriverName() string {
|
|
return *s.api.GetConfig().SqlSettings.DriverName
|
|
}
|
|
|
|
func (s *StoreService) initializeMaster() error {
|
|
if s.initializedMaster {
|
|
return nil
|
|
}
|
|
|
|
if s.driver == nil {
|
|
return errors.New("no db driver was provided")
|
|
}
|
|
|
|
// Set up master db
|
|
db := sql.OpenDB(driver.NewConnector(s.driver, true /* IsMaster */))
|
|
if err := db.Ping(); err != nil {
|
|
return errors.Wrap(err, "failed to connect to master db")
|
|
}
|
|
s.masterDB = db
|
|
|
|
s.initializedMaster = true
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *StoreService) initializeReplica() error {
|
|
if s.initializedReplica {
|
|
return nil
|
|
}
|
|
|
|
config := s.api.GetUnsanitizedConfig()
|
|
// Set up replica db
|
|
if len(config.SqlSettings.DataSourceReplicas) > 0 {
|
|
db := sql.OpenDB(driver.NewConnector(s.driver, false /* IsMaster */))
|
|
if err := db.Ping(); err != nil {
|
|
return errors.Wrap(err, "failed to connect to replica db")
|
|
}
|
|
s.replicaDB = db
|
|
}
|
|
|
|
s.initializedReplica = true
|
|
return nil
|
|
}
|