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

Cluster Discovery store sqlx migration (#18605)

Просмотр исходного кода 
Automatic Merge
Этот коммит содержится в:
Archit Mathur
2021-10-07 20:20:03 +05:30
коммит произвёл GitHub
родитель 9bbac354d8
Коммит d13b3604ea
1 изменённых файлов: 21 добавлений и 9 удалений
Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

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

@@ -35,7 +35,13 @@ func (s sqlClusterDiscoveryStore) Save(ClusterDiscovery *model.ClusterDiscovery)
return err return err
} }
if err := s.GetMaster().Insert(ClusterDiscovery); err != nil { if _, err := s.GetMasterX().NamedExec(`
INSERT INTO
ClusterDiscovery
(Id, Type, ClusterName, Hostname, GossipPort, Port, CreateAt, LastPingAt)
VALUES
(:Id, :Type, :ClusterName, :Hostname, :GossipPort, :Port, :CreateAt, :LastPingAt)
`, ClusterDiscovery); err != nil {
return errors.Wrap(err, "failed to save ClusterDiscovery") return errors.Wrap(err, "failed to save ClusterDiscovery")
} }
return nil return nil
@@ -53,10 +59,16 @@ func (s sqlClusterDiscoveryStore) Delete(ClusterDiscovery *model.ClusterDiscover
return false, errors.Wrap(err, "cluster_discovery_tosql") return false, errors.Wrap(err, "cluster_discovery_tosql")
} }
count, err := s.GetMaster().SelectInt(queryString, args...) res, err := s.GetMasterX().Exec(queryString, args...)
if err != nil { if err != nil {
return false, errors.Wrap(err, "failed to delete ClusterDiscovery") return false, errors.Wrap(err, "failed to delete ClusterDiscovery")
} }
count, err := res.RowsAffected()
if err != nil {
return false, errors.Wrap(err, "failed to count rows affected")
}
if count == 0 { if count == 0 {
return false, nil return false, nil
} }
@@ -76,8 +88,8 @@ func (s sqlClusterDiscoveryStore) Exists(ClusterDiscovery *model.ClusterDiscover
return false, errors.Wrap(err, "cluster_discovery_tosql") return false, errors.Wrap(err, "cluster_discovery_tosql")
} }
count, err := s.GetMaster().SelectInt(queryString, args...) var count int
if err != nil { if err := s.GetMasterX().Get(&count, queryString, args...); err != nil {
return false, errors.Wrap(err, "failed to count ClusterDiscovery") return false, errors.Wrap(err, "failed to count ClusterDiscovery")
} }
if count == 0 { if count == 0 {
@@ -99,9 +111,9 @@ func (s sqlClusterDiscoveryStore) GetAll(ClusterDiscoveryType, clusterName strin
return nil, errors.Wrap(err, "cluster_discovery_tosql") return nil, errors.Wrap(err, "cluster_discovery_tosql")
} }
var list []*model.ClusterDiscovery list := []*model.ClusterDiscovery{}
if _, err := s.GetMaster().Select(&list, queryString, args...); err != nil { if err := s.GetMasterX().Select(&list, queryString, args...); err != nil {
return nil, errors.Wrapf(err, "failed to find ClusterDiscovery") return nil, errors.Wrap(err, "failed to find ClusterDiscovery")
} }
return list, nil return list, nil
} }
@@ -119,7 +131,7 @@ func (s sqlClusterDiscoveryStore) SetLastPingAt(ClusterDiscovery *model.ClusterD
return errors.Wrap(err, "cluster_discovery_tosql") return errors.Wrap(err, "cluster_discovery_tosql")
} }
if _, err := s.GetMaster().Exec(queryString, args...); err != nil { if _, err := s.GetMasterX().Exec(queryString, args...); err != nil {
return errors.Wrap(err, "failed to update ClusterDiscovery") return errors.Wrap(err, "failed to update ClusterDiscovery")
} }
return nil return nil
@@ -135,7 +147,7 @@ func (s sqlClusterDiscoveryStore) Cleanup() error {
return errors.Wrap(err, "cluster_discovery_tosql") return errors.Wrap(err, "cluster_discovery_tosql")
} }
if _, err := s.GetMaster().Exec(queryString, args...); err != nil { if _, err := s.GetMasterX().Exec(queryString, args...); err != nil {
return errors.Wrap(err, "failed to delete ClusterDiscoveries") return errors.Wrap(err, "failed to delete ClusterDiscoveries")
} }
return nil return nil