105 lines
2.4 KiB
Go
105 lines
2.4 KiB
Go
// Copyright (C) 2014 The Syncthing Authors.
|
|
//
|
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this file,
|
|
// You can obtain one at https://mozilla.org/MPL/2.0/.
|
|
|
|
package beacon
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net"
|
|
"time"
|
|
|
|
"github.com/thejerf/suture/v4"
|
|
|
|
"github.com/syncthing/syncthing/lib/svcutil"
|
|
)
|
|
|
|
type recv struct {
|
|
data []byte
|
|
src net.Addr
|
|
}
|
|
|
|
type Interface interface {
|
|
suture.Service
|
|
fmt.Stringer
|
|
Send(data []byte)
|
|
Recv() ([]byte, net.Addr)
|
|
Error() error
|
|
}
|
|
|
|
type cast struct {
|
|
*suture.Supervisor
|
|
name string
|
|
reader svcutil.ServiceWithError
|
|
writer svcutil.ServiceWithError
|
|
outbox chan recv
|
|
inbox chan []byte
|
|
stopped chan struct{}
|
|
}
|
|
|
|
// newCast creates a base object for multi- or broadcasting. Afterwards the
|
|
// caller needs to set reader and writer with the addReader and addWriter
|
|
// methods to get a functional implementation of Interface.
|
|
func newCast(name string) *cast {
|
|
// Only log restarts in debug mode.
|
|
spec := svcutil.SpecWithDebugLogger(l)
|
|
// Don't retry too frenetically: an error to open a socket or
|
|
// whatever is usually something that is either permanent or takes
|
|
// a while to get solved...
|
|
spec.FailureThreshold = 2
|
|
spec.FailureBackoff = 60 * time.Second
|
|
c := &cast{
|
|
Supervisor: suture.New(name, spec),
|
|
name: name,
|
|
inbox: make(chan []byte),
|
|
outbox: make(chan recv, 16),
|
|
stopped: make(chan struct{}),
|
|
}
|
|
svcutil.OnSupervisorDone(c.Supervisor, func() { close(c.stopped) })
|
|
return c
|
|
}
|
|
|
|
func (c *cast) addReader(svc func(context.Context) error) {
|
|
c.reader = c.createService(svc, "reader")
|
|
c.Add(c.reader)
|
|
}
|
|
|
|
func (c *cast) addWriter(svc func(ctx context.Context) error) {
|
|
c.writer = c.createService(svc, "writer")
|
|
c.Add(c.writer)
|
|
}
|
|
|
|
func (c *cast) createService(svc func(context.Context) error, suffix string) svcutil.ServiceWithError {
|
|
return svcutil.AsService(svc, fmt.Sprintf("%s/%s", c, suffix))
|
|
}
|
|
|
|
func (c *cast) String() string {
|
|
return fmt.Sprintf("%s@%p", c.name, c)
|
|
}
|
|
|
|
func (c *cast) Send(data []byte) {
|
|
select {
|
|
case c.inbox <- data:
|
|
case <-c.stopped:
|
|
}
|
|
}
|
|
|
|
func (c *cast) Recv() ([]byte, net.Addr) {
|
|
select {
|
|
case recv := <-c.outbox:
|
|
return recv.data, recv.src
|
|
case <-c.stopped:
|
|
}
|
|
return nil, nil
|
|
}
|
|
|
|
func (c *cast) Error() error {
|
|
if err := c.reader.Error(); err != nil {
|
|
return err
|
|
}
|
|
return c.writer.Error()
|
|
}
|