| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094 |
- // Copyright 2019 Yunion
- //
- // Licensed under the Apache License, Version 2.0 (the "License");
- // you may not use this file except in compliance with the License.
- // You may obtain a copy of the License at
- //
- // http://www.apache.org/licenses/LICENSE-2.0
- //
- // Unless required by applicable law or agreed to in writing, software
- // distributed under the License is distributed on an "AS IS" BASIS,
- // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- // See the License for the specific language governing permissions and
- // limitations under the License.
- package models
- import (
- "context"
- "database/sql"
- "strings"
- "time"
- "yunion.io/x/cloudmux/pkg/cloudprovider"
- "yunion.io/x/jsonutils"
- "yunion.io/x/log"
- "yunion.io/x/pkg/errors"
- "yunion.io/x/pkg/tristate"
- "yunion.io/x/pkg/util/compare"
- "yunion.io/x/sqlchemy"
- "yunion.io/x/onecloud/pkg/apis"
- billing_api "yunion.io/x/onecloud/pkg/apis/billing"
- api "yunion.io/x/onecloud/pkg/apis/compute"
- "yunion.io/x/onecloud/pkg/cloudcommon/db"
- "yunion.io/x/onecloud/pkg/cloudcommon/db/lockman"
- "yunion.io/x/onecloud/pkg/compute/options"
- "yunion.io/x/onecloud/pkg/mcclient"
- "yunion.io/x/onecloud/pkg/util/stringutils2"
- )
- func (manager *SGuestManager) FetchCustomizeColumns(
- ctx context.Context,
- userCred mcclient.TokenCredential,
- query jsonutils.JSONObject,
- objs []interface{},
- fields stringutils2.SSortedStrings,
- isList bool,
- ) []api.ServerDetails {
- rows := make([]api.ServerDetails, len(objs))
- virtRows := manager.SVirtualResourceBaseManager.FetchCustomizeColumns(ctx, userCred, query, objs, fields, isList)
- hostRows := manager.SHostResourceBaseManager.FetchCustomizeColumns(ctx, userCred, query, objs, fields, isList)
- encRows := manager.SEncryptedResourceManager.FetchCustomizeColumns(ctx, userCred, query, objs, fields, isList)
- guestIds := make([]string, len(objs))
- guests := make([]SGuest, len(objs))
- backupHostIds := make([]string, len(objs))
- for i := range objs {
- rows[i] = api.ServerDetails{
- VirtualResourceDetails: virtRows[i],
- HostResourceInfo: hostRows[i],
- EncryptedResourceDetails: encRows[i],
- }
- guest := objs[i].(*SGuest)
- guestIds[i] = guest.GetId()
- backupHostIds[i] = guest.BackupHostId
- guests[i] = *guest
- }
- if len(fields) == 0 || fields.Contains("disk") {
- gds := fetchGuestDisksInfo(guestIds)
- if gds != nil {
- for i := range rows {
- rows[i].DisksInfo = []api.GuestDiskInfo{}
- if disks, ok := gds[guestIds[i]]; ok {
- for j := range disks {
- if len(rows[i].ManagerId) > 0 {
- disks[j].Iops = disks[j].DiskIops
- }
- rows[i].DisksInfo = append(rows[i].DisksInfo, disks[j].GuestDiskInfo)
- }
- }
- rows[i].DiskCount = len(rows[i].DisksInfo)
- shortDescs := []string{}
- for _, info := range rows[i].DisksInfo {
- rows[i].DiskSizeMb += int64(info.SizeMb)
- shortDescs = append(shortDescs, info.ShortDesc())
- }
- rows[i].Disks = strings.Join(shortDescs, "\n")
- }
- }
- }
- if len(fields) == 0 || fields.Contains("snapshotpolicy") {
- counts := fetchGuestSnapshotpolicyInfo(guestIds)
- for i := range rows {
- rows[i].SnapshotpolicyCount = counts[guestIds[i]]
- for j := range rows[i].DisksInfo {
- rows[i].DisksSnapshotpolicyCount += counts[rows[i].DisksInfo[j].Id]
- }
- }
- }
- /*if len(fields) == 0 || fields.Contains("ips") {
- gips := fetchGuestIPs(guestIds, tristate.False)
- if gips != nil {
- for i := range rows {
- if gip, ok := gips[guestIds[i]]; ok {
- rows[i].IPs = strings.Join(gip, ",")
- }
- }
- }
- }*/
- if len(fields) == 0 || fields.Contains("vip") {
- gvips := fetchGuestVips(guestIds)
- if gvips != nil {
- for i := range rows {
- if vips, ok := gvips[guestIds[i]]; ok {
- rows[i].Vip = strings.Join(vips, ",")
- }
- }
- }
- }
- if len(fields) == 0 || fields.Contains("vip_eip") {
- gvips := fetchGuestVipEips(guestIds)
- if gvips != nil {
- for i := range rows {
- if vips, ok := gvips[guestIds[i]]; ok {
- rows[i].VipEip = strings.Join(vips, ",")
- }
- }
- }
- }
- if len(fields) == 0 || fields.Contains("ips") || fields.Contains("macs") || fields.Contains("nics") || fields.Contains("subips") {
- nicsMap := fetchGuestNICs(ctx, guestIds, tristate.False)
- if nicsMap != nil {
- for i := range rows {
- if nics, ok := nicsMap[guestIds[i]]; ok {
- if len(fields) == 0 || fields.Contains("nics") {
- rows[i].Nics = nics
- }
- if len(fields) == 0 || fields.Contains("macs") {
- macs := make([]string, 0, len(nics))
- for _, nic := range nics {
- macs = append(macs, nic.Mac)
- }
- rows[i].Macs = strings.Join(macs, ",")
- }
- if len(fields) == 0 || fields.Contains("ips") {
- ips := make([]string, 0, len(nics))
- for _, nic := range nics {
- if len(nic.IpAddr) > 0 {
- ips = append(ips, nic.IpAddr)
- }
- if len(nic.Ip6Addr) > 0 {
- ips = append(ips, nic.Ip6Addr)
- }
- }
- rows[i].IPs = strings.Join(ips, ",")
- }
- if len(fields) == 0 || fields.Contains("subips") {
- subips := make([]string, 0)
- for _, nic := range nics {
- if len(nic.SubIps) > 0 {
- ips := strings.Split(nic.SubIps, ",")
- subips = append(subips, ips...)
- }
- }
- rows[i].SubIPs = subips
- }
- }
- }
- }
- }
- if len(fields) == 0 || fields.Contains("vpc") || fields.Contains("vpc_id") {
- gvpcs := fetchGuestVpcs(guestIds)
- if gvpcs != nil {
- for i := range rows {
- if gvpc, ok := gvpcs[guestIds[i]]; ok {
- if len(fields) == 0 || fields.Contains("vpc") {
- rows[i].Vpc = strings.Join(gvpc.Vpc, ",")
- }
- if len(fields) == 0 || fields.Contains("vpc_id") {
- rows[i].VpcId = strings.Join(gvpc.VpcId, ",")
- }
- if len(fields) == 0 || fields.Contains("external_access_mode") {
- rows[i].VpcExternalAccessMode = strings.Join(gvpc.ExternalAccessMode, ",")
- }
- }
- }
- }
- }
- if len(fields) == 0 || fields.Contains("secgroups") || fields.Contains("secgroup") {
- gsgs := fetchSecgroups(guestIds)
- if gsgs != nil {
- for i := range rows {
- if gsg, ok := gsgs[guestIds[i]]; ok {
- if len(fields) == 0 || fields.Contains("secgroups") {
- rows[i].Secgroups = gsg
- }
- if len(fields) == 0 || fields.Contains("secgroup") {
- rows[i].Secgroup = gsg[0].Name
- }
- }
- }
- }
- }
- if len(fields) == 0 || fields.Contains("network_secgroups") {
- gnss := fetchGuestNetworkSecgroups(guestIds)
- if gnss != nil {
- for i := range rows {
- if gns, ok := gnss[guestIds[i]]; ok {
- rows[i].NetworkSecgroups = gns
- }
- }
- }
- }
- if len(fields) == 0 || fields.Contains("eip") || fields.Contains("eip_mode") {
- geips := fetchGuestEips(guestIds)
- if geips != nil {
- for i := range rows {
- if eip, ok := geips[guestIds[i]]; ok {
- if len(fields) == 0 || fields.Contains("eip") {
- rows[i].Eip = eip.IpAddr
- }
- if len(fields) == 0 || fields.Contains("eip_mode") {
- rows[i].EipMode = eip.Mode
- }
- }
- }
- }
- }
- if len(fields) == 0 || fields.Contains("keypair") {
- gkps := fetchGuestKeypairs(guestIds)
- if gkps != nil {
- for i := range rows {
- if kps, ok := gkps[guestIds[i]]; ok {
- rows[i].Keypair = kps.Keypair
- }
- }
- }
- }
- if len(fields) == 0 || fields.Contains("isolated_devices") || fields.Contains("is_gpu") {
- gdevs := fetchGuestIsolatedDevices(guestIds)
- if gdevs != nil {
- for i := range rows {
- if gdev, ok := gdevs[guestIds[i]]; ok {
- if len(fields) == 0 || fields.Contains("isolated_devices") {
- rows[i].IsolatedDevices = gdev
- }
- if len(fields) == 0 || fields.Contains("is_gpu") {
- if len(gdev) > 0 {
- rows[i].IsGpu = true
- } else {
- rows[i].IsGpu = false
- }
- }
- } else {
- if len(fields) == 0 || fields.Contains("is_gpu") {
- rows[i].IsGpu = false
- }
- }
- }
- }
- info, _ := fetchGuestGpuInstanceTypes(guestIds)
- if len(info) > 0 {
- for i := range rows {
- gpu, ok := info[guests[i].InstanceType]
- if ok {
- rows[i].IsGpu = true
- rows[i].GpuModel = gpu.Model
- rows[i].GpuCount = gpu.Amount
- }
- }
- }
- }
- if len(fields) == 0 || fields.Contains("cdrom") {
- gcds := fetchGuestCdroms(guestIds)
- if gcds != nil {
- for i := range rows {
- for _, gcd := range gcds[guestIds[i]] {
- if details := gcd.GetDetails(); len(details) > 0 {
- t := api.Cdrom{Ordinal: gcd.Ordinal, Detail: details, BootIndex: gcd.BootIndex, Name: gcd.Name}
- rows[i].Cdrom = append(rows[i].Cdrom, t)
- }
- }
- }
- }
- }
- if len(fields) == 0 || fields.Contains("floppy") {
- gfloppys := fetchGuestFloppys(guestIds)
- if gfloppys != nil {
- for i := range rows {
- for _, gfl := range gfloppys[guestIds[i]] {
- if details := gfl.GetDetails(); len(details) > 0 {
- t := api.Floppy{Ordinal: gfl.Ordinal, Detail: details}
- rows[i].Floppy = append(rows[i].Floppy, t)
- }
- }
- }
- }
- }
- if len(fields) == 0 || fields.Contains("scaling_group") {
- sggs := fetchScalingGroupGuest(guestIds...)
- if sggs != nil && len(sggs) != 0 {
- for i := range rows {
- if sgg, ok := sggs[guestIds[i]]; ok {
- rows[i].ScalingStatus = sgg.GuestStatus
- rows[i].ScalingGroupId = sgg.ScalingGroupId
- }
- }
- }
- }
- if len(fields) == 0 || fields.Contains("backup_host_name") || fields.Contains("backup_host_status") && len(backupHostIds) > 0 {
- backups, _ := fetchGuestBackupInfo(backupHostIds)
- meta := []db.SMetadata{}
- db.Metadata.Query().In("obj_id", guestIds).Equals("obj_type", manager.Keyword()).Equals("key", api.MIRROR_JOB).All(&meta)
- syncStatus := map[string]string{}
- for i := range meta {
- v := meta[i]
- syncStatus[v.ObjId] = v.Value
- }
- if len(backups) > 0 || len(syncStatus) > 0 {
- for i := range rows {
- rows[i].BackupInfo, _ = backups[backupHostIds[i]]
- rows[i].BackupGuestSyncStatus, _ = syncStatus[guestIds[i]]
- }
- }
- }
- if len(fields) == 0 || fields.Contains("container") {
- containers, _ := fetchContainers(guestIds)
- if len(containers) > 0 {
- for i := range rows {
- rows[i].Containers, _ = containers[guestIds[i]]
- }
- }
- }
- for i := range rows {
- if len(fields) == 0 || fields.Contains("auto_delete_at") {
- if guests[i].PendingDeleted {
- pendingDeletedAt := guests[i].PendingDeletedAt.Add(time.Second * time.Duration(options.Options.PendingDeleteExpireSeconds))
- rows[i].AutoDeleteAt = pendingDeletedAt
- }
- }
- if len(fields) == 0 || fields.Contains("can_recycle") {
- if guests[i].BillingType == billing_api.BILLING_TYPE_PREPAID && !guests[i].ExpiredAt.Before(time.Now()) && len(rows[i].ManagerId) > 0 {
- rows[i].CanRecycle = true
- }
- }
- rows[i].IsPrepaidRecycle = (rows[i].HostResourceType == api.HostResourceTypePrepaidRecycle && rows[i].HostBillingType == billing_api.BILLING_TYPE_PREPAID)
- drv, _ := GetDriver(guests[i].Hypervisor, rows[i].Provider)
- if drv != nil {
- rows[i].CdromSupport, _ = drv.IsSupportCdrom(&guests[i])
- rows[i].FloppySupport, _ = drv.IsSupportFloppy(&guests[i])
- rows[i].MonitorUrl = drv.FetchMonitorUrl(ctx, &guests[i])
- }
- if len(guests[i].HostId) == 0 && guests[i].Status == api.VM_SCHEDULE_FAILED {
- rows[i].Brand = "Unknown"
- rows[i].Provider = "Unknown"
- }
- if !isList {
- rows[i].Networks = guests[i].getNetworksDetails()
- rows[i].VirtualIps = strings.Join(guests[i].getVirtualIPs(), ",")
- rows[i].SecurityRules = guests[i].getSecurityGroupsRules()
- osName := guests[i].GetOS()
- if len(osName) > 0 {
- rows[i].OsName = osName
- if len(guests[i].OsType) == 0 {
- rows[i].OsType = osName
- }
- }
- if userCred.HasSystemAdminPrivilege() {
- rows[i].AdminSecurityRules = guests[i].getAdminSecurityRules()
- }
- }
- }
- return rows
- }
- type sGustDiskSize struct {
- GuestId string
- DiskSizeMb int64
- DiskCount int
- }
- type GuestDiskInfo struct {
- DiskIops int
- api.GuestDiskInfo
- }
- type sGuestDiskInfo struct {
- GuestDiskInfo
- GuestId string
- }
- func fetchGuestDisksInfo(guestIds []string) map[string][]GuestDiskInfo {
- disks := DiskManager.Query().SubQuery()
- guestdisks := GuestdiskManager.Query().SubQuery()
- storages := StorageManager.Query().SubQuery()
- q := disks.Query(
- disks.Field("id"),
- disks.Field("name"),
- disks.Field("fs_format"),
- disks.Field("disk_type"),
- guestdisks.Field("index"),
- disks.Field("disk_size").Label("size"),
- disks.Field("disk_format"),
- guestdisks.Field("driver"),
- guestdisks.Field("cache_mode"),
- guestdisks.Field("aio_mode"),
- disks.Field("auto_reset"),
- storages.Field("medium_type"),
- storages.Field("storage_type"),
- guestdisks.Field("iops"),
- disks.Field("iops").Label("disk_iops"),
- disks.Field("throughput"),
- guestdisks.Field("bps"),
- disks.Field("template_id").Label("image_id"),
- guestdisks.Field("guest_id"),
- guestdisks.Field("boot_index"),
- disks.Field("storage_id"),
- disks.Field("preallocation"),
- )
- q = q.Join(guestdisks, sqlchemy.Equals(guestdisks.Field("disk_id"), disks.Field("id")))
- q = q.Join(storages, sqlchemy.Equals(disks.Field("storage_id"), storages.Field("id")))
- q = q.Filter(sqlchemy.In(guestdisks.Field("guest_id"), guestIds)).Asc(guestdisks.Field("index"))
- gds := []sGuestDiskInfo{}
- err := q.All(&gds)
- if err != nil {
- log.Errorf("fetchGuestDisksInfo: %v", err)
- return nil
- }
- imageIds := []string{}
- ret := map[string][]GuestDiskInfo{}
- for i := range gds {
- if len(gds[i].ImageId) > 0 {
- imageIds = append(imageIds, gds[i].ImageId)
- }
- _, ok := ret[gds[i].GuestId]
- if !ok {
- ret[gds[i].GuestId] = []GuestDiskInfo{}
- }
- ret[gds[i].GuestId] = append(ret[gds[i].GuestId], gds[i].GuestDiskInfo)
- }
- imageNames, err := db.FetchIdNameMap2(CachedimageManager, imageIds)
- if err != nil {
- return ret
- }
- for guestId, infos := range ret {
- for i := range infos {
- if len(infos[i].ImageId) > 0 {
- ret[guestId][i].Image, _ = imageNames[infos[i].ImageId]
- }
- }
- }
- return ret
- }
- func fetchGuestSnapshotpolicyInfo(guestIds []string) map[string]int {
- ret := map[string]int{}
- disks := GuestdiskManager.Query("disk_id").In("guest_id", guestIds).SubQuery()
- spq := SnapshotPolicyResourceManager.Query()
- spq = spq.Filter(sqlchemy.OR(
- sqlchemy.In(spq.Field("resource_id"), guestIds),
- sqlchemy.In(spq.Field("resource_id"), disks),
- ))
- sq := spq.SubQuery()
- q := sq.Query(
- sq.Field("resource_id"),
- sqlchemy.COUNT("count", sq.Field("snapshotpolicy_id")),
- ).GroupBy(sq.Field("resource_id"))
- counts := []struct {
- ResourceId string
- Count int
- }{}
- err := q.All(&counts)
- if err != nil {
- return nil
- }
- for _, count := range counts {
- ret[count.ResourceId] = count.Count
- }
- return ret
- }
- func (guest *SGuest) GetDisksSize() int {
- return guest.getDiskSize()
- }
- func (guest *SGuest) getDiskSize() int {
- result := fetchGuestDisksInfo([]string{guest.Id})
- if result == nil {
- return -1
- }
- gds, ok := result[guest.Id]
- if !ok {
- return -1
- }
- size := 0
- for _, gd := range gds {
- size += gd.SizeMb
- }
- return size
- }
- func fetchGuestIPs(guestIds []string, virtual tristate.TriState) map[string][]string {
- guestnetworks := GuestnetworkManager.Query().SubQuery()
- q := guestnetworks.Query(guestnetworks.Field("guest_id"), guestnetworks.Field("ip_addr"))
- q = q.In("guest_id", guestIds)
- if virtual.IsTrue() {
- q = q.IsTrue("virtual")
- } else if virtual.IsFalse() {
- q = q.IsFalse("virtual")
- }
- q = q.IsNotEmpty("ip_addr")
- q = q.Asc("ip_addr")
- type sGuestIdIpAddr struct {
- GuestId string
- IpAddr string
- }
- gias := make([]sGuestIdIpAddr, 0)
- err := q.All(&gias)
- if err != nil && errors.Cause(err) != sql.ErrNoRows {
- return nil
- }
- ret := make(map[string][]string)
- for i := range gias {
- if _, ok := ret[gias[i].GuestId]; !ok {
- ret[gias[i].GuestId] = make([]string, 0)
- }
- ret[gias[i].GuestId] = append(ret[gias[i].GuestId], gias[i].IpAddr)
- }
- return ret
- }
- func fetchGuestVips(guestIds []string) map[string][]string {
- groupguests := GroupguestManager.Query().SubQuery()
- groupnetworks := GroupnetworkManager.Query().SubQuery()
- q := groupnetworks.Query(groupnetworks.Field("ip_addr"), groupnetworks.Field("ip6_addr"), groupguests.Field("guest_id"))
- q = q.Join(groupguests, sqlchemy.Equals(q.Field("group_id"), groupguests.Field("group_id")))
- q = q.In("guest_id", guestIds)
- type sGuestVip struct {
- IpAddr string
- Ip6Addr string
- GuestId string
- }
- gvips := make([]sGuestVip, 0)
- err := q.All(&gvips)
- if err != nil && errors.Cause(err) != sql.ErrNoRows {
- return nil
- }
- ret := make(map[string][]string)
- for i := range gvips {
- if _, ok := ret[gvips[i].GuestId]; !ok {
- ret[gvips[i].GuestId] = make([]string, 0)
- }
- if len(gvips[i].IpAddr) > 0 {
- ret[gvips[i].GuestId] = append(ret[gvips[i].GuestId], gvips[i].IpAddr)
- }
- if len(gvips[i].Ip6Addr) > 0 {
- ret[gvips[i].GuestId] = append(ret[gvips[i].GuestId], gvips[i].Ip6Addr)
- }
- }
- return ret
- }
- func fetchGuestVipEips(guestIds []string) map[string][]string {
- groupguests := GroupguestManager.Query().SubQuery()
- eips := ElasticipManager.Query().Equals("associate_type", api.EIP_ASSOCIATE_TYPE_INSTANCE_GROUP).SubQuery()
- q := eips.Query(eips.Field("ip_addr"), groupguests.Field("guest_id"))
- q = q.Join(groupguests, sqlchemy.Equals(eips.Field("associate_id"), groupguests.Field("group_id")))
- q = q.In("guest_id", guestIds)
- type sGuestVip struct {
- IpAddr string
- GuestId string
- }
- gvips := make([]sGuestVip, 0)
- err := q.All(&gvips)
- if err != nil && errors.Cause(err) != sql.ErrNoRows {
- return nil
- }
- ret := make(map[string][]string)
- for i := range gvips {
- if _, ok := ret[gvips[i].GuestId]; !ok {
- ret[gvips[i].GuestId] = make([]string, 0)
- }
- ret[gvips[i].GuestId] = append(ret[gvips[i].GuestId], gvips[i].IpAddr)
- }
- return ret
- }
- func fetchGuestNICs(ctx context.Context, guestIds []string, virtual tristate.TriState) map[string][]api.GuestnetworkShortDesc {
- netq := NetworkManager.Query().SubQuery()
- wirq := WireManager.Query().SubQuery()
- subIPQ := NetworkAddressManager.fetchSubIpsQuery(api.NetworkAddressParentTypeGuestnetwork)
- subIP := subIPQ.SubQuery()
- gnwq := GuestnetworkManager.Query()
- q := gnwq.AppendField(
- gnwq.Field("guest_id"),
- gnwq.Field("index"),
- gnwq.Field("ip_addr"),
- gnwq.Field("ip6_addr"),
- gnwq.Field("mac_addr").Label("mac"),
- gnwq.Field("team_with"),
- gnwq.Field("network_id"), // caution: do not alias netq.id as network_id
- gnwq.Field("port_mappings"),
- gnwq.Field("ifname"),
- gnwq.Field("is_default"),
- wirq.Field("vpc_id"),
- subIP.Field("sub_ips"),
- )
- q = q.Join(netq, sqlchemy.Equals(netq.Field("id"), gnwq.Field("network_id")))
- q = q.Join(wirq, sqlchemy.Equals(wirq.Field("id"), netq.Field("wire_id")))
- q = q.LeftJoin(subIP, sqlchemy.Equals(q.Field("row_id"), subIP.Field("parent_id")))
- q = q.In("guest_id", guestIds)
- var descs []struct {
- GuestId string `json:"guest_id"`
- api.GuestnetworkShortDesc
- }
- if err := q.All(&descs); err != nil {
- if errors.Cause(err) != sql.ErrNoRows {
- log.Errorf("query guest nics info: %v", err)
- }
- return nil
- }
- ret := map[string][]api.GuestnetworkShortDesc{}
- for i := range descs {
- desc := &descs[i]
- guestId := desc.GuestId
- if _, ok := ret[guestId]; !ok {
- ret[guestId] = []api.GuestnetworkShortDesc{desc.GuestnetworkShortDesc}
- } else {
- ret[guestId] = append(ret[guestId], desc.GuestnetworkShortDesc)
- }
- }
- return ret
- }
- func (self *SGuest) GetRealIPs() []string {
- result := fetchGuestIPs([]string{self.Id}, tristate.False)
- if result == nil {
- return nil
- }
- if ret, ok := result[self.Id]; ok {
- return ret
- }
- return nil
- }
- func (self *SGuest) fetchNICShortDesc(ctx context.Context) []api.GuestnetworkShortDesc {
- nicsMap := fetchGuestNICs(ctx, []string{self.Id}, tristate.False)
- if nicsMap == nil {
- return nil
- }
- return nicsMap[self.Id]
- }
- type sGuestVpcsInfo struct {
- GuestId string
- Vpc []string
- VpcId []string
- ExternalAccessMode []string
- }
- func fetchGuestVpcs(guestIds []string) map[string]sGuestVpcsInfo {
- vpcs := VpcManager.Query().SubQuery()
- wires := WireManager.Query().SubQuery()
- networks := NetworkManager.Query().SubQuery()
- guestnetworks := GuestnetworkManager.Query().SubQuery()
- q := vpcs.Query(guestnetworks.Field("guest_id"), vpcs.Field("id"), vpcs.Field("name"), vpcs.Field("external_access_mode"))
- q = q.Join(wires, sqlchemy.Equals(vpcs.Field("id"), wires.Field("vpc_id")))
- q = q.Join(networks, sqlchemy.Equals(wires.Field("id"), networks.Field("wire_id")))
- q = q.Join(guestnetworks, sqlchemy.Equals(networks.Field("id"), guestnetworks.Field("network_id")))
- q = q.Filter(sqlchemy.In(guestnetworks.Field("guest_id"), guestIds))
- q = q.Distinct()
- type sGuestVpcInfo struct {
- GuestId string
- Id string
- Name string
- ExternalAccessMode string
- }
- gvpcs := make([]sGuestVpcInfo, 0)
- err := q.All(&gvpcs)
- if err != nil {
- return nil
- }
- ret := make(map[string]sGuestVpcsInfo)
- for i := range gvpcs {
- gvpc, ok := ret[gvpcs[i].GuestId]
- if !ok {
- gvpc = sGuestVpcsInfo{
- GuestId: gvpcs[i].GuestId,
- Vpc: make([]string, 0),
- VpcId: make([]string, 0),
- ExternalAccessMode: make([]string, 0),
- }
- }
- gvpc.VpcId = append(gvpc.VpcId, gvpcs[i].Id)
- gvpc.Vpc = append(gvpc.Vpc, gvpcs[i].Name)
- gvpc.ExternalAccessMode = append(gvpc.ExternalAccessMode, gvpcs[i].ExternalAccessMode)
- ret[gvpcs[i].GuestId] = gvpc
- }
- return ret
- }
- func fetchSecgroups(guestIds []string) map[string][]apis.StandaloneShortDesc {
- secgroups := SecurityGroupManager.Query().SubQuery()
- guestsecgroups := GuestsecgroupManager.Query().SubQuery()
- guests := GuestManager.Query().SubQuery()
- q1 := guests.Query(guests.Field("id").Label("guest_id"),
- guests.Field("secgrp_id").Label("secgroup_id"))
- q1 = q1.Filter(sqlchemy.In(guests.Field("id"), guestIds))
- q2 := guestsecgroups.Query(guestsecgroups.Field("guest_id"), guestsecgroups.Field("secgroup_id"))
- q2 = q2.Filter(sqlchemy.In(guestsecgroups.Field("guest_id"), guestIds))
- uq := sqlchemy.Union(q1, q2)
- q := uq.Query(uq.Field("guest_id"), uq.Field("secgroup_id"), secgroups.Field("name").Label("secgroup_name"))
- q = q.Join(secgroups, sqlchemy.Equals(uq.Field("secgroup_id"), secgroups.Field("id")))
- type sGuestSecgroupInfo struct {
- SecgroupId string
- SecgroupName string
- GuestId string
- }
- gsgs := make([]sGuestSecgroupInfo, 0)
- err := q.All(&gsgs)
- if err != nil {
- return nil
- }
- ret := make(map[string][]apis.StandaloneShortDesc)
- for i := range gsgs {
- gsg, ok := ret[gsgs[i].GuestId]
- if !ok {
- gsg = make([]apis.StandaloneShortDesc, 0)
- }
- gsg = append(gsg, apis.StandaloneShortDesc{
- Id: gsgs[i].SecgroupId,
- Name: gsgs[i].SecgroupName,
- })
- ret[gsgs[i].GuestId] = gsg
- }
- return ret
- }
- func fetchGuestNetworkSecgroups(guestIds []string) map[string][]api.GuestnetworkSecgroupShortDesc {
- guestnetworks := GuestnetworkManager.Query().SubQuery()
- guestnetworksecgroups := GuestnetworksecgroupManager.Query().SubQuery()
- secgroups := SecurityGroupManager.Query().SubQuery()
- q := guestnetworksecgroups.Query(
- guestnetworksecgroups.Field("guest_id"),
- guestnetworksecgroups.Field("network_index"),
- guestnetworksecgroups.Field("secgroup_id"),
- guestnetworks.Field("mac_addr").Label("mac"),
- secgroups.Field("name").Label("secgroup_name"),
- )
- q = q.Join(guestnetworks, sqlchemy.AND(
- sqlchemy.Equals(guestnetworks.Field("guest_id"), guestnetworksecgroups.Field("guest_id")),
- sqlchemy.Equals(guestnetworks.Field("index"), guestnetworksecgroups.Field("network_index")),
- ))
- q = q.Join(secgroups, sqlchemy.Equals(secgroups.Field("id"), guestnetworksecgroups.Field("secgroup_id")))
- q = q.Filter(sqlchemy.In(guestnetworksecgroups.Field("guest_id"), guestIds))
- type sGuestNetworkSecgroupInfo struct {
- GuestId string
- NetworkIndex int
- SecgroupId string
- SecgroupName string
- Mac string
- }
- gnss := make([]sGuestNetworkSecgroupInfo, 0)
- err := q.All(&gnss)
- if err != nil && errors.Cause(err) != sql.ErrNoRows {
- log.Errorf("fetchGuestNetworkSecgroups query error: %s", err)
- return nil
- }
- groupedSecgroups := make(map[string][]sGuestNetworkSecgroupInfo)
- for i := range gnss {
- groupedSecgroup, ok := groupedSecgroups[gnss[i].GuestId]
- if !ok {
- groupedSecgroup = make([]sGuestNetworkSecgroupInfo, 0)
- }
- groupedSecgroups[gnss[i].GuestId] = append(groupedSecgroup, gnss[i])
- }
- ret := make(map[string][]api.GuestnetworkSecgroupShortDesc)
- for guestId, secgroups := range groupedSecgroups {
- networkGroupedSecgroups := make(map[int][]sGuestNetworkSecgroupInfo)
- for i := range secgroups {
- secgroupInfos, ok := networkGroupedSecgroups[secgroups[i].NetworkIndex]
- if !ok {
- secgroupInfos = make([]sGuestNetworkSecgroupInfo, 0)
- }
- networkGroupedSecgroups[secgroups[i].NetworkIndex] = append(secgroupInfos, secgroups[i])
- }
- guestnetworkSecgroups := make([]api.GuestnetworkSecgroupShortDesc, 0)
- for networkIndex, secgroups := range networkGroupedSecgroups {
- //networkSecgroupsDesc := make([]api.GuestnetworkSecgroupShortDesc, 0)
- nsDesc := api.GuestnetworkSecgroupShortDesc{
- NetworkIndex: networkIndex,
- Mac: secgroups[0].Mac,
- }
- for i := range secgroups {
- nsDesc.Secgroups = append(nsDesc.Secgroups, apis.StandaloneShortDesc{
- Id: secgroups[i].SecgroupId,
- Name: secgroups[i].SecgroupName,
- })
- }
- guestnetworkSecgroups = append(guestnetworkSecgroups, nsDesc)
- }
- ret[guestId] = guestnetworkSecgroups
- }
- return ret
- }
- type sEipInfo struct {
- IpAddr string
- Mode string
- GuestId string
- }
- func fetchGuestEips(guestIds []string) map[string]sEipInfo {
- eips := ElasticipManager.Query().SubQuery()
- q := eips.Query(eips.Field("ip_addr"), eips.Field("mode"), eips.Field("associate_id").Label("guest_id"))
- q = q.Equals("associate_type", api.EIP_ASSOCIATE_TYPE_SERVER)
- q = q.In("associate_id", guestIds)
- geips := make([]sEipInfo, 0)
- err := q.All(&geips)
- if err != nil {
- return nil
- }
- ret := make(map[string]sEipInfo)
- for i := range geips {
- ret[geips[i].GuestId] = geips[i]
- }
- return ret
- }
- type sGuestKeypair struct {
- GuestId string
- Keypair string
- }
- func fetchGuestKeypairs(guestIds []string) map[string]sGuestKeypair {
- keypairs := KeypairManager.Query().SubQuery()
- guests := GuestManager.Query().SubQuery()
- q := guests.Query(guests.Field("id").Label("guest_id"), keypairs.Field("name").Label("keypair"))
- q = q.Join(keypairs, sqlchemy.Equals(guests.Field("keypair_id"), keypairs.Field("id")))
- q = q.Filter(sqlchemy.In(guests.Field("id"), guestIds))
- gkps := make([]sGuestKeypair, 0)
- err := q.All(&gkps)
- if err != nil {
- return nil
- }
- ret := make(map[string]sGuestKeypair)
- for i := range gkps {
- ret[gkps[i].GuestId] = gkps[i]
- }
- return ret
- }
- func fetchGuestGpuInstanceTypes(guestIds []string) (map[string]*GpuSpec, error) {
- ret := map[string]*GpuSpec{}
- sq := GuestManager.Query("instance_type").In("id", guestIds).SubQuery()
- q := ServerSkuManager.Query("name", "gpu_spec", "gpu_count").In("name", sq).IsNotEmpty("gpu_spec").Distinct()
- gpus := []struct {
- Name string
- GpuSpec string
- GpuCount string
- }{}
- err := q.All(&gpus)
- if err != nil {
- return ret, err
- }
- for _, gpu := range gpus {
- ret[gpu.Name] = &GpuSpec{Model: gpu.GpuSpec, Amount: gpu.GpuCount}
- }
- return ret, nil
- }
- func fetchGuestBackupInfo(hostIds []string) (map[string]api.BackupInfo, error) {
- ret := map[string]api.BackupInfo{}
- hosts := []SHost{}
- err := HostManager.Query().In("id", hostIds).All(&hosts)
- if err != nil {
- return nil, err
- }
- for _, host := range hosts {
- ret[host.Id] = api.BackupInfo{BackupHostName: host.Name, BackupHostStatus: host.HostStatus}
- }
- return ret, nil
- }
- func fetchContainers(guestIds []string) (map[string][]*api.PodContainerDesc, error) {
- ret := map[string][]*api.PodContainerDesc{}
- containers := []SContainer{}
- err := GetContainerManager().Query().In("guest_id", guestIds).Asc("created_at").All(&containers)
- if err != nil {
- return nil, err
- }
- for i := range containers {
- container := containers[i]
- _, ok := ret[container.GuestId]
- if !ok {
- ret[container.GuestId] = []*api.PodContainerDesc{}
- }
- desc := &api.PodContainerDesc{
- Id: container.GetId(),
- Name: container.GetName(),
- Status: container.Status,
- }
- if container.Spec != nil {
- desc.Image = container.Spec.Image
- }
- ret[container.GuestId] = append(ret[container.GuestId], desc)
- }
- return ret, nil
- }
- func fetchGuestIsolatedDevices(guestIds []string) map[string][]api.SIsolatedDevice {
- q := IsolatedDeviceManager.Query().In("guest_id", guestIds)
- devs := make([]SIsolatedDevice, 0)
- err := q.All(&devs)
- if err != nil {
- return nil
- }
- ret := make(map[string][]api.SIsolatedDevice)
- for i := range devs {
- dev := api.SIsolatedDevice{}
- dev.Id = devs[i].Id
- dev.HostId = devs[i].HostId
- dev.DevType = devs[i].DevType
- dev.Model = devs[i].Model
- dev.GuestId = devs[i].GuestId
- dev.Addr = devs[i].Addr
- dev.VendorDeviceId = devs[i].VendorDeviceId
- dev.NumaNode = byte(devs[i].NumaNode)
- gdevs, ok := ret[devs[i].GuestId]
- if !ok {
- gdevs = make([]api.SIsolatedDevice, 0)
- }
- gdevs = append(gdevs, dev)
- ret[devs[i].GuestId] = gdevs
- }
- return ret
- }
- func fetchGuestCdroms(guestIds []string) map[string][]SGuestcdrom {
- sq := GuestcdromManager.Query().In("id", guestIds).SubQuery()
- image := CachedimageManager.Query().SubQuery()
- q := sq.Query(
- sq.Field("id"),
- sq.Field("path"),
- sq.Field("boot_index"),
- sq.Field("image_id"),
- image.Field("size"),
- image.Field("name"),
- )
- q = q.LeftJoin(image, sqlchemy.Equals(sq.Field("image_id"), image.Field("id")))
- gcds := make([]SGuestcdrom, 0)
- err := q.All(&gcds)
- if err != nil {
- return nil
- }
- ret := make(map[string][]SGuestcdrom)
- for i := range gcds {
- ret[gcds[i].Id] = append(ret[gcds[i].Id], gcds[i])
- }
- return ret
- }
- func fetchGuestFloppys(guestIds []string) map[string][]SGuestfloppy {
- q := GuestFloppyManager.Query().In("id", guestIds)
- gfls := make([]SGuestfloppy, 0)
- err := q.All(&gfls)
- if err != nil {
- return nil
- }
- ret := make(map[string][]SGuestfloppy)
- for i := range gfls {
- ret[gfls[i].Id] = append(ret[gfls[i].Id], gfls[i])
- }
- return ret
- }
- func fetchScalingGroupGuest(guestIds ...string) map[string]SScalingGroupGuest {
- q := ScalingGroupGuestManager.Query().In("guest_id", guestIds)
- sggs := make([]SScalingGroupGuest, 0)
- err := q.All(&sggs)
- if err != nil {
- return nil
- }
- ret := make(map[string]SScalingGroupGuest)
- for i := range sggs {
- ret[sggs[i].GuestId] = sggs[i]
- }
- return ret
- }
- func (self *SGuest) SyncInstanceSnapshots(ctx context.Context, userCred mcclient.TokenCredential, provider *SCloudprovider) compare.SyncResult {
- syncResult := compare.SyncResult{}
- extGuest, err := self.GetIVM(ctx)
- if err != nil {
- syncResult.Error(err)
- return syncResult
- }
- extSnapshots, err := extGuest.GetInstanceSnapshots()
- if errors.Cause(err) == errors.ErrNotImplemented {
- return syncResult
- }
- syncOwnerId := provider.GetOwnerId()
- localSnapshots, err := self.GetInstanceSnapshots()
- if err != nil {
- syncResult.Error(err)
- return syncResult
- }
- lockman.LockClass(ctx, InstanceSnapshotManager, db.GetLockClassKey(InstanceSnapshotManager, syncOwnerId))
- defer lockman.ReleaseClass(ctx, InstanceSnapshotManager, db.GetLockClassKey(InstanceSnapshotManager, syncOwnerId))
- removed := make([]SInstanceSnapshot, 0)
- commondb := make([]SInstanceSnapshot, 0)
- commonext := make([]cloudprovider.ICloudInstanceSnapshot, 0)
- added := make([]cloudprovider.ICloudInstanceSnapshot, 0)
- err = compare.CompareSets(localSnapshots, extSnapshots, &removed, &commondb, &commonext, &added)
- if err != nil {
- syncResult.Error(err)
- return syncResult
- }
- for i := 0; i < len(removed); i += 1 {
- err = removed[i].syncRemoveCloudInstanceSnapshot(ctx, userCred)
- if err != nil {
- syncResult.DeleteError(err)
- } else {
- syncResult.Delete()
- }
- }
- for i := 0; i < len(commondb); i += 1 {
- err = commondb[i].SyncWithCloudInstanceSnapshot(ctx, userCred, commonext[i], self)
- if err != nil {
- syncResult.UpdateError(err)
- } else {
- syncResult.Update()
- }
- }
- for i := 0; i < len(added); i += 1 {
- _, err := InstanceSnapshotManager.newFromCloudInstanceSnapshot(ctx, userCred, added[i], self)
- if err != nil {
- syncResult.AddError(err)
- } else {
- syncResult.Add()
- }
- }
- return syncResult
- }
|