multicast_listener.go

package netx

import (
	"fmt"
	"io"

	"github.com/simia-tech/netx/value"
)

var listenMulticastFuncs = map[string]ListenMulticastFunc{}

// ListenMulticastFunc defines the signature of the ListenMulticast function.
type ListenMulticastFunc func(string, *value.Options) (io.ReadCloser, error)

// RegisterListenMulticast registers the provided Listen method under the provided network name.
func RegisterListenMulticast(network string, listenMulticastFunc ListenMulticastFunc) {
	listenMulticastFuncs[network] = listenMulticastFunc
}

// RegisteredListenMulticastNetworks returns the available networks for the ListenMulticast function.
func RegisteredListenMulticastNetworks() []string {
	networks := []string{}
	for network := range listenMulticastFuncs {
		networks = append(networks, network)
	}
	return networks
}

// ListenMulticast creates a multicast connection on the provided network at the provided address.
func ListenMulticast(network, address string, options ...value.Option) (io.ReadCloser, error) {
	o := &value.Options{}
	for _, option := range options {
		if option == nil {
			continue
		}
		if err := option(o); err != nil {
			return nil, err
		}
	}

	listenMulticastFunc, ok := listenMulticastFuncs[network]
	if ok {
		return listenMulticastFunc(address, o)
	}
	return nil, fmt.Errorf("no ListenMulticast function registered for network [%s]", network)
}