server_skus.go 51 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564
  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. "database/sql"
  18. "fmt"
  19. "math"
  20. "sort"
  21. "strconv"
  22. "strings"
  23. "sync"
  24. "yunion.io/x/cloudmux/pkg/cloudprovider"
  25. "yunion.io/x/jsonutils"
  26. "yunion.io/x/log"
  27. "yunion.io/x/pkg/errors"
  28. "yunion.io/x/pkg/tristate"
  29. "yunion.io/x/pkg/util/compare"
  30. "yunion.io/x/pkg/utils"
  31. "yunion.io/x/sqlchemy"
  32. "yunion.io/x/onecloud/pkg/apis"
  33. api "yunion.io/x/onecloud/pkg/apis/compute"
  34. "yunion.io/x/onecloud/pkg/cloudcommon/db"
  35. "yunion.io/x/onecloud/pkg/cloudcommon/db/lockman"
  36. "yunion.io/x/onecloud/pkg/cloudcommon/db/taskman"
  37. "yunion.io/x/onecloud/pkg/cloudcommon/validators"
  38. "yunion.io/x/onecloud/pkg/compute/options"
  39. "yunion.io/x/onecloud/pkg/httperrors"
  40. "yunion.io/x/onecloud/pkg/mcclient"
  41. "yunion.io/x/onecloud/pkg/mcclient/auth"
  42. "yunion.io/x/onecloud/pkg/mcclient/modules/scheduler"
  43. "yunion.io/x/onecloud/pkg/util/stringutils2"
  44. "yunion.io/x/onecloud/pkg/util/yunionmeta"
  45. )
  46. // +onecloud:swagger-gen-model-singular=serversku
  47. // +onecloud:swagger-gen-model-plural=serverskus
  48. type SServerSkuManager struct {
  49. db.SEnabledStatusStandaloneResourceBaseManager
  50. SCloudregionResourceBaseManager
  51. SZoneResourceBaseManager
  52. }
  53. var ServerSkuManager *SServerSkuManager
  54. func init() {
  55. ServerSkuManager = &SServerSkuManager{
  56. SEnabledStatusStandaloneResourceBaseManager: db.NewEnabledStatusStandaloneResourceBaseManager(
  57. SServerSku{},
  58. "serverskus_tbl",
  59. "serversku",
  60. "serverskus",
  61. ),
  62. }
  63. ServerSkuManager.NameRequireAscii = false
  64. ServerSkuManager.SetVirtualObject(ServerSkuManager)
  65. // CREATE INDEX sku_index ON serverskus_tbl (`deleted`, `is_emulated`, `provider`, `cloudregion_id`, `postpaid_status`, `prepaid_status`)
  66. ServerSkuManager.TableSpec().AddIndex(false, "deleted", "is_emulated", "provider", "cloudregion_id", "postpaid_status", "prepaid_status")
  67. }
  68. // SServerSku 实际对应的是instance type清单. 这里的Sku实际指的是instance type。
  69. type SServerSku struct {
  70. db.SEnabledStatusStandaloneResourceBase
  71. db.SExternalizedResourceBase
  72. SCloudregionResourceBase
  73. SZoneResourceBase
  74. InstanceTypeFamily string `width:"32" charset:"ascii" nullable:"true" list:"user" create:"admin_optional" update:"admin"` // x2
  75. InstanceTypeCategory string `width:"32" charset:"utf8" nullable:"true" list:"user" create:"admin_optional" update:"admin"` // 通用型
  76. LocalCategory string `width:"32" charset:"utf8" nullable:"true" list:"user" create:"admin_optional" update:"admin" default:""` // 记录本地分类
  77. PrepaidStatus string `width:"32" charset:"utf8" nullable:"true" list:"user" update:"admin" create:"admin_optional" default:"available"` // 预付费资源状态 available|soldout
  78. PostpaidStatus string `width:"32" charset:"utf8" nullable:"true" list:"user" update:"admin" create:"admin_optional" default:"available"` // 按需付费资源状态 available|soldout
  79. CpuArch string `width:"16" charset:"ascii" nullable:"true" list:"user" create:"admin_optional" update:"admin"` // CPU 架构 x86|xarm
  80. CpuCoreCount int `nullable:"false" list:"user" create:"admin_required"`
  81. MemorySizeMB int `nullable:"false" list:"user" create:"admin_required"`
  82. OsName string `width:"32" charset:"ascii" nullable:"true" list:"user" create:"admin_optional" update:"admin" default:"Any"` // Windows|Linux|Any
  83. SysDiskResizable tristate.TriState `default:"true" list:"user" create:"admin_optional" update:"admin"`
  84. SysDiskType string `width:"128" charset:"ascii" nullable:"true" list:"user" create:"admin_optional" update:"admin"`
  85. SysDiskMinSizeGB int `nullable:"true" list:"user" create:"admin_optional" update:"admin"` // not required。 windows比较新的版本都是50G左右。
  86. SysDiskMaxSizeGB int `nullable:"true" list:"user" create:"admin_optional" update:"admin"` // not required
  87. AttachedDiskType string `width:"32" nullable:"true" list:"user" create:"admin_optional" update:"admin"`
  88. AttachedDiskSizeGB int `nullable:"true" list:"user" create:"admin_optional" update:"admin"`
  89. AttachedDiskCount int `nullable:"true" list:"user" create:"admin_optional" update:"admin"`
  90. DataDiskTypes string `width:"128" charset:"ascii" nullable:"true" list:"user" create:"admin_optional" update:"admin"`
  91. DataDiskMaxCount int `nullable:"true" list:"user" create:"admin_optional" update:"admin"`
  92. NicType string `width:"32" nullable:"true" list:"user" create:"admin_optional" update:"admin"`
  93. NicMaxCount int `default:"1" nullable:"true" list:"user" create:"admin_optional" update:"admin"`
  94. GpuAttachable tristate.TriState `default:"true" list:"user" create:"admin_optional" update:"admin"`
  95. GpuSpec string `width:"128" charset:"ascii" nullable:"true" list:"user" create:"admin_optional" update:"admin"`
  96. GpuCount string `width:"16" nullable:"true" list:"user" create:"admin_optional" update:"admin"`
  97. GpuMaxCount int `nullable:"true" list:"user" create:"admin_optional" update:"admin"`
  98. Provider string `width:"64" charset:"ascii" nullable:"true" list:"user" default:"OneCloud" create:"admin_optional"`
  99. Md5 string `width:"32" charset:"utf8" nullable:"true"`
  100. }
  101. func (manager *SServerSkuManager) FetchUniqValues(ctx context.Context, data jsonutils.JSONObject) jsonutils.JSONObject {
  102. regionId, _ := data.GetString("cloudregion_id")
  103. zoneId, _ := data.GetString("zone_id")
  104. return jsonutils.Marshal(map[string]string{"cloudregion_id": regionId, "zone_id": zoneId})
  105. }
  106. func (manager *SServerSkuManager) FilterByUniqValues(q *sqlchemy.SQuery, values jsonutils.JSONObject) *sqlchemy.SQuery {
  107. regionId, _ := values.GetString("cloudregion_id")
  108. if len(regionId) > 0 {
  109. q = q.Equals("cloudregion_id", regionId)
  110. }
  111. zoneId, _ := values.GetString("zone_id")
  112. if len(zoneId) > 0 {
  113. q = q.Equals("zone_id", zoneId)
  114. }
  115. return q
  116. }
  117. func sliceToJsonObject(items []int) jsonutils.JSONObject {
  118. sort.Slice(items, func(i, j int) bool {
  119. if items[i] < items[j] {
  120. return true
  121. }
  122. return false
  123. })
  124. ret := jsonutils.NewArray()
  125. for _, item := range items {
  126. ret.Add(jsonutils.NewInt(int64(item)))
  127. }
  128. return ret
  129. }
  130. func genInstanceType(family string, cpu, memMb int64) (string, error) {
  131. if cpu <= 0 {
  132. return "", fmt.Errorf("cpu_core_count should great than zero")
  133. }
  134. if memMb <= 0 {
  135. return "", fmt.Errorf("memory_size_mb should great than zero")
  136. }
  137. if memMb%1024 != 0 && memMb != 512 {
  138. return "", fmt.Errorf("memory_size_mb should be 512 or integral multiple of 1024")
  139. }
  140. switch memMb {
  141. case 512:
  142. return fmt.Sprintf("ecs.%s.c%dm1.nano", family, cpu), nil
  143. default:
  144. return fmt.Sprintf("ecs.%s.c%dm%d", family, cpu, memMb/1024), nil
  145. }
  146. }
  147. func (self SServerSku) GetGlobalId() string {
  148. return self.ExternalId
  149. }
  150. func (manager *SServerSkuManager) FetchCustomizeColumns(
  151. ctx context.Context,
  152. userCred mcclient.TokenCredential,
  153. query jsonutils.JSONObject,
  154. objs []interface{},
  155. fields stringutils2.SSortedStrings,
  156. isList bool,
  157. ) []api.ServerSkuDetails {
  158. rows := make([]api.ServerSkuDetails, len(objs))
  159. stdRows := manager.SEnabledStatusStandaloneResourceBaseManager.FetchCustomizeColumns(ctx, userCred, query, objs, fields, isList)
  160. regRows := manager.SCloudregionResourceBaseManager.FetchCustomizeColumns(ctx, userCred, query, objs, fields, isList)
  161. zoneRows := manager.SZoneResourceBaseManager.FetchCustomizeColumns(ctx, userCred, query, objs, fields, isList)
  162. instanceTypes := []string{}
  163. for i := range rows {
  164. rows[i] = api.ServerSkuDetails{
  165. EnabledStatusStandaloneResourceDetails: stdRows[i],
  166. ZoneResourceInfoBase: zoneRows[i].ZoneResourceInfoBase,
  167. CloudregionResourceInfo: regRows[i],
  168. }
  169. sku := objs[i].(*SServerSku)
  170. if !utils.IsInStringArray(sku.Name, instanceTypes) {
  171. instanceTypes = append(instanceTypes, sku.Name)
  172. }
  173. rows[i].CloudEnv = strings.Split(zoneRows[i].RegionExternalId, "/")[0]
  174. }
  175. ret := []struct {
  176. InstanceType string
  177. CloudregionId string
  178. ZoneId string
  179. }{}
  180. guestsQ := GuestManager.Query().SubQuery()
  181. hostsQ := HostManager.Query().SubQuery()
  182. zonesQ := ZoneManager.Query().SubQuery()
  183. q := guestsQ.Query(
  184. guestsQ.Field("instance_type"),
  185. hostsQ.Field("zone_id"),
  186. zonesQ.Field("cloudregion_id"),
  187. ).
  188. Join(hostsQ, sqlchemy.Equals(guestsQ.Field("host_id"), hostsQ.Field("id"))).
  189. Join(zonesQ, sqlchemy.Equals(hostsQ.Field("zone_id"), zonesQ.Field("id"))).
  190. Filter(sqlchemy.In(guestsQ.Field("instance_type"), instanceTypes))
  191. err := q.All(&ret)
  192. if err != nil {
  193. log.Errorf("query instance cnt error: %v", err)
  194. return rows
  195. }
  196. skuMap := map[string]map[string]int{}
  197. for _, sku := range ret {
  198. _, ok := skuMap[sku.InstanceType]
  199. if !ok {
  200. skuMap[sku.InstanceType] = map[string]int{}
  201. }
  202. _, ok = skuMap[sku.InstanceType][sku.ZoneId]
  203. if !ok {
  204. skuMap[sku.InstanceType][sku.ZoneId] = 0
  205. }
  206. skuMap[sku.InstanceType][sku.ZoneId] += 1
  207. _, ok = skuMap[sku.InstanceType][sku.CloudregionId]
  208. if !ok {
  209. skuMap[sku.InstanceType][sku.CloudregionId] = 0
  210. }
  211. skuMap[sku.InstanceType][sku.CloudregionId] += 1
  212. }
  213. for i := range rows {
  214. sku := objs[i].(*SServerSku)
  215. if len(sku.ZoneId) > 0 {
  216. rows[i].TotalGuestCount = skuMap[sku.Name][sku.ZoneId]
  217. } else {
  218. rows[i].TotalGuestCount = skuMap[sku.Name][sku.CloudregionId]
  219. }
  220. }
  221. return rows
  222. }
  223. func (self *SServerSkuManager) ValidateCreateData(ctx context.Context, userCred mcclient.TokenCredential, ownerId mcclient.IIdentityProvider, query jsonutils.JSONObject, input api.ServerSkuCreateInput) (api.ServerSkuCreateInput, error) {
  224. var region *SCloudregion
  225. if len(input.CloudregionId) > 0 {
  226. _region, err := validators.ValidateModel(ctx, userCred, CloudregionManager, &input.CloudregionId)
  227. if err != nil {
  228. return input, err
  229. }
  230. region = _region.(*SCloudregion)
  231. }
  232. if len(input.ZoneId) > 0 {
  233. _zone, err := validators.ValidateModel(ctx, userCred, ZoneManager, &input.ZoneId)
  234. if err != nil {
  235. return input, err
  236. }
  237. zone := _zone.(*SZone)
  238. if len(input.CloudregionId) == 0 {
  239. input.CloudregionId = zone.CloudregionId
  240. }
  241. if input.CloudregionId != zone.CloudregionId {
  242. return input, httperrors.NewConflictError("zone %s not in cloudregion %s", zone.Name, input.CloudregionId)
  243. }
  244. region, _ = zone.GetRegion()
  245. }
  246. if input.Enabled == nil {
  247. enabled := true
  248. input.Enabled = &enabled
  249. }
  250. input.Provider = api.CLOUD_PROVIDER_ONECLOUD
  251. input.Status = api.SkuStatusReady
  252. if region != nil {
  253. input.Provider = region.Provider
  254. }
  255. if input.Provider == api.CLOUD_PROVIDER_ONECLOUD {
  256. input.CloudregionId = api.DEFAULT_REGION_ID
  257. } else if utils.IsInStringArray(input.Provider, api.PRIVATE_CLOUD_PROVIDERS) {
  258. input.Status = api.SkuStatusCreating
  259. }
  260. if !utils.IsInStringArray(input.Provider, api.PUBLIC_CLOUD_PROVIDERS) {
  261. if input.CpuCoreCount < 1 || input.CpuCoreCount > options.Options.SkuMaxCpuCount {
  262. return input, httperrors.NewOutOfRangeError("cpu_core_count should be range of 1~%d", options.Options.SkuMaxCpuCount)
  263. }
  264. if input.MemorySizeMB < 512 || input.MemorySizeMB > 1024*options.Options.SkuMaxMemSize {
  265. return input, httperrors.NewOutOfRangeError("memory_size_mb, shoud be range of 512~%d", 1024*options.Options.SkuMaxMemSize)
  266. }
  267. if len(input.InstanceTypeCategory) == 0 {
  268. input.InstanceTypeCategory = api.SkuCategoryGeneralPurpose
  269. }
  270. if !utils.IsInStringArray(input.InstanceTypeCategory, api.SKU_FAMILIES) {
  271. return input, httperrors.NewInputParameterError("instance_type_category shoud be one of %s", api.SKU_FAMILIES)
  272. }
  273. input.LocalCategory = input.InstanceTypeCategory
  274. input.InstanceTypeFamily = api.InstanceFamilies[input.InstanceTypeCategory]
  275. }
  276. if len(input.LocalCategory) == 0 {
  277. input.LocalCategory = input.InstanceTypeCategory
  278. }
  279. if len(input.InstanceTypeFamily) == 0 {
  280. input.InstanceTypeFamily = api.InstanceFamilies[input.InstanceTypeCategory]
  281. }
  282. var err error
  283. if len(input.Name) == 0 {
  284. // 格式 ecs.g1.c1m1
  285. input.Name, err = genInstanceType(input.InstanceTypeFamily, input.CpuCoreCount, input.MemorySizeMB)
  286. if err != nil {
  287. return input, httperrors.NewInputParameterError("%v", err)
  288. }
  289. q := self.Query().Equals("name", input.Name)
  290. if len(input.CloudregionId) > 0 {
  291. q = q.Equals("cloudregion_id", input.CloudregionId)
  292. }
  293. count, err := q.CountWithError()
  294. if err != nil {
  295. return input, httperrors.NewInternalServerError("checkout server sku name duplicate error: %v", err)
  296. }
  297. if count > 0 {
  298. return input, httperrors.NewDuplicateResourceError("Duplicate sku %s", input.Name)
  299. }
  300. }
  301. input.EnabledStatusStandaloneResourceCreateInput, err = self.SEnabledStatusStandaloneResourceBaseManager.ValidateCreateData(ctx, userCred, ownerId, query, input.EnabledStatusStandaloneResourceCreateInput)
  302. if err != nil {
  303. return input, err
  304. }
  305. return input, nil
  306. }
  307. func (self *SServerSku) PostCreate(ctx context.Context, userCred mcclient.TokenCredential, ownerId mcclient.IIdentityProvider, query jsonutils.JSONObject, data jsonutils.JSONObject) {
  308. self.SEnabledStatusStandaloneResourceBase.PostCreate(ctx, userCred, ownerId, query, data)
  309. if utils.IsInStringArray(self.Provider, api.PRIVATE_CLOUD_PROVIDERS) {
  310. self.StartSkuCreateTask(ctx, userCred)
  311. }
  312. }
  313. func (self *SServerSku) StartSkuCreateTask(ctx context.Context, userCred mcclient.TokenCredential) error {
  314. task, err := taskman.TaskManager.NewTask(ctx, "ServerSkuCreateTask", self, userCred, nil, "", "", nil)
  315. if err != nil {
  316. return errors.Wrapf(err, "NewTask")
  317. }
  318. return task.ScheduleRun(nil)
  319. }
  320. func (self *SServerSku) GetPrivateCloudproviders() ([]SCloudprovider, error) {
  321. providers := []SCloudprovider{}
  322. q := CloudproviderManager.Query().In("provider", CloudproviderManager.GetPrivateProviderProvidersQuery())
  323. err := db.FetchModelObjects(CloudproviderManager, q, &providers)
  324. if err != nil {
  325. return nil, err
  326. }
  327. return providers, nil
  328. }
  329. func (self *SServerSkuManager) ClearSchedDescCache(wait bool) error {
  330. s := auth.GetAdminSession(context.Background(), options.Options.Region)
  331. _, err := scheduler.SchedManager.SyncSku(s, true)
  332. if err != nil {
  333. return errors.Wrapf(err, "chedManager.SyncSku")
  334. }
  335. return nil
  336. }
  337. func (self *SServerSkuManager) FetchByZoneExtId(zoneExtId string, name string) (db.IModel, error) {
  338. zoneObj, err := db.FetchByExternalId(ZoneManager, zoneExtId)
  339. if err != nil {
  340. return nil, err
  341. }
  342. return self.FetchByZoneId(zoneObj.GetId(), name)
  343. }
  344. func (self *SServerSkuManager) FetchByZoneId(zoneId string, name string) (db.IModel, error) {
  345. q := self.Query().Equals("zone_id", zoneId).Equals("name", name)
  346. count, err := q.CountWithError()
  347. if err != nil {
  348. return nil, err
  349. }
  350. if count == 1 {
  351. obj, err := db.NewModelObject(self)
  352. if err != nil {
  353. return nil, err
  354. }
  355. err = q.First(obj)
  356. if err != nil {
  357. return nil, err
  358. } else {
  359. return obj.(db.IStandaloneModel), nil
  360. }
  361. } else if count > 1 {
  362. return nil, sqlchemy.ErrDuplicateEntry
  363. } else {
  364. return nil, sql.ErrNoRows
  365. }
  366. }
  367. // 四舍五入
  368. func round(n int, step int) int {
  369. q := float64(n) / float64(step)
  370. return int(math.Floor(q+0.5)) * step
  371. }
  372. // 内存按GB取整
  373. func roundMem(n int) int {
  374. if n <= 512 {
  375. return 512
  376. }
  377. return round(n, 1024)
  378. }
  379. // step必须是偶数
  380. func interval(n int, step int) (int, int) {
  381. r := round(n, step)
  382. start := r - step/2
  383. end := r + step/2
  384. return start, end
  385. }
  386. // 计算内存所在区间范围
  387. func intervalMem(n int) (int, int) {
  388. if n <= 512 {
  389. return 0, 512
  390. }
  391. return interval(n, 1024)
  392. }
  393. func networkUsableRegionQueries(f sqlchemy.IQueryField) []sqlchemy.ICondition {
  394. providers := usableCloudProviders()
  395. networks := NetworkManager.Query("wire_id").Equals("status", api.NETWORK_STATUS_AVAILABLE)
  396. wires := WireManager.Query("vpc_id").In("id", networks)
  397. _vpcs := VpcManager.Query("cloudregion_id").
  398. Equals("status", api.VPC_STATUS_AVAILABLE).
  399. In("id", wires)
  400. filters := sqlchemy.OR(sqlchemy.In(_vpcs.Field("manager_id"), providers), sqlchemy.IsNullOrEmpty(_vpcs.Field("manager_id")))
  401. vpcs := _vpcs.Filter(filters).SubQuery()
  402. sq := vpcs.Query(sqlchemy.DISTINCT("cloudregion_id", vpcs.Field("cloudregion_id")))
  403. return []sqlchemy.ICondition{sqlchemy.In(f, sq.SubQuery())}
  404. }
  405. func usableFilter(q *sqlchemy.SQuery, public_cloud bool) (*sqlchemy.SQuery, error) {
  406. // 过滤出公有云provider状态健康的sku
  407. if public_cloud {
  408. providerTable := usableCloudProviders().SubQuery()
  409. providerRegionTable := CloudproviderRegionManager.Query().SubQuery()
  410. _subq := providerRegionTable.Query(sqlchemy.DISTINCT("cloudregion_id", providerRegionTable.Field("cloudregion_id")))
  411. subq := _subq.Join(providerTable, sqlchemy.Equals(providerRegionTable.Field("cloudprovider_id"), providerTable.Field("id"))).SubQuery()
  412. q.Join(subq, sqlchemy.Equals(q.Field("cloudregion_id"), subq.Field("cloudregion_id")))
  413. }
  414. // 过滤出network usable的sku
  415. if public_cloud {
  416. zoneIds, err := NetworkUsableZoneIds(true, true, nil)
  417. if err != nil {
  418. return nil, errors.Wrap(err, "NetworkUsableZoneIds")
  419. }
  420. q = q.Filter(sqlchemy.OR(sqlchemy.In(q.Field("zone_id"), zoneIds), sqlchemy.IsNullOrEmpty(q.Field("zone_id")))) //Azure的zone_id可能为空
  421. } else {
  422. // 本地IDC sku 只定义到region层级, zone id 为空.因此只能按region查询
  423. iconditions := networkUsableRegionQueries(q.Field("cloudregion_id"))
  424. // 私有云 sku region及zone为空
  425. iconditions = append(iconditions, sqlchemy.IsNullOrEmpty(q.Field("cloudregion_id")))
  426. q = q.Filter(sqlchemy.OR(iconditions...))
  427. }
  428. return q, nil
  429. }
  430. func (manager *SServerSkuManager) GetPropertyInstanceSpecs(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject) (jsonutils.JSONObject, error) {
  431. listQuery := api.ServerSkuListInput{}
  432. err := query.Unmarshal(&listQuery)
  433. if err != nil {
  434. return nil, errors.Wrap(err, "query.Unmarshal")
  435. }
  436. q, err := manager.ListItemFilter(ctx, manager.Query(), userCred, listQuery)
  437. if err != nil {
  438. return nil, errors.Wrap(err, "manager.ListItemFilter")
  439. }
  440. skus := make([]SServerSku, 0)
  441. q = q.GroupBy(q.Field("cpu_core_count"), q.Field("memory_size_mb"))
  442. q = q.Asc(q.Field("cpu_core_count"), q.Field("memory_size_mb"))
  443. err = db.FetchModelObjects(manager, q, &skus)
  444. if err != nil {
  445. log.Errorf("FetchModelObjects %s: %s", q.DebugString(), err)
  446. return nil, httperrors.NewBadRequestError("instance specs list query error")
  447. }
  448. cpus := jsonutils.NewArray()
  449. mems_mb := []int{}
  450. cpu_mems_mb := map[string][]int{}
  451. mems := map[int]bool{}
  452. oc := 0
  453. for i := range skus {
  454. nc := skus[i].CpuCoreCount
  455. nm := roundMem(skus[i].MemorySizeMB) // 内存按GB取整
  456. if nc > oc {
  457. cpus.Add(jsonutils.NewInt(int64(nc)))
  458. oc = nc
  459. }
  460. if _, exists := mems[nm]; !exists {
  461. mems_mb = append(mems_mb, nm)
  462. mems[nm] = true
  463. }
  464. k := strconv.Itoa(nc)
  465. if _, exists := cpu_mems_mb[k]; !exists {
  466. cpu_mems_mb[k] = []int{nm}
  467. } else {
  468. idx := len(cpu_mems_mb[k]) - 1
  469. if cpu_mems_mb[k][idx] != nm {
  470. cpu_mems_mb[k] = append(cpu_mems_mb[k], nm)
  471. }
  472. }
  473. }
  474. ret := jsonutils.NewDict()
  475. ret.Add(cpus, "cpus")
  476. ret.Add(sliceToJsonObject(mems_mb), "mems_mb")
  477. r_obj := jsonutils.Marshal(&cpu_mems_mb)
  478. ret.Add(r_obj, "cpu_mems_mb")
  479. return ret, nil
  480. }
  481. func (self *SServerSku) ValidateUpdateData(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, input api.ServerSkuUpdateInput) (api.ServerSkuUpdateInput, error) {
  482. if len(input.Name) > 0 {
  483. return input, httperrors.NewUnsupportOperationError("Cannot change server sku name")
  484. }
  485. var err error
  486. input.EnabledStatusStandaloneResourceBaseUpdateInput, err = self.SEnabledStatusStandaloneResourceBase.ValidateUpdateData(ctx, userCred, query, input.EnabledStatusStandaloneResourceBaseUpdateInput)
  487. if err != nil {
  488. return input, errors.Wrap(err, "SEnabledStatusStandaloneResourceBase.ValidateUpdateData")
  489. }
  490. return input, nil
  491. }
  492. func (self *SServerSku) PostUpdate(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) {
  493. ServerSkuManager.ClearSchedDescCache(true)
  494. if utils.IsInStringArray(self.Provider, api.PUBLIC_CLOUD_PROVIDERS) && (data.Contains("prepaid_status") || data.Contains("postpaid_status")) {
  495. self.SetMetadata(ctx, api.SERVER_SKU_PROJECT_SRC_KEY, api.SERVER_SKU_PROJECT_SRC_VALUE_LOCAL, userCred)
  496. }
  497. self.SEnabledStatusStandaloneResourceBase.PostUpdate(ctx, userCred, query, data)
  498. }
  499. func (self *SServerSku) Delete(ctx context.Context, userCred mcclient.TokenCredential) error {
  500. log.Infof("SServerSku delete do nothing")
  501. return nil
  502. }
  503. func (self *SServerSku) RealDelete(ctx context.Context, userCred mcclient.TokenCredential) error {
  504. ServerSkuManager.ClearSchedDescCache(true)
  505. return db.RealDeleteModel(ctx, userCred, self)
  506. }
  507. func (self *SServerSku) CustomizeDelete(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) error {
  508. return self.StartServerSkuDeleteTask(ctx, userCred, jsonutils.QueryBoolean(data, "purge", false), "")
  509. }
  510. func (self *SServerSku) StartServerSkuDeleteTask(ctx context.Context, userCred mcclient.TokenCredential, purge bool, parentTaskId string) error {
  511. params := jsonutils.NewDict()
  512. params.Add(jsonutils.NewBool(purge), "purge")
  513. task, err := taskman.TaskManager.NewTask(ctx, "ServerSkuDeleteTask", self, userCred, params, parentTaskId, "", nil)
  514. if err != nil {
  515. log.Errorf("newTask ServerSkuDeleteTask fail %s", err)
  516. return err
  517. }
  518. self.SetStatus(ctx, userCred, api.SkuStatusDeleting, "start to delete")
  519. task.ScheduleRun(nil)
  520. return nil
  521. }
  522. func (self *SServerSku) GetGuestCount() (int, error) {
  523. guestsQ := GuestManager.Query().SubQuery()
  524. hostsQ := HostManager.Query().SubQuery()
  525. zonesQ := ZoneManager.Query().SubQuery()
  526. q := guestsQ.Query().
  527. Join(hostsQ, sqlchemy.Equals(guestsQ.Field("host_id"), hostsQ.Field("id"))).
  528. Join(zonesQ, sqlchemy.Equals(hostsQ.Field("zone_id"), zonesQ.Field("id"))).
  529. Filter(sqlchemy.Equals(guestsQ.Field("instance_type"), self.Name))
  530. if len(self.ZoneId) > 0 {
  531. q = q.Filter(sqlchemy.Equals(hostsQ.Field("zone_id"), self.ZoneId))
  532. } else {
  533. q = q.Filter(sqlchemy.Equals(zonesQ.Field("cloudregion_id"), self.CloudregionId))
  534. }
  535. return q.CountWithError()
  536. }
  537. func (self *SServerSku) ValidateDeleteCondition(ctx context.Context, info *api.ServerSkuDetails) error {
  538. totalGuestCnt := 0
  539. if info != nil {
  540. totalGuestCnt = info.TotalGuestCount
  541. } else {
  542. totalGuestCnt, _ = self.GetGuestCount()
  543. }
  544. if totalGuestCnt > 0 {
  545. return httperrors.NewNotEmptyError("now allow to delete inuse instance_type.please remove related servers first: %s", self.Name)
  546. }
  547. if !options.Options.EnableDeletePublicCloudSku && utils.IsInStringArray(self.Provider, api.PUBLIC_CLOUD_PROVIDERS) {
  548. return httperrors.NewForbiddenError("not allow to delete public cloud instance_type: %s", self.Name)
  549. }
  550. return nil
  551. }
  552. func (self *SServerSku) GetZoneExternalId() (string, error) {
  553. zoneObj, err := ZoneManager.FetchById(self.ZoneId)
  554. if err != nil {
  555. return "", err
  556. }
  557. zone := zoneObj.(*SZone)
  558. return zone.GetExternalId(), nil
  559. }
  560. func listItemDomainFilter(q *sqlchemy.SQuery, providers []string, domainId string) *sqlchemy.SQuery {
  561. // CLOUD_PROVIDER_ONECLOUD 没有对应的cloudaccount
  562. if len(domainId) > 0 {
  563. if len(providers) >= 1 && !utils.IsInStringArray(api.CLOUD_PROVIDER_ONECLOUD, providers) {
  564. // 明确指定只查询公有云provider的情况,只查询公有云skus
  565. q = q.In("provider", getDomainManagerProviderSubq(domainId))
  566. } else if len(providers) == 1 && utils.IsInStringArray(api.CLOUD_PROVIDER_ONECLOUD, providers) {
  567. // 明确指定只查询私有云provider的情况
  568. } else {
  569. // 公有云skus & 私有云skus 混合查询
  570. publicSkusQ := sqlchemy.In(q.Field("provider"), getDomainManagerProviderSubq(domainId))
  571. privateSkusQ := sqlchemy.Equals(q.Field("provider"), api.CLOUD_PROVIDER_ONECLOUD)
  572. q = q.Filter(sqlchemy.OR(publicSkusQ, privateSkusQ))
  573. }
  574. }
  575. return q
  576. }
  577. // 主机套餐规格列表
  578. func (manager *SServerSkuManager) ListItemFilter(
  579. ctx context.Context,
  580. q *sqlchemy.SQuery,
  581. userCred mcclient.TokenCredential,
  582. query api.ServerSkuListInput,
  583. ) (*sqlchemy.SQuery, error) {
  584. publicCloud := false
  585. cloudEnvStr := query.CloudEnv
  586. if cloudEnvStr == api.CLOUD_ENV_PUBLIC_CLOUD {
  587. publicCloud = true
  588. pq := CloudproviderManager.GetPublicProviderProvidersQuery()
  589. q = q.Join(pq, sqlchemy.Equals(q.Field("provider"), pq.Field("provider")))
  590. }
  591. if cloudEnvStr == api.CLOUD_ENV_PRIVATE_CLOUD {
  592. pq := CloudproviderManager.GetPrivateProviderProvidersQuery()
  593. q = q.Join(pq, sqlchemy.Equals(q.Field("provider"), pq.Field("provider")))
  594. }
  595. if cloudEnvStr == api.CLOUD_ENV_ON_PREMISE {
  596. q = q.Filter(
  597. sqlchemy.OR(
  598. sqlchemy.Equals(q.Field("provider"), api.CLOUD_PROVIDER_ONECLOUD),
  599. sqlchemy.In(q.Field("provider"), CloudproviderManager.GetOnPremiseProviderProvidersQuery()),
  600. ),
  601. )
  602. }
  603. if cloudEnvStr == api.CLOUD_ENV_PRIVATE_ON_PREMISE {
  604. q = q.Filter(sqlchemy.OR(
  605. sqlchemy.In(q.Field("provider"), CloudproviderManager.GetPrivateProviderProvidersQuery()), //私有云
  606. sqlchemy.Equals(q.Field("provider"), api.CLOUD_PROVIDER_ONECLOUD), //本地IDC
  607. ),
  608. )
  609. }
  610. if domainStr := query.ProjectDomainId; len(domainStr) > 0 {
  611. domain, err := db.TenantCacheManager.FetchDomainByIdOrName(ctx, domainStr)
  612. if err != nil {
  613. if errors.Cause(err) == sql.ErrNoRows {
  614. return nil, httperrors.NewResourceNotFoundError2("domains", domainStr)
  615. }
  616. return nil, httperrors.NewGeneralError(err)
  617. }
  618. query.ProjectDomainId = domain.GetId()
  619. }
  620. q = listItemDomainFilter(q, query.Providers, query.ProjectDomainId)
  621. providers := query.Providers
  622. if len(providers) > 0 {
  623. q = q.Filter(sqlchemy.In(q.Field("provider"), providers))
  624. if len(providers) == 1 && utils.IsInStringArray(providers[0], cloudprovider.GetPublicProviders()) {
  625. publicCloud = true
  626. }
  627. }
  628. conditions := []sqlchemy.ICondition{}
  629. for _, arch := range query.CpuArch {
  630. if len(arch) == 0 {
  631. continue
  632. }
  633. if arch == apis.OS_ARCH_X86 {
  634. conditions = append(conditions, sqlchemy.OR(
  635. sqlchemy.Startswith(q.Field("cpu_arch"), arch),
  636. sqlchemy.Equals(q.Field("cpu_arch"), apis.OS_ARCH_I386),
  637. sqlchemy.IsNullOrEmpty(q.Field("cpu_arch")),
  638. ))
  639. } else if arch == apis.OS_ARCH_ARM {
  640. conditions = append(conditions, sqlchemy.OR(
  641. sqlchemy.Startswith(q.Field("cpu_arch"), arch),
  642. sqlchemy.Equals(q.Field("cpu_arch"), apis.OS_ARCH_AARCH32),
  643. sqlchemy.Equals(q.Field("cpu_arch"), apis.OS_ARCH_AARCH64),
  644. sqlchemy.IsNullOrEmpty(q.Field("cpu_arch")),
  645. ))
  646. } else if arch == apis.OS_ARCH_RISCV {
  647. conditions = append(conditions, sqlchemy.OR(
  648. sqlchemy.Startswith(q.Field("cpu_arch"), arch),
  649. sqlchemy.Equals(q.Field("cpu_arch"), apis.OS_ARCH_RISCV32),
  650. sqlchemy.Equals(q.Field("cpu_arch"), apis.OS_ARCH_RISCV64),
  651. sqlchemy.IsNullOrEmpty(q.Field("cpu_arch")),
  652. ))
  653. } else {
  654. conditions = append(conditions, sqlchemy.Startswith(q.Field("cpu_arch"), arch))
  655. }
  656. }
  657. if len(conditions) > 0 {
  658. q = q.Filter(sqlchemy.OR(conditions...))
  659. }
  660. if query.Distinct {
  661. q = q.GroupBy(q.Field("name"))
  662. }
  663. brands := query.Brands
  664. if len(brands) > 0 {
  665. q = q.Filter(sqlchemy.In(q.Field("brand"), brands))
  666. }
  667. q, err := manager.SEnabledStatusStandaloneResourceBaseManager.ListItemFilter(ctx, q, userCred, query.EnabledStatusStandaloneResourceListInput)
  668. if err != nil {
  669. return nil, errors.Wrap(err, "SEnabledStatusStandaloneResourceBaseManager.ListItemFilter")
  670. }
  671. if query.Usable != nil && *query.Usable {
  672. q, err := usableFilter(q, publicCloud)
  673. if err != nil {
  674. return nil, err
  675. }
  676. q = q.IsTrue("enabled")
  677. }
  678. zoneStr := query.ZoneId
  679. if len(zoneStr) > 0 {
  680. zoneObj, err := validators.ValidateModel(ctx, userCred, ZoneManager, &zoneStr)
  681. if err != nil {
  682. return nil, err
  683. }
  684. zone := zoneObj.(*SZone)
  685. region, err := zone.GetRegion()
  686. if err != nil {
  687. return nil, errors.Wrapf(err, "GetRegion %s", zone.Name)
  688. }
  689. q = q.Filter(
  690. sqlchemy.OR(
  691. sqlchemy.AND(
  692. sqlchemy.Equals(q.Field("cloudregion_id"), region.Id),
  693. sqlchemy.IsNullOrEmpty(q.Field("zone_id")),
  694. ),
  695. sqlchemy.Equals(q.Field("zone_id"), zone.Id),
  696. ),
  697. )
  698. }
  699. q, err = managedResourceFilterByRegion(ctx, q, query.RegionalFilterListInput, "", nil)
  700. if err != nil {
  701. return nil, errors.Wrap(err, "managedResourceFilterByRegion")
  702. }
  703. if len(query.PostpaidStatus) > 0 {
  704. q = q.Equals("postpaid_status", query.PostpaidStatus)
  705. }
  706. if len(query.PrepaidStatus) > 0 {
  707. q = q.Equals("prepaid_status", query.PrepaidStatus)
  708. }
  709. conditions = []sqlchemy.ICondition{}
  710. for _, sizeMb := range query.MemorySizeMb {
  711. // 按区间查询内存, 避免0.75G这样的套餐不好过滤
  712. if sizeMb > 0 {
  713. s, e := intervalMem(sizeMb)
  714. conditions = append(
  715. conditions,
  716. sqlchemy.AND(
  717. sqlchemy.GT(q.Field("memory_size_mb"), s),
  718. sqlchemy.LE(q.Field("memory_size_mb"), e),
  719. ),
  720. )
  721. }
  722. }
  723. if len(conditions) > 0 {
  724. q = q.Filter(sqlchemy.OR(conditions...))
  725. }
  726. if len(query.CpuCoreCount) > 0 {
  727. q = q.In("cpu_core_count", query.CpuCoreCount)
  728. }
  729. return q, err
  730. }
  731. func (manager *SServerSkuManager) OrderByExtraFields(
  732. ctx context.Context,
  733. q *sqlchemy.SQuery,
  734. userCred mcclient.TokenCredential,
  735. query api.ServerSkuListInput,
  736. ) (*sqlchemy.SQuery, error) {
  737. var err error
  738. q, err = manager.SEnabledStatusStandaloneResourceBaseManager.OrderByExtraFields(ctx, q, userCred, query.EnabledStatusStandaloneResourceListInput)
  739. if err != nil {
  740. return nil, errors.Wrap(err, "SEnabledStatusStandaloneResourceBaseManager.OrderByExtraFields")
  741. }
  742. q, err = manager.SCloudregionResourceBaseManager.OrderByExtraFields(ctx, q, userCred, query.RegionalFilterListInput)
  743. if err != nil {
  744. return nil, errors.Wrap(err, "SCloudregionResourceBaseManager.OrderByExtraFields")
  745. }
  746. if db.NeedOrderQuery([]string{query.OrderByTotalGuestCount}) {
  747. guestQ := GuestManager.Query()
  748. guestQ = guestQ.AppendField(guestQ.Field("instance_type"), sqlchemy.COUNT("total_guest_count"))
  749. guestQ = guestQ.GroupBy(guestQ.Field("instance_type"))
  750. guestSQ := guestQ.SubQuery()
  751. q = q.Join(guestSQ, sqlchemy.Equals(guestSQ.Field("instance_type"), q.Field("name")))
  752. q = q.AppendField(q.QueryFields()...)
  753. q = q.AppendField(guestSQ.Field("total_guest_count"))
  754. q = db.OrderByFields(q, []string{query.OrderByTotalGuestCount}, []sqlchemy.IQueryField{q.Field("total_guest_count")})
  755. }
  756. return q, nil
  757. }
  758. func (manager *SServerSkuManager) QueryDistinctExtraField(q *sqlchemy.SQuery, field string) (*sqlchemy.SQuery, error) {
  759. var err error
  760. q, err = manager.SEnabledStatusStandaloneResourceBaseManager.QueryDistinctExtraField(q, field)
  761. if err == nil {
  762. return q, nil
  763. }
  764. q, err = manager.SCloudregionResourceBaseManager.QueryDistinctExtraField(q, field)
  765. if err == nil {
  766. return q, nil
  767. }
  768. return q, httperrors.ErrNotFound
  769. }
  770. func (manager *SServerSkuManager) GetMatchedSku(regionId string, cpu int64, memMB int64) (*SServerSku, error) {
  771. ret := &SServerSku{}
  772. _region, err := CloudregionManager.FetchById(regionId)
  773. if err != nil {
  774. return nil, errors.Wrapf(err, "CloudregionManager.FetchById(%s)", regionId)
  775. }
  776. region := _region.(*SCloudregion)
  777. if utils.IsInStringArray(region.Provider, api.PRIVATE_CLOUD_PROVIDERS) {
  778. regionId = api.DEFAULT_REGION_ID
  779. }
  780. q := manager.Query()
  781. q = q.Equals("cpu_core_count", cpu).Equals("memory_size_mb", memMB).Equals("cloudregion_id", regionId).Equals("postpaid_status", api.SkuStatusAvailable)
  782. err = q.First(ret)
  783. if err != nil {
  784. return nil, errors.Wrap(err, "ServerSkuManager.GetMatchedSku")
  785. }
  786. return ret, nil
  787. }
  788. func (manager *SServerSkuManager) FetchSkuByNameAndProvider(name string, provider string, checkConsistency bool) (*SServerSku, error) {
  789. q := manager.Query().IsTrue("enabled")
  790. q = q.Equals("name", name)
  791. if utils.IsInStringArray(provider, []string{api.CLOUD_PROVIDER_ONECLOUD, api.CLOUD_PROVIDER_VMWARE, api.CLOUD_PROVIDER_NUTANIX}) {
  792. q = q.Filter(
  793. sqlchemy.Equals(q.Field("provider"), api.CLOUD_PROVIDER_ONECLOUD),
  794. )
  795. } else if utils.IsInStringArray(provider, api.PRIVATE_CLOUD_PROVIDERS) {
  796. q = q.Filter(sqlchemy.OR(
  797. sqlchemy.Equals(q.Field("provider"), api.CLOUD_PROVIDER_ONECLOUD),
  798. sqlchemy.Equals(q.Field("provider"), provider),
  799. ))
  800. } else {
  801. q = q.Equals("provider", provider)
  802. }
  803. skus := make([]SServerSku, 0)
  804. err := db.FetchModelObjects(manager, q, &skus)
  805. if err != nil {
  806. log.Errorf("fetch sku fail %s", err)
  807. return nil, err
  808. }
  809. if len(skus) == 0 {
  810. log.Errorf("no sku found for %s %s", name, provider)
  811. return nil, httperrors.NewResourceNotFoundError2(manager.Keyword(), name)
  812. }
  813. if len(skus) == 1 {
  814. return &skus[0], nil
  815. }
  816. if checkConsistency {
  817. for i := 1; i < len(skus); i += 1 {
  818. if skus[i].CpuCoreCount != skus[0].CpuCoreCount || skus[i].MemorySizeMB != skus[0].MemorySizeMB {
  819. log.Errorf("inconsistent sku %s %s", jsonutils.Marshal(&skus[0]), jsonutils.Marshal(&skus[i]))
  820. return nil, httperrors.NewDuplicateResourceError("duplicate instanceType %s", name)
  821. }
  822. }
  823. }
  824. return &skus[0], nil
  825. }
  826. func (manager *SServerSkuManager) GetPublicCloudSkuCount() (int, error) {
  827. q := manager.Query()
  828. q = q.Filter(sqlchemy.In(q.Field("provider"), cloudprovider.GetPublicProviders()))
  829. return q.CountWithError()
  830. }
  831. func (manager *SServerSkuManager) GetSkuCountByRegion(regionId string) (int, error) {
  832. q := manager.Query()
  833. if len(regionId) == 0 {
  834. q = q.IsNotEmpty("cloudregion_id")
  835. } else {
  836. q = q.Equals("cloudregion_id", regionId)
  837. }
  838. return q.CountWithError()
  839. }
  840. func (manager *SServerSkuManager) GetSkuCountByZone(zoneId string) []SServerSku {
  841. skus := []SServerSku{}
  842. q := manager.Query().Equals("zone_id", zoneId)
  843. if err := db.FetchModelObjects(manager, q, &skus); err != nil {
  844. log.Errorf("failed to get skus by zoneId %s error: %v", zoneId, err)
  845. }
  846. return skus
  847. }
  848. func (manager *SServerSkuManager) GetOneCloudSkus() ([]string, error) {
  849. skus := []SServerSku{}
  850. q := manager.Query().Equals("provider", api.CLOUD_PROVIDER_ONECLOUD)
  851. err := db.FetchModelObjects(manager, q, &skus)
  852. if err != nil {
  853. return nil, err
  854. }
  855. result := []string{}
  856. for _, sku := range skus {
  857. result = append(result, fmt.Sprintf("%d/%d", sku.CpuCoreCount, sku.MemorySizeMB))
  858. }
  859. return result, nil
  860. }
  861. func (manager *SServerSkuManager) GetSkus(provider string, cpu, memMB int) ([]SServerSku, error) {
  862. skus := []SServerSku{}
  863. q := manager.Query()
  864. if provider == api.CLOUD_PROVIDER_ONECLOUD {
  865. providerFilter := sqlchemy.OR(sqlchemy.Equals(q.Field("provider"), provider), sqlchemy.IsNullOrEmpty(q.Field("provider")))
  866. q = q.Equals("cpu_core_count", cpu).Equals("memory_size_mb", memMB).Filter(providerFilter)
  867. } else {
  868. q = q.Equals("cpu_core_count", cpu).Equals("memory_size_mb", memMB).Equals("provider", provider)
  869. }
  870. if err := db.FetchModelObjects(manager, q, &skus); err != nil {
  871. log.Errorf("failed to get skus with provider %s cpu %d mem %d error: %v", provider, cpu, memMB, err)
  872. return nil, err
  873. }
  874. return skus, nil
  875. }
  876. // 删除表中zone not found的记录
  877. func (manager *SServerSkuManager) DeleteInvalidSkus() error {
  878. _, err := sqlchemy.GetDB().Exec(
  879. fmt.Sprintf(
  880. "delete from %s where length(zone_id) > 0 and zone_id not in (select id from zones_tbl where deleted=0)",
  881. manager.TableSpec().Name(),
  882. ),
  883. )
  884. return err
  885. }
  886. func (manager *SServerSkuManager) SyncPrivateCloudSkus(
  887. ctx context.Context,
  888. userCred mcclient.TokenCredential,
  889. region *SCloudregion,
  890. skus []cloudprovider.ICloudSku,
  891. xor bool,
  892. ) compare.SyncResult {
  893. lockman.LockRawObject(ctx, manager.Keyword(), region.Id)
  894. defer lockman.ReleaseRawObject(ctx, manager.Keyword(), region.Id)
  895. result := compare.SyncResult{}
  896. dbSkus, err := region.GetServerSkus()
  897. if err != nil {
  898. result.Error(errors.Wrapf(err, "GetServerSkus"))
  899. return result
  900. }
  901. removed := make([]SServerSku, 0)
  902. commondb := make([]SServerSku, 0)
  903. commonext := make([]cloudprovider.ICloudSku, 0)
  904. added := make([]cloudprovider.ICloudSku, 0)
  905. err = compare.CompareSets(dbSkus, skus, &removed, &commondb, &commonext, &added)
  906. if err != nil {
  907. result.Error(errors.Wrapf(err, "CompareSets"))
  908. return result
  909. }
  910. for i := 0; i < len(removed); i += 1 {
  911. err = removed[i].RealDelete(ctx, userCred)
  912. if err != nil {
  913. result.DeleteError(err)
  914. continue
  915. }
  916. result.Delete()
  917. }
  918. if !xor {
  919. for i := 0; i < len(commondb); i += 1 {
  920. err = commondb[i].SyncWithPrivateCloudSku(ctx, userCred, commonext[i])
  921. if err != nil {
  922. result.UpdateError(err)
  923. }
  924. result.Update()
  925. }
  926. }
  927. for i := 0; i < len(added); i += 1 {
  928. err := manager.newPrivateCloudSku(ctx, userCred, region, added[i])
  929. if err != nil {
  930. result.AddError(err)
  931. continue
  932. }
  933. result.Add()
  934. }
  935. return result
  936. }
  937. func (self *SServerSku) SyncWithPrivateCloudSku(ctx context.Context, userCred mcclient.TokenCredential, sku cloudprovider.ICloudSku) error {
  938. _, err := db.Update(self, func() error {
  939. self.Status = api.SkuStatusAvailable
  940. self.constructSku(sku)
  941. return nil
  942. })
  943. return err
  944. }
  945. func (self *SServerSku) constructSku(extSku cloudprovider.ICloudSku) {
  946. self.ExternalId = extSku.GetGlobalId()
  947. self.InstanceTypeFamily = extSku.GetInstanceTypeFamily()
  948. self.InstanceTypeCategory = extSku.GetInstanceTypeCategory()
  949. self.PrepaidStatus = extSku.GetPrepaidStatus()
  950. self.PostpaidStatus = extSku.GetPostpaidStatus()
  951. self.CpuArch = extSku.GetCpuArch()
  952. self.CpuCoreCount = extSku.GetCpuCoreCount()
  953. self.MemorySizeMB = extSku.GetMemorySizeMB()
  954. self.OsName = extSku.GetOsName()
  955. self.SysDiskResizable = tristate.NewFromBool(extSku.GetSysDiskResizable())
  956. self.SysDiskType = extSku.GetSysDiskType()
  957. self.SysDiskMinSizeGB = extSku.GetSysDiskMinSizeGB()
  958. self.SysDiskMaxSizeGB = extSku.GetSysDiskMaxSizeGB()
  959. self.AttachedDiskType = extSku.GetAttachedDiskType()
  960. self.AttachedDiskSizeGB = extSku.GetAttachedDiskSizeGB()
  961. self.AttachedDiskCount = extSku.GetAttachedDiskCount()
  962. self.DataDiskTypes = extSku.GetDataDiskTypes()
  963. self.DataDiskMaxCount = extSku.GetDataDiskMaxCount()
  964. self.NicType = extSku.GetNicType()
  965. self.NicMaxCount = extSku.GetNicMaxCount()
  966. self.GpuAttachable = tristate.NewFromBool(extSku.GetGpuAttachable())
  967. self.GpuSpec = extSku.GetGpuSpec()
  968. self.GpuCount = extSku.GetGpuCount()
  969. self.GpuMaxCount = extSku.GetGpuMaxCount()
  970. self.Name = extSku.GetName()
  971. }
  972. func (region *SCloudregion) getMetaUrl(base string, externalId string) string {
  973. if region.Provider == api.CLOUD_PROVIDER_HUAWEI && strings.Contains(region.ExternalId, "_") {
  974. idx := strings.Index(region.ExternalId, "_")
  975. return fmt.Sprintf("%s/%s/%s.json", base, region.ExternalId[:idx], externalId)
  976. }
  977. return fmt.Sprintf("%s/%s/%s.json", base, region.ExternalId, externalId)
  978. }
  979. func (region *SCloudregion) newPublicCloudSku(ctx context.Context, userCred mcclient.TokenCredential, extSku SServerSku) error {
  980. meta, err := yunionmeta.FetchYunionmeta(ctx)
  981. if err != nil {
  982. return err
  983. }
  984. zones, err := region.GetZones()
  985. if err != nil {
  986. return errors.Wrap(err, "GetZones")
  987. }
  988. zoneMaps := map[string]string{}
  989. for _, zone := range zones {
  990. zoneMaps[zone.ExternalId] = zone.Id
  991. }
  992. sku := &SServerSku{}
  993. sku.SetModelManager(ServerSkuManager, sku)
  994. skuUrl := region.getMetaUrl(meta.ServerBase, extSku.ExternalId)
  995. err = meta.Get(skuUrl, sku)
  996. if err != nil {
  997. return errors.Wrapf(err, "Get")
  998. }
  999. if len(sku.ZoneId) > 0 {
  1000. zoneId := sku.ZoneId
  1001. sku.ZoneId = yunionmeta.GetZoneIdBySuffix(zoneMaps, zoneId)
  1002. if len(sku.ZoneId) == 0 {
  1003. return errors.Wrapf(cloudprovider.ErrNotFound, "%v", zoneId)
  1004. }
  1005. }
  1006. // 第一次同步新建的套餐是启用状态
  1007. sku.Enabled = tristate.True
  1008. sku.Status = api.SkuStatusReady
  1009. sku.CloudregionId = region.Id
  1010. sku.Provider = region.Provider
  1011. return ServerSkuManager.TableSpec().Insert(ctx, sku)
  1012. }
  1013. func (manager *SServerSkuManager) newPrivateCloudSku(ctx context.Context, userCred mcclient.TokenCredential, region *SCloudregion, extSku cloudprovider.ICloudSku) error {
  1014. sku := &SServerSku{Provider: region.Provider}
  1015. sku.SetModelManager(manager, sku)
  1016. // 第一次同步新建的套餐是启用状态
  1017. sku.Enabled = tristate.True
  1018. sku.Status = api.SkuStatusReady
  1019. sku.constructSku(extSku)
  1020. sku.CloudregionId = region.Id
  1021. sku.Name = extSku.GetName()
  1022. return manager.TableSpec().Insert(ctx, sku)
  1023. }
  1024. func (self *SServerSku) syncWithCloudSku(ctx context.Context, region *SCloudregion, isLocalChangedStatus bool, extSku SServerSku) error {
  1025. if self.Md5 == extSku.Md5 {
  1026. return nil
  1027. }
  1028. meta, err := yunionmeta.FetchYunionmeta(ctx)
  1029. if err != nil {
  1030. return err
  1031. }
  1032. sku := &SServerSku{}
  1033. skuUrl := region.getMetaUrl(meta.ServerBase, extSku.ExternalId)
  1034. err = meta.Get(skuUrl, sku)
  1035. if err != nil {
  1036. return errors.Wrapf(err, "Get")
  1037. }
  1038. _, err = db.Update(self, func() error {
  1039. if !isLocalChangedStatus {
  1040. self.PrepaidStatus = sku.PrepaidStatus
  1041. self.PostpaidStatus = sku.PostpaidStatus
  1042. }
  1043. self.SysDiskType = sku.SysDiskType
  1044. self.DataDiskTypes = sku.DataDiskTypes
  1045. self.CpuArch = sku.CpuArch
  1046. self.InstanceTypeFamily = sku.InstanceTypeFamily
  1047. self.InstanceTypeCategory = sku.InstanceTypeCategory
  1048. self.LocalCategory = sku.LocalCategory
  1049. self.NicType = sku.NicType
  1050. self.GpuAttachable = sku.GpuAttachable
  1051. self.GpuSpec = sku.GpuSpec
  1052. self.GpuCount = sku.GpuCount
  1053. self.CpuCoreCount = sku.CpuCoreCount
  1054. self.MemorySizeMB = sku.MemorySizeMB
  1055. self.Md5 = sku.Md5
  1056. return nil
  1057. })
  1058. return err
  1059. }
  1060. func (self *SServerSku) MarkAsSoldout(ctx context.Context) error {
  1061. _, err := db.UpdateWithLock(ctx, self, func() error {
  1062. self.PrepaidStatus = api.SkuStatusSoldout
  1063. self.PostpaidStatus = api.SkuStatusSoldout
  1064. return nil
  1065. })
  1066. return errors.Wrap(err, "SServerSku.MarkAsSoldout")
  1067. }
  1068. func (region *SCloudregion) FetchSkusByRegion() ([]SServerSku, error) {
  1069. q := ServerSkuManager.Query().Equals("cloudregion_id", region.Id)
  1070. skus := make([]SServerSku, 0)
  1071. err := db.FetchModelObjects(ServerSkuManager, q, &skus)
  1072. if err != nil {
  1073. return nil, errors.Wrapf(err, "FetchSkusByRegion %s", region.ExternalId)
  1074. }
  1075. return skus, nil
  1076. }
  1077. func (region *SCloudregion) GetUsedSkus() (map[string]bool, error) {
  1078. hosts := HostManager.Query().SubQuery()
  1079. zones := ZoneManager.Query().Equals("cloudregion_id", region.Id).SubQuery()
  1080. q := GuestManager.Query("instance_type").Distinct()
  1081. q = q.Join(hosts, sqlchemy.Equals(q.Field("host_id"), hosts.Field("id")))
  1082. q = q.Join(zones, sqlchemy.Equals(hosts.Field("zone_id"), zones.Field("id")))
  1083. ret := []struct {
  1084. InstanceType string `json:"instance_type"`
  1085. }{}
  1086. err := q.All(&ret)
  1087. if err != nil {
  1088. return nil, errors.Wrapf(err, "GetUsedSkus %s", region.ExternalId)
  1089. }
  1090. usedSkus := make(map[string]bool, 0)
  1091. for _, item := range ret {
  1092. usedSkus[item.InstanceType] = true
  1093. }
  1094. return usedSkus, nil
  1095. }
  1096. // 获取本地已变更过套餐状态的公有云套餐
  1097. func (manager *SServerSkuManager) GetLocalSkus() (map[string]bool, error) {
  1098. q := db.Metadata.Query("obj_id").Equals("obj_type", manager.Keyword()).Equals("key", api.SERVER_SKU_PROJECT_SRC_KEY).Equals("value", api.SERVER_SKU_PROJECT_SRC_VALUE_LOCAL)
  1099. ret := []struct {
  1100. ObjId string `json:"obj_id"`
  1101. }{}
  1102. err := q.All(&ret)
  1103. if err != nil {
  1104. return nil, errors.Wrapf(err, "GetLocalSkus")
  1105. }
  1106. localSkus := make(map[string]bool, 0)
  1107. for _, item := range ret {
  1108. localSkus[item.ObjId] = true
  1109. }
  1110. return localSkus, nil
  1111. }
  1112. func (region *SCloudregion) SyncServerSkus(ctx context.Context, userCred mcclient.TokenCredential, xor bool) compare.SyncResult {
  1113. lockman.LockRawObject(ctx, ServerSkuManager.Keyword(), region.Id)
  1114. defer lockman.ReleaseRawObject(ctx, ServerSkuManager.Keyword(), region.Id)
  1115. result := compare.SyncResult{}
  1116. meta, err := yunionmeta.FetchYunionmeta(ctx)
  1117. if err != nil {
  1118. result.Error(errors.Wrapf(err, "FetchYunionmeta"))
  1119. return result
  1120. }
  1121. extSkus := []SServerSku{}
  1122. err = meta.List(ServerSkuManager.Keyword(), region.ExternalId, &extSkus)
  1123. if err != nil {
  1124. result.Error(errors.Wrapf(err, "List"))
  1125. return result
  1126. }
  1127. dbSkus, err := region.FetchSkusByRegion()
  1128. if err != nil {
  1129. result.Error(errors.Wrapf(err, "FetchSkusByRegion %s", region.ExternalId))
  1130. return result
  1131. }
  1132. removed := make([]SServerSku, 0)
  1133. commondb := make([]SServerSku, 0)
  1134. commonext := make([]SServerSku, 0)
  1135. added := make([]SServerSku, 0)
  1136. err = compare.CompareSets(dbSkus, extSkus, &removed, &commondb, &commonext, &added)
  1137. if err != nil {
  1138. result.Error(errors.Wrapf(err, "CompareSets %s", region.ExternalId))
  1139. return result
  1140. }
  1141. usedSkus, err := region.GetUsedSkus()
  1142. if err != nil {
  1143. result.Error(errors.Wrapf(err, "GetUsedSkus %s", region.ExternalId))
  1144. return result
  1145. }
  1146. localSkus, err := ServerSkuManager.GetLocalSkus()
  1147. if err != nil {
  1148. result.Error(errors.Wrapf(err, "GetLocalSkus"))
  1149. return result
  1150. }
  1151. purgeIds := []string{}
  1152. for i := 0; i < len(removed); i += 1 {
  1153. if usedSkus[removed[i].Name] {
  1154. continue
  1155. }
  1156. purgeIds = append(purgeIds, removed[i].Id)
  1157. }
  1158. if len(purgeIds) > 0 {
  1159. err = db.Purge(ServerSkuManager, "id", purgeIds, true)
  1160. if err != nil {
  1161. result.Error(errors.Wrapf(err, "Purge %s", region.ExternalId))
  1162. } else {
  1163. result.DelCnt += len(purgeIds)
  1164. }
  1165. }
  1166. if !xor {
  1167. for i := 0; i < len(commondb); i += 1 {
  1168. _, isLocalChangedStatus := localSkus[commondb[i].Id]
  1169. err = commondb[i].syncWithCloudSku(ctx, region, isLocalChangedStatus, commonext[i])
  1170. if err != nil {
  1171. result.UpdateError(err)
  1172. } else {
  1173. result.Update()
  1174. }
  1175. }
  1176. }
  1177. ch := make(chan struct{}, options.Options.SkuBatchSync)
  1178. defer close(ch)
  1179. var wg sync.WaitGroup
  1180. for i := 0; i < len(added); i += 1 {
  1181. ch <- struct{}{}
  1182. wg.Add(1)
  1183. go func(sku SServerSku) {
  1184. defer func() {
  1185. wg.Done()
  1186. <-ch
  1187. }()
  1188. err = region.newPublicCloudSku(ctx, userCred, sku)
  1189. if err != nil {
  1190. result.AddError(err)
  1191. } else {
  1192. result.Add()
  1193. }
  1194. }(added[i])
  1195. }
  1196. wg.Wait()
  1197. // notfiy sched manager
  1198. _, err = scheduler.SchedManager.SyncSku(auth.GetAdminSession(ctx, options.Options.Region), false)
  1199. if err != nil {
  1200. log.Errorf("SchedManager SyncSku %s", err)
  1201. }
  1202. return result
  1203. }
  1204. func (manager *SServerSkuManager) InitializeData() error {
  1205. count, err := manager.Query().Equals("cloudregion_id", api.DEFAULT_REGION_ID).IsNullOrEmpty("zone_id").CountWithError()
  1206. if err != nil {
  1207. return errors.Wrapf(err, "fetch default region skus")
  1208. }
  1209. if count == 0 {
  1210. skus := []struct {
  1211. cpu int
  1212. memGb int
  1213. }{
  1214. {1, 1}, {1, 2}, {1, 4}, {1, 8},
  1215. {2, 2}, {2, 4}, {2, 8}, {2, 12}, {2, 16},
  1216. {4, 4}, {4, 12}, {4, 16}, {4, 24}, {4, 32},
  1217. {8, 8}, {8, 16}, {8, 24}, {8, 32}, {8, 64},
  1218. {12, 12}, {12, 16}, {12, 24}, {12, 32}, {12, 64},
  1219. {16, 16}, {16, 24}, {16, 32}, {16, 48}, {16, 64},
  1220. {24, 24}, {24, 32}, {24, 48}, {24, 64}, {24, 128},
  1221. {32, 32}, {32, 48}, {32, 64}, {32, 128},
  1222. }
  1223. for _, item := range skus {
  1224. sku := &SServerSku{}
  1225. sku.CloudregionId = api.DEFAULT_REGION_ID
  1226. sku.CpuCoreCount = item.cpu
  1227. sku.MemorySizeMB = item.memGb * 1024
  1228. sku.IsEmulated = false
  1229. sku.Enabled = tristate.True
  1230. sku.InstanceTypeCategory = api.SkuCategoryGeneralPurpose
  1231. sku.LocalCategory = api.SkuCategoryGeneralPurpose
  1232. sku.InstanceTypeFamily = api.InstanceFamilies[api.SkuCategoryGeneralPurpose]
  1233. sku.Name, _ = genInstanceType(sku.InstanceTypeFamily, int64(item.cpu), int64(item.memGb*1024))
  1234. sku.PrepaidStatus = api.SkuStatusAvailable
  1235. sku.PostpaidStatus = api.SkuStatusAvailable
  1236. sku.SetModelManager(manager, sku)
  1237. err := manager.TableSpec().Insert(context.TODO(), sku)
  1238. if err != nil {
  1239. log.Errorf("ServerSkuManager Initialize local sku %s", err)
  1240. }
  1241. }
  1242. }
  1243. return nil
  1244. }
  1245. func (manager *SServerSkuManager) ListItemExportKeys(ctx context.Context,
  1246. q *sqlchemy.SQuery,
  1247. userCred mcclient.TokenCredential,
  1248. keys stringutils2.SSortedStrings,
  1249. ) (*sqlchemy.SQuery, error) {
  1250. var err error
  1251. q, err = manager.SEnabledStatusStandaloneResourceBaseManager.ListItemExportKeys(ctx, q, userCred, keys)
  1252. if err != nil {
  1253. return nil, errors.Wrap(err, "SEnabledStatusStandaloneResourceBaseManager.ListItemExportKeys")
  1254. }
  1255. if keys.ContainsAny(manager.SCloudregionResourceBaseManager.GetExportKeys()...) {
  1256. q, err = manager.SCloudregionResourceBaseManager.ListItemExportKeys(ctx, q, userCred, keys)
  1257. if err != nil {
  1258. return nil, errors.Wrap(err, "SCloudregionResourceBaseManager.ListItemExportKeys")
  1259. }
  1260. }
  1261. if keys.Contains("zone") {
  1262. q, err = manager.SZoneResourceBaseManager.ListItemExportKeys(ctx, q, userCred, stringutils2.NewSortedStrings([]string{"zone"}))
  1263. if err != nil {
  1264. return nil, errors.Wrap(err, "SZoneResourceBaseManager.ListItemExportKeys")
  1265. }
  1266. }
  1267. return q, nil
  1268. }
  1269. func (manager *SServerSkuManager) PerformSyncSkus(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, input api.SkuSyncInput) (jsonutils.JSONObject, error) {
  1270. return PerformActionSyncSkus(ctx, userCred, manager.Keyword(), input)
  1271. }
  1272. func (manager *SServerSkuManager) GetPropertySyncTasks(ctx context.Context, userCred mcclient.TokenCredential, query api.SkuTaskQueryInput) (jsonutils.JSONObject, error) {
  1273. return GetPropertySkusSyncTasks(ctx, userCred, query)
  1274. }
  1275. func (self *SServerSku) GetICloudSku(ctx context.Context) (cloudprovider.ICloudSku, error) {
  1276. region, err := self.GetRegion()
  1277. if err != nil {
  1278. if errors.Cause(err) == sql.ErrNoRows {
  1279. return nil, errors.Wrapf(cloudprovider.ErrNotFound, "GetRegion")
  1280. }
  1281. return nil, errors.Wrapf(err, "GetRegion")
  1282. }
  1283. providers, err := region.GetCloudproviders()
  1284. if err != nil {
  1285. return nil, errors.Wrapf(err, "GetCloudprovider")
  1286. }
  1287. for i := range providers {
  1288. provider := providers[i]
  1289. driver, err := provider.GetProvider(ctx)
  1290. if err != nil {
  1291. return nil, errors.Wrapf(err, "GetDriver()")
  1292. }
  1293. iRegion, err := driver.GetIRegionById(region.ExternalId)
  1294. if err != nil {
  1295. return nil, errors.Wrapf(err, "GetIRegionById(%s)", region.ExternalId)
  1296. }
  1297. skus, err := iRegion.GetISkus()
  1298. if err != nil {
  1299. return nil, errors.Wrapf(err, "GetICloudSku")
  1300. }
  1301. for i := range skus {
  1302. if skus[i].GetGlobalId() == self.ExternalId {
  1303. return skus[i], nil
  1304. }
  1305. }
  1306. }
  1307. return nil, errors.Wrapf(cloudprovider.ErrNotFound, "%v", self.ExternalId)
  1308. }
  1309. func fetchSkuSyncCloudregions() []SCloudregion {
  1310. cloudregions := []SCloudregion{}
  1311. q := CloudregionManager.Query()
  1312. q = q.In("provider", CloudproviderManager.GetPublicProviderProvidersQuery())
  1313. err := db.FetchModelObjects(CloudregionManager, q, &cloudregions)
  1314. if err != nil {
  1315. log.Errorf("fetchSkuSyncCloudregions.FetchCloudregions failed: %v", err)
  1316. return nil
  1317. }
  1318. return cloudregions
  1319. }
  1320. // 全量同步sku列表.
  1321. func SyncServerSkus(ctx context.Context, userCred mcclient.TokenCredential, isStart bool) {
  1322. // 清理无效的sku
  1323. log.Debugf("DeleteInvalidSkus in processing...")
  1324. ServerSkuManager.DeleteInvalidSkus()
  1325. cloudregions := fetchSkuSyncCloudregions()
  1326. if len(cloudregions) == 0 {
  1327. return
  1328. }
  1329. meta, err := yunionmeta.FetchYunionmeta(ctx)
  1330. if err != nil {
  1331. log.Errorf("FetchYunionmeta %v", err)
  1332. return
  1333. }
  1334. index, err := meta.Index(ServerSkuManager.Keyword())
  1335. if err != nil {
  1336. log.Errorf("getServerSkuIndex error: %v", err)
  1337. return
  1338. }
  1339. for i := range cloudregions {
  1340. region := &cloudregions[i]
  1341. skuMeta := &SServerSku{}
  1342. skuMeta.SetModelManager(ServerSkuManager, skuMeta)
  1343. skuMeta.Id = region.ExternalId
  1344. oldMd5 := db.Metadata.GetStringValue(ctx, skuMeta, db.SKU_METADAT_KEY, userCred)
  1345. newMd5, ok := index[region.ExternalId]
  1346. if !ok || newMd5 == yunionmeta.EMPTY_MD5 || len(oldMd5) > 0 && newMd5 == oldMd5 {
  1347. continue
  1348. }
  1349. db.Metadata.SetValue(ctx, skuMeta, db.SKU_METADAT_KEY, newMd5, userCred)
  1350. result := region.SyncServerSkus(ctx, userCred, false)
  1351. notes := fmt.Sprintf("SyncServerSkusByRegion %s result: %v", region.Name, result.Result())
  1352. log.Debugf("%s", notes)
  1353. }
  1354. }
  1355. // 同步指定region sku列表
  1356. func SyncServerSkusByRegion(ctx context.Context, userCred mcclient.TokenCredential, region *SCloudregion, xor bool) compare.SyncResult {
  1357. result := compare.SyncResult{}
  1358. result = region.SyncServerSkus(ctx, userCred, xor)
  1359. notes := fmt.Sprintf("SyncServerSkusByRegion %s result: %v", region.Name, result.Result())
  1360. log.Infof("%s", notes)
  1361. return result
  1362. }