various: remove unused prober field; change expiration to 30s by default
Signed-off-by: NotAShelf <raf@notashelf.dev> Change-Id: I877beeca7c653f7bb2479c4ef2eeed266a6a6964
This commit is contained in:
parent
2d2c49bbe7
commit
00bbf5608c
2 changed files with 3 additions and 6 deletions
|
|
@ -136,7 +136,7 @@ func runServer(_ *cobra.Command, _ []string) error {
|
||||||
// Setup mDNS discovery if enabled
|
// Setup mDNS discovery if enabled
|
||||||
var discoveryMgr *discovery.Discovery
|
var discoveryMgr *discovery.Discovery
|
||||||
if cfg.Discovery.Enabled {
|
if cfg.Discovery.Enabled {
|
||||||
discoveryMgr, err = discovery.New(cfg.Discovery, p)
|
discoveryMgr, err = discovery.New(cfg.Discovery)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("create discovery manager: %w", err)
|
return fmt.Errorf("create discovery manager: %w", err)
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -10,13 +10,11 @@ import (
|
||||||
|
|
||||||
"github.com/grandcat/zeroconf"
|
"github.com/grandcat/zeroconf"
|
||||||
"notashelf.dev/ncro/internal/config"
|
"notashelf.dev/ncro/internal/config"
|
||||||
"notashelf.dev/ncro/internal/prober"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// Tracks discovered nix-serve instances and maintains the upstream list.
|
// Tracks discovered nix-serve instances and maintains the upstream list.
|
||||||
type Discovery struct {
|
type Discovery struct {
|
||||||
cfg config.DiscoveryConfig
|
cfg config.DiscoveryConfig
|
||||||
prober *prober.Prober
|
|
||||||
resolver *zeroconf.Resolver
|
resolver *zeroconf.Resolver
|
||||||
discovered map[string]*discoveredPeer
|
discovered map[string]*discoveredPeer
|
||||||
mu sync.RWMutex
|
mu sync.RWMutex
|
||||||
|
|
@ -33,7 +31,7 @@ type discoveredPeer struct {
|
||||||
}
|
}
|
||||||
|
|
||||||
// Creates a new Discovery manager.
|
// Creates a new Discovery manager.
|
||||||
func New(cfg config.DiscoveryConfig, p *prober.Prober) (*Discovery, error) {
|
func New(cfg config.DiscoveryConfig) (*Discovery, error) {
|
||||||
resolver, err := zeroconf.NewResolver(nil)
|
resolver, err := zeroconf.NewResolver(nil)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("create zeroconf resolver: %w", err)
|
return nil, fmt.Errorf("create zeroconf resolver: %w", err)
|
||||||
|
|
@ -41,7 +39,6 @@ func New(cfg config.DiscoveryConfig, p *prober.Prober) (*Discovery, error) {
|
||||||
|
|
||||||
return &Discovery{
|
return &Discovery{
|
||||||
cfg: cfg,
|
cfg: cfg,
|
||||||
prober: p,
|
|
||||||
resolver: resolver,
|
resolver: resolver,
|
||||||
discovered: make(map[string]*discoveredPeer),
|
discovered: make(map[string]*discoveredPeer),
|
||||||
stopCh: make(chan struct{}),
|
stopCh: make(chan struct{}),
|
||||||
|
|
@ -187,7 +184,7 @@ func (d *Discovery) cleanupPeers() {
|
||||||
// Use 3x TTL as the expiration window.
|
// Use 3x TTL as the expiration window.
|
||||||
expiration := d.cfg.DiscoveryTime.Duration * 3
|
expiration := d.cfg.DiscoveryTime.Duration * 3
|
||||||
if expiration == 0 {
|
if expiration == 0 {
|
||||||
expiration = time.Second
|
expiration = 30 * time.Second
|
||||||
}
|
}
|
||||||
|
|
||||||
for key, peer := range d.discovered {
|
for key, peer := range d.discovered {
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue