cellohelper

type InstanceLimitManager interface {
	// GetLimit get InstanceLimits of ecs instance
	GetLimit() InstanceLimits
	// Update update InstanceLimits of ecs instance
	Update()
	// UpdateTrunk update trunk eni to InstanceLimits
	UpdateTrunk(trunk *types.ENI)
}

type InstanceLimits struct {
	InstanceLimitsAttr
	ENICustomer int
	// currently support only one
	TrunkENI *types.ENI
}

type InstanceLimitsAttr struct {
	ENITotal       int
	ENIQuota       int
	IPv4MaxPerENI  int
	IPv6MaxPerENI  int
	TrunkSupported bool
}

type InstanceMetadata struct {
	VpcId            string
	InstanceId       string
	InstanceType     string
	PrimaryENIId     string
	PrimaryENIMac    string
	AvailabilityZone string
	Region           string
}

type InstanceMetadataGetter interface {
	GetVpcId() string
	GetInstanceId() string
	GetInstanceType() string
	GetPrimaryENIId() string
	GetPrimaryENIMac() string
	GetAvailabilityZone() string
	GetRegion() string
}

type PodSubnet struct {
	ZoneId         string `json:"zoneId"`
	VpcId          string `json:"vpcId"`
	SubnetId       string `json:"subnetId"`
	IPv4Cidr       string `json:"ipv4Cidr,omitempty"`
	IPv6Cidr       string `json:"ipv6Cidr,omitempty"`
	TotalIpv4Count int    `json:"totalIpv4Count"`

	// status
	sync.RWMutex
	Disable                 bool      `json:"disable,omitempty"`
	AvailableIpAddressCount int       `json:"availableIpAddressCount"`
	LastUpdate              time.Time `json:"lastUpdate"`
}

type PodSubnetManagerConfig struct {
	EventLimiter                *rate.Limiter
	SubnetInsufficientThreshold float64
	// contains filtered or unexported fields
}

type PodSubnetManagerOption func(config *PodSubnetManagerConfig)

type SecurityGroupManager interface {
	// GetSecurityGroups get security groups used by pods
	GetSecurityGroups() []string
	// UpdateSecurityGroups update security groups used by pods
	UpdateSecurityGroups(sec []string) error
}

type SortablePodSubnets []*PodSubnet

type Status struct {
	ZoneId           string                `json:"zone_id"`
	VpcId            string                `json:"vpc_id"`
	PodSubnets       map[string]*PodSubnet `json:"pod_subnets,omitempty"`
	LegacyPodSubnets map[string]*PodSubnet `json:"legacy_pod_subnets,omitempty"`
}

type SubnetManager interface {
	// FlushSubnets flush PodSubnet in SubnetManager
	FlushSubnets(subnetIds ...string) error

	// GetPodSubnet get PodSubnet, get by openapi if not exist in SubnetManager
	GetPodSubnet(subnetId string) *PodSubnet

	// GetUpdatedPodSubnet update PodSubnet and then get it
	GetUpdatedPodSubnet(subnetId string) (subnet *PodSubnet, err error)

	// SelectSubnet select a PodSubnet from available PodSubnets in SubnetManager
	SelectSubnet(ipFamily types.IPFamily, options ...UpdateSubnetsStatusOption) *PodSubnet

	// UpdateSubnetsStatus update status of all PodSubnet in SubnetManager
	// If the option of aging is carried, subnets within the time limit will not be updated
	UpdateSubnetsStatus(options ...UpdateSubnetsStatusOption) error

	// DisableSubnet disable a PodSubnet, it will not be used
	DisableSubnet(subnetId string)

	// Status return status of SubnetManager
	Status() *Status
}

type UpdateSubnetsStatusConfig struct {
	Aging time.Duration
}

type UpdateSubnetsStatusOption func(config *UpdateSubnetsStatusConfig)

type VolcAPI interface {

	// AllocENI creates an ENI and attaches it to the instance
	AllocENI(subnet string, securityGroups []string, trunk bool, ipCnt int) (*types.ENI, error)

	// FreeENI detaches ENI interface and deletes it
	FreeENI(eniID string) error

	// GetENI get eni by mac from metadata
	GetENI(mac string) (*types.ENI, error)

	// GetAttachedENIs return counts of attached enis except rdma eni and list of eni created by cello
	GetAttachedENIs(withTrunk bool) (int, []*types.ENI, error)

	GetSecondaryENIMACs() ([]string, error)

	// GetENIIPList returns the IPs for a given ENI from instance metadata service
	GetENIIPList(eniMac string) ([]net.IP, []net.IP, error)

	// AllocIPAddresses allocates numIPs IP addresses on an ENI
	AllocIPAddresses(eniID, eniMac string, v4Cnt, v6Cnt int) ([]net.IP, []net.IP, error)

	// DeallocIPAddresses deallocates the IP addresses from an ENI
	DeallocIPAddresses(eniID, eniMac string, ipv4s, ipv6s []net.IP) error

	// GetInstanceLimit return instance InstanceLimits
	GetInstanceLimit() (*InstanceLimits, error)
}

type VolcApiImpl struct {

	// metadata info
	InstanceMetadataGetter
	// contains filtered or unexported fields
}