Cloudpods/backend/vendor/yunion.io/x/cloudmux/pkg/multicloud/aws/loadbalancerbackend.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 aws

import (
	"context"
	"fmt"

	api "yunion.io/x/cloudmux/pkg/apis/compute"
	"yunion.io/x/cloudmux/pkg/cloudprovider"
	"yunion.io/x/cloudmux/pkg/multicloud"
	"yunion.io/x/pkg/util/netutils"
)

type SElbBackends struct {
	TargetHealthDescriptions []SElbBackend `xml:"TargetHealthDescriptions>member"`
}

type SElbBackend struct {
	multicloud.SResourceBase
	AwsTags
	group *SElbBackendGroup

	Target       Target       `json:"Target"`
	TargetHealth TargetHealth `json:"TargetHealth"`
}

type Target struct {
	Id   string `json:"Id"`
	Port int    `json:"Port"`
}

type TargetHealth struct {
	State       string `json:"State"`
	Reason      string `json:"Reason"`
	Description string `json:"Description"`
}

func (self *SElbBackend) GetId() string {
	return fmt.Sprintf("%s::%s::%d", self.group.GetId(), self.Target.Id, self.Target.Port)
}

func (self *SElbBackend) GetName() string {
	return self.GetId()
}

func (self *SElbBackend) GetGlobalId() string {
	return self.GetId()
}

func (self *SElbBackend) GetStatus() string {
	return api.LB_STATUS_ENABLED
}

func (self *SElbBackend) Refresh() error {
	return nil
}

func (self *SElbBackend) IsEmulated() bool {
	return false
}

func (self *SElbBackend) GetProjectId() string {
	return ""
}

func (self *SElbBackend) GetWeight() int {
	return 0
}

func (self *SElbBackend) GetPort() int {
	return self.Target.Port
}

func (self *SElbBackend) GetBackendType() string {
	return api.LB_BACKEND_GUEST
}

func (self *SElbBackend) GetBackendRole() string {
	return api.LB_BACKEND_ROLE_DEFAULT
}

func (self *SElbBackend) GetBackendId() string {
	return self.Target.Id
}

func (self *SElbBackend) Update(ctx context.Context, opts *cloudprovider.SLoadbalancerBackend) error {
	return self.group.lb.region.SyncElbBackend(self.GetId(), self.GetBackendId(), self.Target.Port, opts.Port)
}

func (self *SElbBackend) GetIpAddress() string {
	_, err := netutils.NewIPV4Addr(self.Target.Id)
	if err != nil {
		return ""
	}
	return self.Target.Id
}

func (self *SRegion) SyncElbBackend(backendId, serverId string, oldPort, newPort int) error {
	err := self.RemoveElbBackend(backendId, serverId, 0, oldPort)
	if err != nil {
		return err
	}

	_, err = self.AddElbBackend(backendId, serverId, 0, newPort)
	if err != nil {
		return err
	}

	return nil
}

func (self *SElbBackend) GetDescription() string {
	return self.AwsTags.GetDescription()
}