mirror of
https://github.com/yunionio/cloudpods.git
synced 2026-05-07 06:02:09 +08:00
1095 lines
31 KiB
Go
1095 lines
31 KiB
Go
// 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
|
|
}
|