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".
95 lines
2.7 KiB
Go
95 lines
2.7 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package sqlstore
|
|
|
|
import (
|
|
sq "github.com/mattermost/squirrel"
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/mattermost/mattermost/server/public/model"
|
|
"github.com/mattermost/mattermost/server/public/shared/request"
|
|
"github.com/mattermost/mattermost/server/v8/channels/store"
|
|
)
|
|
|
|
// SqlLicenseStore encapsulates the database writes and reads for
|
|
// model.LicenseRecord objects.
|
|
type SqlLicenseStore struct {
|
|
*SqlStore
|
|
}
|
|
|
|
func newSqlLicenseStore(sqlStore *SqlStore) store.LicenseStore {
|
|
return &SqlLicenseStore{sqlStore}
|
|
}
|
|
|
|
// Save validates and stores the license instance in the database. The Id
|
|
// and Bytes fields are mandatory. The Bytes field is limited to a maximum
|
|
// of 10000 bytes. Provided license is saved only if missing.
|
|
func (ls SqlLicenseStore) Save(license *model.LicenseRecord) error {
|
|
license.PreSave()
|
|
if err := license.IsValid(); err != nil {
|
|
return err
|
|
}
|
|
|
|
query := ls.getQueryBuilder().
|
|
Insert("Licenses").
|
|
Columns("Id", "CreateAt", "Bytes").
|
|
Values(license.Id, license.CreateAt, license.Bytes)
|
|
|
|
if ls.DriverName() == model.DatabaseDriverMysql {
|
|
query = query.SuffixExpr(sq.Expr("ON DUPLICATE KEY UPDATE Id=Id"))
|
|
} else {
|
|
query = query.SuffixExpr(sq.Expr("ON CONFLICT (Id) DO NOTHING"))
|
|
}
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return errors.Wrap(err, "license_tosql")
|
|
}
|
|
|
|
if _, err := ls.GetMaster().Exec(queryString, args...); err != nil {
|
|
return errors.Wrapf(err, "failed to insert License with licenseId=%s", license.Id)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Get obtains the license with the provided id parameter from the database.
|
|
// If the license doesn't exist it returns a model.AppError with
|
|
// http.StatusNotFound in the StatusCode field.
|
|
func (ls SqlLicenseStore) Get(c request.CTX, id string) (*model.LicenseRecord, error) {
|
|
query := ls.getQueryBuilder().
|
|
Select("Id, CreateAt, Bytes").
|
|
From("Licenses").
|
|
Where(sq.Eq{"Id": id})
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "license_record_tosql")
|
|
}
|
|
|
|
license := &model.LicenseRecord{}
|
|
if err := ls.DBXFromContext(c.Context()).Get(license, queryString, args...); err != nil {
|
|
return nil, store.NewErrNotFound("License", id)
|
|
}
|
|
return license, nil
|
|
}
|
|
|
|
func (ls SqlLicenseStore) GetAll() ([]*model.LicenseRecord, error) {
|
|
query := ls.getQueryBuilder().
|
|
Select("Id, CreateAt, Bytes").
|
|
From("Licenses")
|
|
|
|
queryString, _, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "license_tosql")
|
|
}
|
|
|
|
licenses := []*model.LicenseRecord{}
|
|
if err := ls.GetReplica().Select(&licenses, queryString); err != nil {
|
|
return nil, errors.Wrap(err, "failed to fetch licenses")
|
|
}
|
|
|
|
return licenses, nil
|
|
}
|