Skip to content

Commit

Permalink
contractor: cleanup PR
Browse files Browse the repository at this point in the history
  • Loading branch information
peterjan committed Oct 4, 2023
1 parent 113b360 commit 102393e
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 10 deletions.
13 changes: 7 additions & 6 deletions autopilot/contractor.go
Original file line number Diff line number Diff line change
Expand Up @@ -126,11 +126,10 @@ type (
)

func newContractor(ap *Autopilot, revisionSubmissionBuffer uint64, revisionBroadcastInterval time.Duration) *contractor {
logger := ap.logger.Named("contractor")
return &contractor{
ap: ap,
resolver: newIPResolver(resolverLookupTimeout, logger),
logger: logger,
resolver: newIPResolver(resolverLookupTimeout, ap.logger.Named("resolver")),
logger: ap.logger.Named("contractor"),
revisionBroadcastInterval: revisionBroadcastInterval,
revisionLastBroadcast: make(map[types.FileContractID]time.Time),
revisionSubmissionBuffer: revisionSubmissionBuffer,
Expand Down Expand Up @@ -818,9 +817,11 @@ func (c *contractor) runContractFormations(ctx context.Context, w Worker, hosts

// prepare an IP filter that contains all used hosts
ipFilter := c.newIPFilter()
for _, h := range hosts {
if _, used := usedHosts[h.PublicKey]; used {
_ = ipFilter.IsRedundantIP(h.NetAddress, h.PublicKey)
if shouldFilter {
for _, h := range hosts {
if _, used := usedHosts[h.PublicKey]; used {
_ = ipFilter.IsRedundantIP(h.NetAddress, h.PublicKey)
}
}
}

Expand Down
9 changes: 5 additions & 4 deletions autopilot/ipfilter.go
Original file line number Diff line number Diff line change
Expand Up @@ -37,18 +37,19 @@ var (
type (
ipFilter struct {
subnetToHostKey map[string]string
resolver *ipResolver

logger *zap.SugaredLogger
resolver *ipResolver
logger *zap.SugaredLogger
}
)

func (c *contractor) newIPFilter() *ipFilter {
c.resolver.pruneCache()
return &ipFilter{
subnetToHostKey: make(map[string]string),
resolver: c.resolver,
logger: c.logger,

resolver: c.resolver,
logger: c.logger,
}
}

Expand Down

0 comments on commit 102393e

Please sign in to comment.