loadbalancerclusters.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633
  1. // Copyright 2019 Yunion
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. package models
  15. import (
  16. "context"
  17. "reflect"
  18. "yunion.io/x/jsonutils"
  19. "yunion.io/x/log"
  20. "yunion.io/x/pkg/errors"
  21. "yunion.io/x/pkg/gotypes"
  22. "yunion.io/x/pkg/util/seclib"
  23. "yunion.io/x/sqlchemy"
  24. "yunion.io/x/onecloud/pkg/apis"
  25. api "yunion.io/x/onecloud/pkg/apis/compute"
  26. "yunion.io/x/onecloud/pkg/cloudcommon/db"
  27. "yunion.io/x/onecloud/pkg/cloudcommon/validators"
  28. "yunion.io/x/onecloud/pkg/httperrors"
  29. "yunion.io/x/onecloud/pkg/mcclient"
  30. "yunion.io/x/onecloud/pkg/util/logclient"
  31. "yunion.io/x/onecloud/pkg/util/stringutils2"
  32. )
  33. // +onecloud:swagger-gen-model-singular=loadbalancercluster
  34. // +onecloud:swagger-gen-model-plural=loadbalancerclusters
  35. type SLoadbalancerClusterManager struct {
  36. db.SStandaloneResourceBaseManager
  37. SZoneResourceBaseManager
  38. SWireResourceBaseManager
  39. }
  40. var LoadbalancerClusterManager *SLoadbalancerClusterManager
  41. func init() {
  42. gotypes.RegisterSerializable(reflect.TypeOf(&SLoadbalancerClusterParams{}), func() gotypes.ISerializable {
  43. return &SLoadbalancerClusterParams{}
  44. })
  45. LoadbalancerClusterManager = &SLoadbalancerClusterManager{
  46. SStandaloneResourceBaseManager: db.NewStandaloneResourceBaseManager(
  47. SLoadbalancerCluster{},
  48. "loadbalancerclusters_tbl",
  49. "loadbalancercluster",
  50. "loadbalancerclusters",
  51. ),
  52. }
  53. LoadbalancerClusterManager.SetVirtualObject(LoadbalancerClusterManager)
  54. }
  55. type SLoadbalancerCluster struct {
  56. db.SStandaloneResourceBase
  57. SZoneResourceBase
  58. SWireResourceBase `width:"36" charset:"ascii" nullable:"true" list:"admin" create:"optional" update:"admin"`
  59. Params *SLoadbalancerClusterParams `nullable:"true" create:"optional" list:"admin" get:"admin"`
  60. }
  61. // 负载均衡集群列表
  62. func (man *SLoadbalancerClusterManager) ListItemFilter(
  63. ctx context.Context,
  64. q *sqlchemy.SQuery,
  65. userCred mcclient.TokenCredential,
  66. query api.LoadbalancerClusterListInput,
  67. ) (*sqlchemy.SQuery, error) {
  68. var err error
  69. q, err = man.SStandaloneResourceBaseManager.ListItemFilter(ctx, q, userCred, query.StandaloneResourceListInput)
  70. if err != nil {
  71. return nil, errors.Wrap(err, "SStandaloneResourceBaseManager.ListItemFilter")
  72. }
  73. q, err = man.SZoneResourceBaseManager.ListItemFilter(ctx, q, userCred, query.ZonalFilterListInput)
  74. if err != nil {
  75. return nil, errors.Wrap(err, "SZoneResourceBaseManager.ListItemFilter")
  76. }
  77. wireQuery := api.WireFilterListInput{
  78. WireFilterListBase: query.WireFilterListBase,
  79. }
  80. q, err = man.SWireResourceBaseManager.ListItemFilter(ctx, q, userCred, wireQuery)
  81. if err != nil {
  82. return nil, errors.Wrap(err, "SWireResourceBaseManager.ListItemFilter")
  83. }
  84. return q, nil
  85. }
  86. func (man *SLoadbalancerClusterManager) OrderByExtraFields(
  87. ctx context.Context,
  88. q *sqlchemy.SQuery,
  89. userCred mcclient.TokenCredential,
  90. query api.LoadbalancerClusterListInput,
  91. ) (*sqlchemy.SQuery, error) {
  92. var err error
  93. q, err = man.SStandaloneResourceBaseManager.OrderByExtraFields(ctx, q, userCred, query.StandaloneResourceListInput)
  94. if err != nil {
  95. return nil, errors.Wrap(err, "SStandaloneResourceBaseManager.OrderByExtraFields")
  96. }
  97. q, err = man.SZoneResourceBaseManager.OrderByExtraFields(ctx, q, userCred, query.ZonalFilterListInput)
  98. if err != nil {
  99. return nil, errors.Wrap(err, "SZoneResourceBaseManager.OrderByExtraFields")
  100. }
  101. wireQuery := api.WireFilterListInput{
  102. WireFilterListBase: query.WireFilterListBase,
  103. }
  104. q, err = man.SWireResourceBaseManager.OrderByExtraFields(ctx, q, userCred, wireQuery)
  105. if err != nil {
  106. return nil, errors.Wrap(err, "SWireResourceBaseManager.OrderByExtraFields")
  107. }
  108. return q, nil
  109. }
  110. func (man *SLoadbalancerClusterManager) QueryDistinctExtraField(q *sqlchemy.SQuery, field string) (*sqlchemy.SQuery, error) {
  111. var err error
  112. q, err = man.SStandaloneResourceBaseManager.QueryDistinctExtraField(q, field)
  113. if err == nil {
  114. return q, nil
  115. }
  116. q, err = man.SZoneResourceBaseManager.QueryDistinctExtraField(q, field)
  117. if err == nil {
  118. return q, nil
  119. }
  120. q, err = man.SWireResourceBaseManager.QueryDistinctExtraField(q, field)
  121. if err == nil {
  122. return q, nil
  123. }
  124. return q, httperrors.ErrNotFound
  125. }
  126. func (man *SLoadbalancerClusterManager) ValidateCreateData(
  127. ctx context.Context,
  128. userCred mcclient.TokenCredential,
  129. ownerId mcclient.IIdentityProvider,
  130. query jsonutils.JSONObject,
  131. data *jsonutils.JSONDict,
  132. ) (*jsonutils.JSONDict, error) {
  133. zoneV := validators.NewModelIdOrNameValidator("zone", "zone", ownerId)
  134. wireV := validators.NewModelIdOrNameValidator("wire", "wire", ownerId)
  135. vs := []validators.IValidator{
  136. zoneV,
  137. wireV.Optional(true),
  138. }
  139. for _, v := range vs {
  140. if err := v.Validate(ctx, data); err != nil {
  141. return nil, err
  142. }
  143. }
  144. zone := zoneV.Model.(*SZone)
  145. if zone.ExternalId != "" {
  146. return nil, httperrors.NewInputParameterError("allow only internal zone, got %s(%s)", zone.Name, zone.Id)
  147. }
  148. if wireV.Model != nil {
  149. wire := wireV.Model.(*SWire)
  150. if wire.ZoneId != zone.Id {
  151. return nil, httperrors.NewInputParameterError("wire zone must match zone parameter, got %s, want %s(%s)",
  152. wire.ZoneId, zone.Name, zone.Id)
  153. }
  154. }
  155. input := apis.StandaloneResourceCreateInput{}
  156. err := data.Unmarshal(&input)
  157. if err != nil {
  158. return nil, httperrors.NewInternalServerError("unmarshal StandaloneResourceCreateInput fail %s", err)
  159. }
  160. input, err = man.SStandaloneResourceBaseManager.ValidateCreateData(ctx, userCred, ownerId, query, input)
  161. if err != nil {
  162. return nil, err
  163. }
  164. data.Update(jsonutils.Marshal(input))
  165. return data, nil
  166. }
  167. func (cluster *SLoadbalancerCluster) PostCreate(
  168. ctx context.Context,
  169. userCred mcclient.TokenCredential,
  170. ownerId mcclient.IIdentityProvider,
  171. query jsonutils.JSONObject,
  172. data jsonutils.JSONObject,
  173. ) {
  174. err := cluster.selfInitParams()
  175. if err != nil {
  176. log.Errorf("fail to generate cluster params: %s", err)
  177. logclient.AddSimpleActionLog(cluster, logclient.ACT_CREATE, err, userCred, false)
  178. } else {
  179. logclient.AddSimpleActionLog(cluster, logclient.ACT_CREATE, cluster.Params, userCred, true)
  180. }
  181. }
  182. func (lbc *SLoadbalancerCluster) ValidateUpdateData(
  183. ctx context.Context,
  184. userCred mcclient.TokenCredential,
  185. query jsonutils.JSONObject,
  186. data *jsonutils.JSONDict,
  187. ) (*jsonutils.JSONDict, error) {
  188. wireV := validators.NewModelIdOrNameValidator("wire", "wire", lbc.GetOwnerId())
  189. wireV.Optional(true)
  190. if err := wireV.Validate(ctx, data); err != nil {
  191. return nil, err
  192. }
  193. if wireV.Model != nil {
  194. wire := wireV.Model.(*SWire)
  195. if wire.ZoneId != lbc.ZoneId {
  196. return nil, httperrors.NewInputParameterError("zone of wire must be %s, got %s", lbc.ZoneId, wire.ZoneId)
  197. }
  198. var from string
  199. if lbc.WireId != "" {
  200. from = "from " + lbc.WireId + " "
  201. }
  202. log.Infof("changing wire attribute of lbcluster %s(%s) %sto %s(%s)",
  203. lbc.Name, lbc.Id, from, wire.Name, wire.Id)
  204. }
  205. input := apis.StandaloneResourceBaseUpdateInput{}
  206. err := data.Unmarshal(&input)
  207. if err != nil {
  208. return nil, errors.Wrap(err, "Unmarshal")
  209. }
  210. input, err = lbc.SStandaloneResourceBase.ValidateUpdateData(ctx, userCred, query, input)
  211. if err != nil {
  212. return nil, errors.Wrap(err, "SStandaloneResourceBase.ValidateUpdateData")
  213. }
  214. data.Update(jsonutils.Marshal(input))
  215. return data, nil
  216. }
  217. func (lbc *SLoadbalancerCluster) ValidateDeleteCondition(ctx context.Context, info jsonutils.JSONObject) error {
  218. refCnts, err := lbc.refCounts()
  219. if err != nil {
  220. return errors.Wrap(err, "refCounts")
  221. }
  222. errs := make([]error, 0)
  223. for k, cnt := range refCnts {
  224. errs = append(errs, httperrors.NewResourceBusyError("lbcluster %s(%s) is still referred to by %d %s",
  225. lbc.Id, lbc.Name, cnt, k))
  226. }
  227. if len(errs) > 0 {
  228. return errors.NewAggregate(errs)
  229. }
  230. return lbc.SStandaloneResourceBase.ValidateDeleteCondition(ctx, nil)
  231. }
  232. func (lbc *SLoadbalancerCluster) refCounts() (map[string]int, error) {
  233. ret := make(map[string]int)
  234. men := []db.IModelManager{
  235. LoadbalancerManager,
  236. LoadbalancerAgentManager,
  237. }
  238. for _, man := range men {
  239. q := man.Query().Equals("cluster_id", lbc.Id)
  240. n, err := q.CountWithError()
  241. if err != nil {
  242. return nil, httperrors.NewInternalServerError("get lbcluster refcount fail %v", err)
  243. }
  244. if n > 0 {
  245. ret[man.KeywordPlural()] = n
  246. }
  247. }
  248. return ret, nil
  249. }
  250. func (man *SLoadbalancerClusterManager) FetchCustomizeColumns(
  251. ctx context.Context,
  252. userCred mcclient.TokenCredential,
  253. query jsonutils.JSONObject,
  254. objs []interface{},
  255. fields stringutils2.SSortedStrings,
  256. isList bool,
  257. ) []api.LoadbalancerClusterDetails {
  258. rows := make([]api.LoadbalancerClusterDetails, len(objs))
  259. stdRows := man.SStandaloneResourceBaseManager.FetchCustomizeColumns(ctx, userCred, query, objs, fields, isList)
  260. zoneRows := man.SZoneResourceBaseManager.FetchCustomizeColumns(ctx, userCred, query, objs, fields, isList)
  261. wireRows := man.SWireResourceBaseManager.FetchCustomizeColumns(ctx, userCred, query, objs, fields, isList)
  262. for i := range rows {
  263. rows[i] = api.LoadbalancerClusterDetails{
  264. StandaloneResourceDetails: stdRows[i],
  265. ZoneResourceInfo: zoneRows[i],
  266. WireResourceInfoBase: wireRows[i].WireResourceInfoBase,
  267. }
  268. lbc := objs[i].(*SLoadbalancerCluster)
  269. rows[i].RefCounts, _ = lbc.refCounts()
  270. }
  271. return rows
  272. }
  273. func (lbc *SLoadbalancerCluster) CustomizeDelete(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) error {
  274. lbagents := []SLoadbalancerAgent{}
  275. q := LoadbalancerAgentManager.Query().Equals("cluster_id", lbc.Id)
  276. if err := db.FetchModelObjects(LoadbalancerAgentManager, q, &lbagents); err != nil {
  277. return errors.Wrapf(err, "lbcluster %s(%s): find lbagents", lbc.Name, lbc.Id)
  278. }
  279. for i := range lbagents {
  280. lbagent := &lbagents[i]
  281. if err := lbagent.ValidateDeleteCondition(ctx, nil); err != nil {
  282. return errors.Wrapf(err, "lbagent %s(%s): validate delete", lbagent.Name, lbagent.Id)
  283. }
  284. if err := lbagent.CustomizeDelete(ctx, userCred, query, data); err != nil {
  285. return errors.Wrapf(err, "lbagent %s(%s): customize delete", lbagent.Name, lbagent.Id)
  286. }
  287. lbagent.PreDelete(ctx, userCred)
  288. if err := lbagent.Delete(ctx, userCred); err != nil {
  289. return errors.Wrapf(err, "lbagent %s(%s): delete", lbagent.Name, lbagent.Id)
  290. }
  291. lbagent.PostDelete(ctx, userCred)
  292. }
  293. return lbc.SStandaloneResourceBase.CustomizeDelete(ctx, userCred, query, data)
  294. }
  295. func (man *SLoadbalancerClusterManager) FindByZoneId(zoneId string) []SLoadbalancerCluster {
  296. r := []SLoadbalancerCluster{}
  297. q := man.Query().Equals("zone_id", zoneId)
  298. if err := db.FetchModelObjects(man, q, &r); err != nil {
  299. log.Errorf("find lbclusters by zone_id %s: %v", zoneId, err)
  300. return nil
  301. }
  302. return r
  303. }
  304. func (man *SLoadbalancerClusterManager) findByVrrpRouterIdInZone(zoneId string, routerId int) ([]*SLoadbalancerCluster, error) {
  305. r := make([]*SLoadbalancerCluster, 0)
  306. peerClusters := man.FindByZoneId(zoneId)
  307. for i := range peerClusters {
  308. peerCluster := &peerClusters[i]
  309. if peerCluster.Params.VirtualRouterId == routerId {
  310. r = append(r, peerCluster)
  311. }
  312. }
  313. return r, nil
  314. }
  315. func (man *SLoadbalancerClusterManager) getVrrpRouterIdsInZone(zoneId string) ([]int, error) {
  316. q := man.Query().IsNotEmpty("params").Equals("zone_id", zoneId)
  317. clusters := make([]SLoadbalancerCluster, 0)
  318. err := db.FetchModelObjects(man, q, &clusters)
  319. if err != nil {
  320. return nil, errors.Wrap(err, "initParams.FetchModelObjects")
  321. }
  322. ret := make([]int, 0)
  323. for i := range clusters {
  324. if clusters[i].Params.VirtualRouterId > 0 {
  325. ret = append(ret, clusters[i].Params.VirtualRouterId)
  326. }
  327. }
  328. return ret, nil
  329. }
  330. func isInArray[K comparable](v K, arr []K) bool {
  331. for i := range arr {
  332. if arr[i] == v {
  333. return true
  334. }
  335. }
  336. return false
  337. }
  338. func (man *SLoadbalancerClusterManager) newVrrpRouterIdsInZone(zoneId string) (int, error) {
  339. idList, err := man.getVrrpRouterIdsInZone(zoneId)
  340. if err != nil {
  341. return -1, errors.Wrap(err, "getVrrpRouterIdsInZone")
  342. }
  343. for i := 17; i < 250; i++ {
  344. if !isInArray(i, idList) {
  345. return i, nil
  346. }
  347. }
  348. return -1, errors.Wrapf(httperrors.ErrNotFound, "no available vrrp router id in zone %s", zoneId)
  349. }
  350. func (man *SLoadbalancerClusterManager) getLoadbalancerAgents(clusterId string) ([]SLoadbalancerAgent, error) {
  351. r := []SLoadbalancerAgent{}
  352. q := LoadbalancerAgentManager.Query().Equals("cluster_id", clusterId)
  353. err := db.FetchModelObjects(LoadbalancerAgentManager, q, &r)
  354. if err != nil {
  355. return nil, err
  356. }
  357. return r, nil
  358. }
  359. func (man *SLoadbalancerClusterManager) InitializeData() error {
  360. // find existing lb with empty clusterid
  361. lbs := []SLoadbalancer{}
  362. lbQ := LoadbalancerManager.Query()
  363. vpcs := VpcManager.Query().SubQuery()
  364. lbQ = lbQ.Join(vpcs, sqlchemy.Equals(lbQ.Field("vpc_id"), vpcs.Field("id")))
  365. lbQ = lbQ.Filter(sqlchemy.IsFalse(lbQ.Field("pending_deleted")))
  366. lbQ = lbQ.Filter(sqlchemy.IsNullOrEmpty(vpcs.Field("manager_id")))
  367. lbQ = lbQ.Filter(sqlchemy.IsNullOrEmpty(lbQ.Field("cluster_id")))
  368. if err := db.FetchModelObjects(LoadbalancerManager, lbQ, &lbs); err != nil {
  369. return errors.Wrap(err, "find lb with empty cluster_id")
  370. }
  371. // create 1 cluster for each zone
  372. zoneCluster := map[string]*SLoadbalancerCluster{}
  373. for i := range lbs {
  374. lb := &lbs[i]
  375. zoneId := lb.ZoneId
  376. if zoneId == "" {
  377. // just in case
  378. log.Warningf("found lb with empty zone_id: %s(%s)", lb.Name, lb.Id)
  379. continue
  380. }
  381. lbc, ok := zoneCluster[zoneId]
  382. if !ok {
  383. lbcs := man.FindByZoneId(zoneId)
  384. if len(lbcs) == 0 {
  385. m, err := db.NewModelObject(man)
  386. if err != nil {
  387. return errors.Wrap(err, "new model object")
  388. }
  389. lbc = m.(*SLoadbalancerCluster)
  390. lbc.Name = "auto-lbc-" + zoneId
  391. lbc.ZoneId = zoneId
  392. if err := man.TableSpec().Insert(context.TODO(), lbc); err != nil {
  393. return errors.Wrap(err, "insert lbcluster model")
  394. }
  395. } else {
  396. if len(lbcs) > 1 {
  397. log.Infof("zone %s has %d lbclusters, select one", zoneId, len(lbcs))
  398. }
  399. lbc = &lbcs[0]
  400. }
  401. zoneCluster[zoneId] = lbc
  402. }
  403. if _, err := db.UpdateWithLock(context.Background(), lb, func() error {
  404. lb.ClusterId = lbc.Id
  405. return nil
  406. }); err != nil {
  407. return errors.Wrapf(err, "lb %s(%s): assign cluster: %s(%s)", lb.Name, lb.Name, lbc.Name, lbc.Id)
  408. }
  409. }
  410. // associate existing lbagents with the cluster
  411. if len(zoneCluster) > 1 {
  412. log.Warningf("found %d zones with lb not assigned to any lbcluster, skip assigning lbagent to lbcluster", len(zoneCluster))
  413. return nil
  414. }
  415. for _, lbc := range zoneCluster {
  416. lbagents := []SLoadbalancerAgent{}
  417. q := LoadbalancerAgentManager.Query().
  418. IsNullOrEmpty("cluster_id")
  419. if err := db.FetchModelObjects(LoadbalancerAgentManager, q, &lbagents); err != nil {
  420. return errors.Wrap(err, "find lbagents with empty cluster_id")
  421. }
  422. for i := range lbagents {
  423. lbagent := &lbagents[i]
  424. if _, err := db.UpdateWithLock(context.Background(), lbagent, func() error {
  425. lbagent.ClusterId = lbc.Id
  426. return nil
  427. }); err != nil {
  428. return errors.Wrapf(err, "lbagent %s(%s): assign cluster: %s(%s)",
  429. lbagent.Name, lbagent.Id, lbc.Name, lbc.Id)
  430. }
  431. }
  432. }
  433. man.initParams()
  434. return nil
  435. }
  436. func (man *SLoadbalancerClusterManager) initParams() error {
  437. q := man.Query().IsNullOrEmpty("params")
  438. clusters := make([]SLoadbalancerCluster, 0)
  439. err := db.FetchModelObjects(man, q, &clusters)
  440. if err != nil {
  441. return errors.Wrap(err, "initParams.FetchModelObjects")
  442. }
  443. for i := range clusters {
  444. err := clusters[i].initParams()
  445. if err != nil {
  446. return errors.Wrap(err, "cluster initParams")
  447. }
  448. }
  449. return nil
  450. }
  451. func (cluster *SLoadbalancerCluster) initParams() error {
  452. lbagents, err := LoadbalancerClusterManager.getLoadbalancerAgents(cluster.Id)
  453. if err != nil {
  454. return errors.Wrap(err, "getLoadbalancerAgents")
  455. }
  456. if len(lbagents) == 0 {
  457. // generate params
  458. return cluster.selfInitParams()
  459. }
  460. _, err = db.Update(cluster, func() error {
  461. cluster.Params = &lbagents[0].Params.Vrrp.SLoadbalancerClusterParams
  462. return nil
  463. })
  464. if err != nil {
  465. return errors.Wrap(err, "UpdateWithLock")
  466. }
  467. return nil
  468. }
  469. func (cluster *SLoadbalancerCluster) selfInitParams() error {
  470. newRouterId, err := LoadbalancerClusterManager.newVrrpRouterIdsInZone(cluster.ZoneId)
  471. if err != nil {
  472. return errors.Wrap(err, "newVrrpRouterIdsInZone")
  473. }
  474. params := SLoadbalancerClusterParams{
  475. VirtualRouterId: newRouterId,
  476. Preempt: false,
  477. AdvertInt: 5,
  478. Pass: seclib.RandomPassword(6),
  479. GarpMasterRefresh: 29,
  480. }
  481. _, err = db.Update(cluster, func() error {
  482. cluster.Params = &params
  483. return nil
  484. })
  485. if err != nil {
  486. return errors.Wrap(err, "UpdateWithLock")
  487. }
  488. return nil
  489. }
  490. func (manager *SLoadbalancerClusterManager) ListItemExportKeys(ctx context.Context,
  491. q *sqlchemy.SQuery,
  492. userCred mcclient.TokenCredential,
  493. keys stringutils2.SSortedStrings,
  494. ) (*sqlchemy.SQuery, error) {
  495. var err error
  496. q, err = manager.SStandaloneResourceBaseManager.ListItemExportKeys(ctx, q, userCred, keys)
  497. if err != nil {
  498. return nil, errors.Wrap(err, "SStandaloneResourceBaseManager.ListItemExportKeys")
  499. }
  500. if keys.ContainsAny(manager.SZoneResourceBaseManager.GetExportKeys()...) {
  501. q, err = manager.SZoneResourceBaseManager.ListItemExportKeys(ctx, q, userCred, keys)
  502. if err != nil {
  503. return nil, errors.Wrap(err, "SZoneResourceBaseManager.ListItemExportKeys")
  504. }
  505. }
  506. if keys.Contains("wire") {
  507. q, err = manager.SWireResourceBaseManager.ListItemExportKeys(ctx, q, userCred, stringutils2.NewSortedStrings([]string{"wire"}))
  508. if err != nil {
  509. return nil, errors.Wrap(err, "SWireResourceBaseManager.ListItemExportKeys")
  510. }
  511. }
  512. return q, nil
  513. }
  514. func (cluster *SLoadbalancerCluster) PerformParamsPatch(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data *jsonutils.JSONDict) (*jsonutils.JSONDict, error) {
  515. oldParams := cluster.Params
  516. params := gotypes.DeepCopy(*cluster.Params).(SLoadbalancerClusterParams)
  517. d := jsonutils.NewDict()
  518. d.Set("params", data)
  519. paramsV := validators.NewStructValidator("params", &params)
  520. if err := paramsV.Validate(ctx, d); err != nil {
  521. return nil, err
  522. }
  523. // new vrrp virtual_router_id should be unique across clusters
  524. if params.VirtualRouterId != oldParams.VirtualRouterId {
  525. otherClusters, err := LoadbalancerClusterManager.findByVrrpRouterIdInZone(cluster.ZoneId, params.VirtualRouterId)
  526. if err != nil {
  527. return nil, errors.Wrap(err, "findByVrrpRouterIdInZone")
  528. }
  529. if len(otherClusters) > 0 {
  530. return nil, httperrors.NewConflictError("lbcluster %s(%s) already has virtual_router_id %d",
  531. otherClusters[0].Name, otherClusters[0].Id, params.VirtualRouterId)
  532. }
  533. }
  534. {
  535. // save name, description, params
  536. input := apis.StandaloneResourceBaseUpdateInput{}
  537. err := data.Unmarshal(&input)
  538. if err != nil {
  539. return nil, errors.Wrap(err, "Unmarshal update input")
  540. }
  541. input, err = cluster.SStandaloneResourceBase.ValidateUpdateData(ctx, userCred, query, input)
  542. if err != nil {
  543. return nil, errors.Wrap(err, "SStandaloneResourceBase.ValidateUpdateData")
  544. }
  545. diff, err := db.Update(cluster, func() error {
  546. cluster.Params = &params
  547. if len(input.Name) > 0 {
  548. cluster.Name = input.Name
  549. }
  550. if len(input.Description) > 0 {
  551. cluster.Description = input.Description
  552. }
  553. return nil
  554. })
  555. if err != nil {
  556. return nil, errors.Wrap(err, "Update")
  557. }
  558. db.OpsLog.LogEvent(cluster, db.ACT_UPDATE, diff, userCred)
  559. }
  560. {
  561. // save metadata
  562. cluster.TrySaveMetadataInput(ctx, userCred, data)
  563. }
  564. {
  565. // populate changes to underlying lbagents
  566. lbagents, err := LoadbalancerAgentManager.getByClusterId(cluster.Id)
  567. if err != nil {
  568. return nil, errors.Wrap(err, "getByClusterId")
  569. }
  570. for i := range lbagents {
  571. lbagent := lbagents[i]
  572. params := *lbagent.Params
  573. params.Vrrp.SLoadbalancerClusterParams = *cluster.Params
  574. _, err := db.Update(&lbagent, func() error {
  575. lbagent.Params = &params
  576. return nil
  577. })
  578. if err != nil {
  579. return nil, errors.Wrap(err, "Update")
  580. }
  581. }
  582. }
  583. return nil, nil
  584. }