Cloudpods/backend/pkg/vpcagent/models/models.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 (
	"fmt"
	"sort"

	"yunion.io/x/log"

	compute_models "yunion.io/x/onecloud/pkg/compute/models"
	"yunion.io/x/onecloud/pkg/util/netutils2"
)

type Vpc struct {
	compute_models.SVpc

	RouteTable *RouteTable `json:"-"`

	Wire     *Wire    `json:"-"`
	Networks Networks `json:"-"`
}

func (el *Vpc) Copy() *Vpc {
	return &Vpc{
		SVpc: el.SVpc,
	}
}

type RouteTable struct {
	compute_models.SRouteTable

	Vpc *Vpc
}

func (el *RouteTable) Copy() *RouteTable {
	return &RouteTable{
		SRouteTable: el.SRouteTable,
	}
}

type Wire struct {
	compute_models.SWire

	Vpc *Vpc
}

func (el *Wire) Copy() *Wire {
	return &Wire{
		SWire: el.SWire,
	}
}

type Network struct {
	compute_models.SNetwork

	Vpc                  *Vpc                 `json:"-"`
	Wire                 *Wire                `json:"-"`
	Guestnetworks        Guestnetworks        `json:"-"`
	Groupnetworks        Groupnetworks        `json:"-"`
	LoadbalancerNetworks LoadbalancerNetworks `json:"-"`
	Elasticips           Elasticips           `json:"-"`
}

func (el *Network) Copy() *Network {
	return &Network{
		SNetwork: el.SNetwork,
	}
}

type Guestnetwork struct {
	compute_models.SGuestnetwork

	// Guest could be nil for when the guest is pending_deleted
	Guest     *Guest           `json:"-"`
	Network   *Network         `json:"-"`
	Elasticip *Elasticip       `json:"-"`
	SubIPs    NetworkAddresses `json:"-"`

	// guest nic level secgroup
	Guestnetworksecgroups Guestnetworksecgroups `json:"-"`
}

func (el *Guestnetwork) Copy() *Guestnetwork {
	return &Guestnetwork{
		SGuestnetwork: el.SGuestnetwork,
	}
}

type NetworkAddress struct {
	compute_models.SNetworkAddress

	Guestnetwork *Guestnetwork `json:"-"`
	Network      *Network      `json:"-"`
}

func (el *NetworkAddress) Copy() *NetworkAddress {
	return &NetworkAddress{
		SNetworkAddress: el.SNetworkAddress,
	}
}

type Guestnetworksecgroup struct {
	compute_models.SGuestnetworksecgroup

	SecurityGroup *SecurityGroup `json:"-"`
}

func (el *Guestnetworksecgroup) ModelSetKey() string {
	return fmt.Sprintf("%s/%d", el.GuestId, el.NetworkIndex)
}

func (el *Guestnetworksecgroup) Copy() *Guestnetworksecgroup {
	return &Guestnetworksecgroup{
		SGuestnetworksecgroup: el.SGuestnetworksecgroup,
	}
}

type Guest struct {
	compute_models.SGuest

	Host               *Host          `json:"-"`
	AdminSecurityGroup *SecurityGroup `json:"-"`
	SecurityGroups     SecurityGroups `json:"-"`
	Guestnetworks      Guestnetworks  `json:"-"`

	Groups Groups `json:"-"`
}

func (el *Guest) Copy() *Guest {
	return &Guest{
		SGuest: el.SGuest,
	}
}

func (el *Guest) GetVips() []string {
	ret := make([]string, 0)
	for _, g := range el.Groups {
		for _, gn := range g.Groupnetworks {
			ret = append(ret, fmt.Sprintf("%s/%d", gn.IpAddr, gn.Network.GuestIpMask))
		}
	}
	return ret
}

type GuestnetworkList []*Guestnetwork

func (a GuestnetworkList) Len() int           { return len(a) }
func (a GuestnetworkList) Swap(i, j int)      { a[i], a[j] = a[j], a[i] }
func (a GuestnetworkList) Less(i, j int) bool { return a[i].Index < a[j].Index }

func (el *Guest) FixIsDefaults() {
	gns := make([]*Guestnetwork, 0, len(el.Guestnetworks))
	for _, v := range el.Guestnetworks {
		gns = append(gns, v)
	}
	sort.Sort(GuestnetworkList(gns))
	defaultCnt := 0
	nics := netutils2.SNicInfoList{}
	for _, gn := range gns {
		if gn.Network == nil {
			log.Debugf("guest %s %s", gn.GuestId, gn.NetworkId)
			continue
		}
		if gn.IsDefault {
			defaultCnt++
		}
		nics = nics.Add(gn.MacAddr, gn.IpAddr, gn.Network.GuestGateway, gn.Ip6Addr, gn.Network.GuestGateway6, gn.IsDefault)
	}
	if defaultCnt != 1 {
		gwMac, _ := nics.FindDefaultNicMac()
		for _, gn := range gns {
			if gn.MacAddr == gwMac {
				gn.IsDefault = true
			} else {
				gn.IsDefault = false
			}
		}
	}
}

type Host struct {
	compute_models.SHost
}

func (el *Host) Copy() *Host {
	return &Host{
		SHost: el.SHost,
	}
}

type Guestsecgroup struct {
	compute_models.SGuestsecgroup

	Guest         *Guest         `json:"-"`
	SecurityGroup *SecurityGroup `json:"-"`
}

func (el *Guestsecgroup) ModelSetKey() string {
	return el.GuestId + "/" + el.SecgroupId
}

func (el *Guestsecgroup) Copy() *Guestsecgroup {
	return &Guestsecgroup{
		SGuestsecgroup: el.SGuestsecgroup,
	}
}

type SecurityGroup struct {
	compute_models.SSecurityGroup

	SecurityGroupRules SecurityGroupRules `json:"-"`
}

func (el *SecurityGroup) Copy() *SecurityGroup {
	return &SecurityGroup{
		SSecurityGroup: el.SSecurityGroup,
	}
}

type SecurityGroupRule struct {
	compute_models.SSecurityGroupRule

	SecurityGroup *SecurityGroup `json:"-"`
}

func (el *SecurityGroupRule) Copy() *SecurityGroupRule {
	return &SecurityGroupRule{
		SSecurityGroupRule: el.SSecurityGroupRule,
	}
}

type Elasticip struct {
	compute_models.SElasticip

	Network             *Network             `json:"-"`
	Guestnetwork        *Guestnetwork        `json:"-"`
	Groupnetwork        *Groupnetwork        `json:"-"`
	LoadbalancerNetwork *LoadbalancerNetwork `json:"-"`
}

func (el *Elasticip) Copy() *Elasticip {
	return &Elasticip{
		SElasticip: el.SElasticip,
	}
}

type DnsRecord struct {
	compute_models.SDnsRecord

	DnsZone *DnsZone
}

func (el *DnsRecord) Copy() *DnsRecord {
	return &DnsRecord{
		SDnsRecord: el.SDnsRecord,
	}
}

type DnsZone struct {
	compute_models.SDnsZone

	Records DnsRecords
}

func (el *DnsZone) Copy() *DnsZone {
	return &DnsZone{
		SDnsZone: el.SDnsZone,
	}
}

type Groupguest struct {
	compute_models.SGroupguest

	Group *Group `json:"-"`
	Guest *Guest `json:"-"`
}

func (el *Groupguest) Copy() *Groupguest {
	return &Groupguest{
		SGroupguest: el.SGroupguest,
	}
}

type Groupnetwork struct {
	compute_models.SGroupnetwork

	Network *Network `json:"-"`
	Group   *Group   `json:"-"`

	Elasticip *Elasticip `json:"-"`
}

func (el *Groupnetwork) Copy() *Groupnetwork {
	return &Groupnetwork{
		SGroupnetwork: el.SGroupnetwork,
	}
}

func (el *Groupnetwork) GetGuestNetworks() []*Guestnetwork {
	ret := make([]*Guestnetwork, 0)
	if el.Group == nil {
		log.Errorf("Nil group for groupnetwork %s %s", el.GroupId, el.NetworkId)
	} else if el.Group.Groupguests == nil {
		log.Errorf("Nil groupguests for group %s", el.GroupId)
	} else {
		for _, gg := range el.Group.Groupguests {
			if gg.Guest == nil {
				log.Errorf("Nil guest for groupguest %s %s", gg.GroupId, gg.GuestId)
			} else if gg.Guest.Guestnetworks == nil {
				log.Errorf("Nil guestnetworks for guest %s", gg.GuestId)
			} else {
				for _, gn := range gg.Guest.Guestnetworks {
					ret = append(ret, gn)
				}
			}
		}
	}
	return ret
}

type Group struct {
	compute_models.SGroup

	Groupguests   Groupguests   `json:"-"`
	Groupnetworks Groupnetworks `json:"-"`
}

func (el *Group) Copy() *Group {
	return &Group{
		SGroup: el.SGroup,
	}
}

type LoadbalancerNetwork struct {
	compute_models.SLoadbalancerNetwork

	Network               *Network              `json:"-"`
	Elasticip             *Elasticip            `json:"-"`
	LoadbalancerListeners LoadbalancerListeners `json:"-"`
}

func (el *LoadbalancerNetwork) Copy() *LoadbalancerNetwork {
	return &LoadbalancerNetwork{
		SLoadbalancerNetwork: el.SLoadbalancerNetwork,
	}
}

func (el *LoadbalancerNetwork) OrderedLoadbalancerListeners() []*LoadbalancerListener {
	lblisteners := make([]*LoadbalancerListener, 0, len(el.LoadbalancerListeners))
	for _, lblistener := range el.LoadbalancerListeners {
		lblisteners = append(lblisteners, lblistener)
	}
	sort.Slice(lblisteners, func(i, j int) bool {
		return lblisteners[i].Id < lblisteners[j].Id
	})
	return lblisteners
}

type LoadbalancerListener struct {
	compute_models.SLoadbalancerListener

	LoadbalancerAcl *LoadbalancerAcl `json:"-"`
}

func (el *LoadbalancerListener) Copy() *LoadbalancerListener {
	return &LoadbalancerListener{
		SLoadbalancerListener: el.SLoadbalancerListener,
	}
}

type LoadbalancerAcl struct {
	compute_models.SLoadbalancerAcl
}

func (el *LoadbalancerAcl) Copy() *LoadbalancerAcl {
	return &LoadbalancerAcl{
		SLoadbalancerAcl: el.SLoadbalancerAcl,
	}
}