package strategy

import "gitee.com/qprodn/gsms/provider"

type RandomLoadBalanceStrategy struct {
	providerMap map[string]provider.SmsProvider
}

func (r *RandomLoadBalanceStrategy) GetSmsProvider() provider.SmsProvider {
	for _, v := range r.providerMap {
		return v
	}
	return nil
}

func (r *RandomLoadBalanceStrategy) SetProviderMap(providerMap map[string]provider.SmsProvider) {
	r.providerMap = providerMap
}