alibabacloudstack/extension_slb.go

package alibabacloudstack

import (
	"fmt"
	"strings"

	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
)

type SchedulerType string

const (
	WRRScheduler = SchedulerType("wrr")
	WLCScheduler = SchedulerType("wlc")
	RRScheduler  = SchedulerType("rr")
)

type FlagType string

const (
	OnFlag  = FlagType("on")
	OffFlag = FlagType("off")
)

type RsType string

const (
	ENI = FlagType("eni")
	ECS = FlagType("ecs")
)

type AclType string

const (
	AclTypeBlack = AclType("black")
	AclTypeWhite = AclType("white")
)

type StickySessionType string

const (
	InsertStickySessionType = StickySessionType("insert")
	ServerStickySessionType = StickySessionType("server")
)

const BackendServerPort = -520

type HealthCheckHttpCodeType string

const (
	HTTP_2XX = HealthCheckHttpCodeType("http_2xx")
	HTTP_3XX = HealthCheckHttpCodeType("http_3xx")
	HTTP_4XX = HealthCheckHttpCodeType("http_4xx")
	HTTP_5XX = HealthCheckHttpCodeType("http_5xx")
)

type HealthCheckType string

const (
	TCPHealthCheckType  = HealthCheckType("tcp")
	HTTPHealthCheckType = HealthCheckType("http")
)

type LoadBalancerSpecType string

const (
	S1Small  = "slb.s1.small"
	S2Small  = "slb.s2.small"
	S2Medium = "slb.s2.medium"
	S3Small  = "slb.s3.small"
	S3Medium = "slb.s3.medium"
	S3Large  = "slb.s3.large"
	S4Large  = "slb.s4.large"
)

type ListenerErr struct {
	ErrType string
	Err     error
}

func (e *ListenerErr) Error() string {
	return e.ErrType + " " + e.Err.Error()

}

func expandBackendServersToString(list []interface{}, weight int, serverType string) string {
	if len(list) < 1 {
		return ""
	}
	var items []string
	for _, id := range list {
		items = append(items, fmt.Sprintf("{'ServerId':'%s','Weight':'%d', 'Type': '%s'}", id, weight, strings.Trim(serverType, " ")))
	}
	return fmt.Sprintf("[%s]", strings.Join(items, COMMA_SEPARATED))
}

func expandBackendServersWithPortToString(items []interface{}) string {

	if len(items) < 1 {
		return ""
	}
	var servers []string
	for _, server := range items {
		s := server.(map[string]interface{})

		var serverIds []interface{}
		var port, weight int
		var serveType, serverId string
		if v, ok := s["port"]; ok {
			port = v.(int)
		}
		if v, ok := s["weight"]; ok {
			weight = v.(int)
		}
		if v, ok := s["type"]; ok {
			serveType = v.(string)
		}
		if v, ok := s["server_id"]; ok {
			serverId = v.(string)
			str := fmt.Sprintf("{'ServerId':'%s','Port':'%d','Weight':'%d', 'Type': '%s'}", strings.Trim(serverId, " "), port, weight, strings.Trim(serveType, " "))
			servers = append(servers, str)
		}
		if v, ok := s["server_ids"]; ok {
			serverIds = v.([]interface{})
			for _, id := range serverIds {
				str := fmt.Sprintf("{'ServerId':'%s','Port':'%d','Weight':'%d', 'Type': '%s'}", strings.Trim(id.(string), " "), port, weight, strings.Trim(serveType, " "))
				servers = append(servers, str)
			}
		}

	}
	return fmt.Sprintf("[%s]", strings.Join(servers, COMMA_SEPARATED))
}

func expandMasterSlaveBackendServersToString(items []interface{}) string {

	if len(items) < 1 {
		return ""
	}
	var servers []string
	for _, server := range items {
		s := server.(map[string]interface{})

		var serverId string
		var port, weight, isBackup int
		var stype, serveType string
		if v, ok := s["server_id"]; ok {
			serverId = v.(string)
		}
		if v, ok := s["port"]; ok {
			port = v.(int)
		}
		if v, ok := s["weight"]; ok {
			weight = v.(int)
		}
		if v, ok := s["type"]; ok {
			stype = v.(string)
		}
		if v, ok := s["server_type"]; ok {
			serveType = v.(string)
		}
		if v, ok := s["is_backup"]; ok {
			isBackup = v.(int)
		}
		str := fmt.Sprintf("{'ServerId':'%s','Port':'%d','Weight':'%d', 'Type': '%s', 'ServerType': '%s', 'IsBackup':'%d'}", strings.Trim(serverId, " "), port, weight, strings.Trim(stype, " "), strings.Trim(serveType, " "), isBackup)
		servers = append(servers, str)

	}
	return fmt.Sprintf("[%s]", strings.Join(servers, COMMA_SEPARATED))
}

func expandBackendServersInfoToString(items []interface{}) string {

	if len(items) < 1 {
		return ""
	}
	var servers []string
	for _, server := range items {
		s := server.(map[string]interface{})

		var serverId string
		var weight int
		var stype string
		if v, ok := s["server_id"]; ok {
			serverId = v.(string)
		}
		if v, ok := s["weight"]; ok {
			weight = v.(int)
		}
		if v, ok := s["type"]; ok {
			stype = v.(string)
		}
		str := fmt.Sprintf("{'ServerId':'%s','Weight':'%d', 'Type': '%s'}", strings.Trim(serverId, " "), weight, strings.Trim(stype, " "))
		servers = append(servers, str)

	}
	return fmt.Sprintf("[%s]", strings.Join(servers, COMMA_SEPARATED))
}

func expandBackendServersWithoutTypeToString(items []interface{}) string {

	if len(items) < 1 {
		return ""
	}
	var servers []string
	for _, server := range items {
		s := server.(map[string]interface{})

		var serverId string
		var weight int
		if v, ok := s["server_id"]; ok {
			serverId = v.(string)
		}
		if v, ok := s["weight"]; ok {
			weight = v.(int)
		}
		str := fmt.Sprintf("{'ServerId':'%s','Weight':'%d'}", strings.Trim(serverId, " "), weight)
		servers = append(servers, str)

	}
	return fmt.Sprintf("[%s]", strings.Join(servers, COMMA_SEPARATED))
}

func getIdSetFromServers(items []interface{}) *schema.Set {
	rmId := make([]interface{}, 0)
	for _, item := range items {
		server := item.(map[string]interface{})
		rmId = append(rmId, fmt.Sprintf("%s", server["server_id"]))
	}
	return schema.NewSet(schema.HashString, rmId)
}

func getIdPortSetFromServers(items []interface{}) *schema.Set {
	rmIdPort := make([]interface{}, 0)
	for _, item := range items {
		server := item.(map[string]interface{})
		if v, ok := server["server_ids"]; ok {
			serverIds := v.([]interface{})
			for _, id := range serverIds {
				rmIdPort = append(rmIdPort, fmt.Sprintf("%s:%d", id, server["port"]))
			}
		}
	}
	return schema.NewSet(schema.HashString, rmIdPort)
}