mirrors/mostlymatter
1
0
Форкнуть 0
Код Задачи Запросы на слияние Действия Пакеты Проекты Релизы Вики Активность

[MM-39231] Migrate license store from gorp to sqlx in store/sqlstore/license_store.go (#19146)

Просмотр исходного кода 
Automatic Merge
Этот коммит содержится в:
Yair Facio
2021-12-12 00:10:31 -06:00
коммит произвёл GitHub
родитель 62d4d8166d
Коммит 3b3ab701b1
1 изменённых файлов: 23 добавлений и 8 удалений
Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

31
store/sqlstore/license_store.go
Просмотреть файл

@@ -52,9 +52,17 @@ func (ls SqlLicenseStore) Save(license *model.LicenseRecord) (*model.LicenseReco
return nil, errors.Wrap(err, "license_tosql") return nil, errors.Wrap(err, "license_tosql")
} }
var storedLicense model.LicenseRecord var storedLicense model.LicenseRecord
if err := ls.GetReplica().SelectOne(&storedLicense, queryString, args...); err != nil { if err := ls.GetReplicaX().Get(&storedLicense, queryString, args...); err != nil {
// Only insert if not exists // Only insert if not exists
if err := ls.GetMaster().Insert(license); err != nil { query, args, err := ls.getQueryBuilder().
Insert("Licenses").
Columns("Id", "CreateAt", "Bytes").
Values(license.Id, license.CreateAt, license.Bytes).
ToSql()
if err != nil {
return nil, errors.Wrap(err, "license_record_tosql")
}
if _, err := ls.GetMasterX().Exec(query, args...); err != nil {
return nil, errors.Wrapf(err, "failed to get License with licenseId=%s", license.Id) return nil, errors.Wrapf(err, "failed to get License with licenseId=%s", license.Id)
} }
return license, nil return license, nil
@@ -66,14 +74,21 @@ func (ls SqlLicenseStore) Save(license *model.LicenseRecord) (*model.LicenseReco
// If the license doesn't exist it returns a model.AppError with // If the license doesn't exist it returns a model.AppError with
// http.StatusNotFound in the StatusCode field. // http.StatusNotFound in the StatusCode field.
func (ls SqlLicenseStore) Get(id string) (*model.LicenseRecord, error) { func (ls SqlLicenseStore) Get(id string) (*model.LicenseRecord, error) {
obj, err := ls.GetReplica().Get(model.LicenseRecord{}, id) query := ls.getQueryBuilder().
Select("*").
From("Licenses").
Where(sq.Eq{"Id": id})
queryString, args, err := query.ToSql()
if err != nil { if err != nil {
return nil, errors.Wrapf(err, "failed to get License with licenseId=%s", id) return nil, errors.Wrap(err, "license_record_tosql")
} }
if obj == nil {
license := &model.LicenseRecord{}
if err := ls.GetReplicaX().Get(license, queryString, args...); err != nil {
return nil, store.NewErrNotFound("License", id) return nil, store.NewErrNotFound("License", id)
} }
return obj.(*model.LicenseRecord), nil return license, nil
} }
func (ls SqlLicenseStore) GetAll() ([]*model.LicenseRecord, error) { func (ls SqlLicenseStore) GetAll() ([]*model.LicenseRecord, error) {
@@ -86,8 +101,8 @@ func (ls SqlLicenseStore) GetAll() ([]*model.LicenseRecord, error) {
return nil, errors.Wrap(err, "license_tosql") return nil, errors.Wrap(err, "license_tosql")
} }
var licenses []*model.LicenseRecord licenses := []*model.LicenseRecord{}
if _, err := ls.GetReplica().Select(&licenses, queryString); err != nil { if err := ls.GetReplicaX().Select(&licenses, queryString); err != nil {
return nil, errors.Wrap(err, "failed to fetch licenses") return nil, errors.Wrap(err, "failed to fetch licenses")
} }