Skip to content

Commit

Permalink
Fix style issues
Browse files Browse the repository at this point in the history
License: MIT
Signed-off-by: Michael Avila <davidmichaelavila@gmail.com>
  • Loading branch information
michaelavila committed Apr 12, 2019
1 parent 0bf621c commit 6c78140
Show file tree
Hide file tree
Showing 3 changed files with 14 additions and 4 deletions.
1 change: 1 addition & 0 deletions core/builder.go
Original file line number Diff line number Diff line change
Expand Up @@ -288,6 +288,7 @@ func setupNode(ctx context.Context, n *IpfsNode, cfg *BuildCfg) error {
tracker := provider.NewTracker(n.Repo.Datastore())
n.Provider = provider.NewProviderSystem(
provider.NewProvider(ctx, queueP, tracker, n.Routing),
// TODO: configure time.Minute and time.Hour*12
provider.NewReprovider(ctx, queueR, tracker, time.Minute, time.Hour*12, n.Blockstore, n.Routing),
)

Expand Down
2 changes: 1 addition & 1 deletion provider/reprovider.go
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,7 @@ func (rp *Reprovider) Run() {
}

// Close stops thhe reprovider
func (rp *Reprovider) Close() error {
func (rp *Reprovider) Close() error {
return rp.queue.Close()
}

Expand Down
15 changes: 12 additions & 3 deletions provider/system.go
Original file line number Diff line number Diff line change
Expand Up @@ -5,23 +5,28 @@ import (
"github.com/ipfs/go-cid"
)

// ProviderSystem bundles together provider and reprovider behavior
// into one system
type ProviderSystem struct {
provider Provider
provider Provider
reprovider *Reprovider
}

// NewProviderSystem creates a new ProviderSystem
func NewProviderSystem(p Provider, r *Reprovider) *ProviderSystem {
return &ProviderSystem{
provider: p,
provider: p,
reprovider: r,
}
}

// Run starts the provider system loops
func (ps *ProviderSystem) Run() {
ps.provider.Run()
ps.reprovider.Run()
}

// Close stops the provider system loops
func (ps *ProviderSystem) Close() error {
var errs []error

Expand All @@ -39,14 +44,18 @@ func (ps *ProviderSystem) Close() error {
return nil
}

// Provide a cid by announcing it to the network
func (ps *ProviderSystem) Provide(cid cid.Cid) {
ps.provider.Provide(cid)
}

// Tracking returns all cids that are currently being tracked and reprovided
// by the provider system.
func (ps *ProviderSystem) Tracking() (<-chan cid.Cid, error) {
return ps.provider.Tracking()
}

// Reprovide triggers a reprovide
func (ps *ProviderSystem) Reprovide(ctx context.Context) error {
return ps.reprovider.Trigger(ctx)
return ps.reprovider.Trigger(ctx)
}

0 comments on commit 6c78140

Please sign in to comment.