unistack-org/micro
4
0
Fork 3
Code Issues 20 Pull Requests 1 Actions Packages Projects Releases 281 Wiki Activity

Use DialMode/ListenMode

Browse Source
This commit is contained in:
Asim Aslam 2019-10-15 16:08:38 +01:00
parent ca934951ad
commit be5093798b
4 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
network/default.go
View File

@ -822,7 +822,7 @@ func (n *network) Connect() error {
} }
// dial into ControlChannel to send route adverts // dial into ControlChannel to send route adverts
ctrlClient, err := n.tunnel.Dial(ControlChannel, tunnel.DialMulticast()) ctrlClient, err := n.tunnel.Dial(ControlChannel, tunnel.DialMode(tunnel.Multicast))
if err != nil { if err != nil {
n.Unlock() n.Unlock()
return err return err
@ -831,14 +831,14 @@ func (n *network) Connect() error {
n.tunClient[ControlChannel] = ctrlClient n.tunClient[ControlChannel] = ctrlClient
// listen on ControlChannel // listen on ControlChannel
ctrlListener, err := n.tunnel.Listen(ControlChannel, tunnel.ListenMulticast()) ctrlListener, err := n.tunnel.Listen(ControlChannel, tunnel.ListenMode(tunnel.Multicast))
if err != nil { if err != nil {
n.Unlock() n.Unlock()
return err return err
} }
// dial into NetworkChannel to send network messages // dial into NetworkChannel to send network messages
netClient, err := n.tunnel.Dial(NetworkChannel, tunnel.DialMulticast()) netClient, err := n.tunnel.Dial(NetworkChannel, tunnel.DialMode(tunnel.Multicast))
if err != nil { if err != nil {
n.Unlock() n.Unlock()
return err return err
@ -847,7 +847,7 @@ func (n *network) Connect() error {
n.tunClient[NetworkChannel] = netClient n.tunClient[NetworkChannel] = netClient
// listen on NetworkChannel // listen on NetworkChannel
netListener, err := n.tunnel.Listen(NetworkChannel, tunnel.ListenMulticast()) netListener, err := n.tunnel.Listen(NetworkChannel, tunnel.ListenMode(tunnel.Multicast))
if err != nil { if err != nil {
n.Unlock() n.Unlock()
return err return err

4
tunnel/broker/broker.go
View File

@ -58,7 +58,7 @@ func (t *tunBroker) Disconnect() error {
func (t *tunBroker) Publish(topic string, m *broker.Message, opts ...broker.PublishOption) error { func (t *tunBroker) Publish(topic string, m *broker.Message, opts ...broker.PublishOption) error {
// TODO: this is probably inefficient, we might want to just maintain an open connection // TODO: this is probably inefficient, we might want to just maintain an open connection
// it may be easier to add broadcast to the tunnel // it may be easier to add broadcast to the tunnel
c, err := t.tunnel.Dial(topic, tunnel.DialMulticast()) c, err := t.tunnel.Dial(topic, tunnel.DialMode(tunnel.Multicast))
if err != nil { if err != nil {
return err return err
} }
@ -71,7 +71,7 @@ func (t *tunBroker) Publish(topic string, m *broker.Message, opts ...broker.Publ
} }
func (t *tunBroker) Subscribe(topic string, h broker.Handler, opts ...broker.SubscribeOption) (broker.Subscriber, error) { func (t *tunBroker) Subscribe(topic string, h broker.Handler, opts ...broker.SubscribeOption) (broker.Subscriber, error) {
l, err := t.tunnel.Listen(topic, tunnel.ListenMulticast()) l, err := t.tunnel.Listen(topic, tunnel.ListenMode(tunnel.Multicast))
if err != nil { if err != nil {
return nil, err return nil, err
} }

8
tunnel/options.go
View File

@ -95,18 +95,18 @@ func DefaultOptions() Options {
} }
// Listen options // Listen options
func ListenMulticast() ListenOption { func ListenMode(m Mode) ListenOption {
return func(o *ListenOptions) { return func(o *ListenOptions) {
o.Mode = Multicast o.Mode = m
} }
} }
// Dial options // Dial options
// Dial multicast sets the multicast option to send only to those mapped // Dial multicast sets the multicast option to send only to those mapped
func DialMulticast() DialOption { func DialMode(m Mode) DialOption {
return func(o *DialOptions) { return func(o *DialOptions) {
o.Mode = Multicast o.Mode = m
} }
} }

2
tunnel/tunnel.go
View File

@ -13,7 +13,7 @@ const (
Unicast Mode = iota Unicast Mode = iota
// send to all channel listeners // send to all channel listeners
Multicast Multicast
// send to all listeners // send to all links
Broadcast Broadcast
) )