mirror of
https://github.com/hanzoai/team.git
synced 2026-09-30 01:57:35 +00:00
Jesse HallamandGitHub
f45e9ad165
Drop the legacy `X` suffix from `GetMasterX` and `GetReplicaX`. The presence of the suffix suggests there's a `non-X` version: but in fact we migrated these away a long time ago, so remove the cognitive overhead. As an aside, this additionally helps avoid trip up LLMs that interpret this as "something to fix".
135 lines
4.0 KiB
Go
135 lines
4.0 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package sqlstore
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
|
|
sq "github.com/mattermost/squirrel"
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/mattermost/mattermost/server/public/model"
|
|
"github.com/mattermost/mattermost/server/v8/channels/store"
|
|
)
|
|
|
|
type SqlSystemStore struct {
|
|
*SqlStore
|
|
}
|
|
|
|
func newSqlSystemStore(sqlStore *SqlStore) store.SystemStore {
|
|
return &SqlSystemStore{sqlStore}
|
|
}
|
|
|
|
func (s SqlSystemStore) Save(system *model.System) error {
|
|
query := "INSERT INTO Systems (Name, Value) VALUES (:Name, :Value)"
|
|
if _, err := s.GetMaster().NamedExec(query, system); err != nil {
|
|
return errors.Wrapf(err, "failed to save system property with name=%s", system.Name)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s SqlSystemStore) SaveOrUpdate(system *model.System) error {
|
|
query := s.getQueryBuilder().
|
|
Insert("Systems").
|
|
Columns("Name", "Value").
|
|
Values(system.Name, system.Value)
|
|
|
|
if s.DriverName() == model.DatabaseDriverMysql {
|
|
query = query.SuffixExpr(sq.Expr("ON DUPLICATE KEY UPDATE Value = ?", system.Value))
|
|
} else {
|
|
query = query.SuffixExpr(sq.Expr("ON CONFLICT (name) DO UPDATE SET Value = ?", system.Value))
|
|
}
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return errors.Wrap(err, "system_tosql")
|
|
}
|
|
|
|
if _, err := s.GetMaster().Exec(queryString, args...); err != nil {
|
|
return errors.Wrap(err, "failed to upsert system property")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s SqlSystemStore) Update(system *model.System) error {
|
|
query := "UPDATE Systems SET Value=:Value WHERE Name=:Name"
|
|
if _, err := s.GetMaster().NamedExec(query, system); err != nil {
|
|
return errors.Wrapf(err, "failed to update system property with name=%s", system.Name)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s SqlSystemStore) Get() (model.StringMap, error) {
|
|
systems := []model.System{}
|
|
props := make(model.StringMap)
|
|
|
|
if err := s.GetReplica().Select(&systems, "SELECT * FROM Systems"); err != nil {
|
|
return nil, errors.Wrap(err, "failed to get System list")
|
|
}
|
|
|
|
for _, prop := range systems {
|
|
props[prop.Name] = prop.Value
|
|
}
|
|
|
|
return props, nil
|
|
}
|
|
|
|
func (s SqlSystemStore) GetByName(name string) (*model.System, error) {
|
|
var system model.System
|
|
if err := s.GetMaster().Get(&system, "SELECT * FROM Systems WHERE Name = ?", name); err != nil {
|
|
if err == sql.ErrNoRows {
|
|
return nil, store.NewErrNotFound("System", fmt.Sprintf("name=%s", system.Name))
|
|
}
|
|
return nil, errors.Wrapf(err, "failed to get system property with name=%s", system.Name)
|
|
}
|
|
|
|
return &system, nil
|
|
}
|
|
|
|
func (s SqlSystemStore) PermanentDeleteByName(name string) (*model.System, error) {
|
|
var system model.System
|
|
if _, err := s.GetMaster().Exec("DELETE FROM Systems WHERE Name = ?", name); err != nil {
|
|
return nil, errors.Wrapf(err, "failed to permanent delete system property with name=%s", system.Name)
|
|
}
|
|
|
|
return &system, nil
|
|
}
|
|
|
|
// InsertIfExists inserts a given system value if it does not already exist. If a value
|
|
// already exists, it returns the old one, else returns the new one.
|
|
func (s SqlSystemStore) InsertIfExists(system *model.System) (_ *model.System, err error) {
|
|
tx, err := s.GetMaster().BeginXWithIsolation(&sql.TxOptions{
|
|
Isolation: sql.LevelSerializable,
|
|
})
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "begin_transaction")
|
|
}
|
|
defer finalizeTransactionX(tx, &err)
|
|
|
|
var origSystem model.System
|
|
if err := tx.Get(&origSystem, `SELECT * FROM Systems
|
|
WHERE Name = ?`, system.Name); err != nil && err != sql.ErrNoRows {
|
|
return nil, errors.Wrapf(err, "failed to get system property with name=%s", system.Name)
|
|
}
|
|
|
|
if origSystem.Value != "" {
|
|
// Already a value exists, return that.
|
|
return &origSystem, nil
|
|
}
|
|
|
|
// Key does not exist, need to insert.
|
|
if _, err := tx.NamedExec("INSERT INTO Systems (Name, Value) VALUES (:Name, :Value)", system); err != nil {
|
|
return nil, errors.Wrapf(err, "failed to save system property with name=%s", system.Name)
|
|
}
|
|
|
|
if err := tx.Commit(); err != nil {
|
|
return nil, errors.Wrap(err, "commit_transaction")
|
|
}
|
|
return system, nil
|
|
}
|