xds_config_endpoint_v3_localitylbendpoints.go

package convert

import (
	envoy_config_endpoint_v3 "github.com/envoyproxy/go-control-plane/envoy/config/endpoint/v3"
	"github.com/pipeproxy/pipe-xds/internal/config"
	"github.com/pipeproxy/pipe/bind"
)

func Convert_config_endpoint_v3_LocalityLbEndpoints(conf *config.ConfigCtx, c *envoy_config_endpoint_v3.LocalityLbEndpoints) (bind.StreamDialer, error) {
	dialers := []bind.LbStreamDialerWeight{}
	for _, lbEndpoint := range c.LbEndpoints {
		dialer, err := Convert_config_endpoint_v3_LbEndpoint(conf, lbEndpoint)
		if err != nil {
			return nil, err
		}

		dialers = append(dialers, bind.LbStreamDialerWeight{
			Weight: uint(lbEndpoint.LoadBalancingWeight.GetValue()),
			Dialer: dialer,
		})
	}

	var d bind.StreamDialer
	switch len(dialers) {
	case 0:
		d = bind.NoneStreamDialer{}
	case 1:
		d = dialers[0].Dialer
	default:
		d = bind.LbStreamDialerConfig{
			Policy:  bind.RoundRobinBalancePolicy{},
			Dialers: dialers,
		}
	}
	return d, nil
}