c9d900fe4d
* Process for updating node stats in kademlia * Mutex lock * Rename and set up the refresher * Wrap errors * Lock should be around the if * Address comments; Build updateSelf funciton in Kademlia Routing Table * Added test * Address comments
95 lines
2.3 KiB
Go
95 lines
2.3 KiB
Go
// Copyright (C) 2018 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package psserver
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"path/filepath"
|
|
"time"
|
|
|
|
"go.uber.org/zap"
|
|
"golang.org/x/net/context"
|
|
monkit "gopkg.in/spacemonkeygo/monkit.v2"
|
|
|
|
"storj.io/storj/pkg/kademlia"
|
|
"storj.io/storj/pkg/pb"
|
|
"storj.io/storj/pkg/piecestore/psserver/agreementsender"
|
|
"storj.io/storj/pkg/piecestore/psserver/psdb"
|
|
"storj.io/storj/pkg/provider"
|
|
)
|
|
|
|
var (
|
|
mon = monkit.Package()
|
|
)
|
|
|
|
// Config contains everything necessary for a server
|
|
type Config struct {
|
|
Path string `help:"path to store data in" default:"$CONFDIR"`
|
|
AllocatedDiskSpace int64 `help:"total allocated disk space, default(1GB)" default:"1073741824"`
|
|
AllocatedBandwidth int64 `help:"total allocated bandwidth, default(100GB)" default:"107374182400"`
|
|
KBucketRefreshInterval time.Duration `help:"how frequently checker should audit segments" default:"3600s"`
|
|
}
|
|
|
|
// Run implements provider.Responsibility
|
|
func (c Config) Run(ctx context.Context, server *provider.Provider) (err error) {
|
|
defer mon.Task()(&ctx)(&err)
|
|
|
|
ctx, cancel := context.WithCancel(ctx)
|
|
|
|
db, err := psdb.Open(ctx, filepath.Join(c.Path, "piece-store-data"), filepath.Join(c.Path, "piecestore.db"))
|
|
if err != nil {
|
|
return ServerError.Wrap(err)
|
|
}
|
|
|
|
s, err := NewEndpoint(zap.L(), c, db, server.Identity().Key)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
pb.RegisterPieceStoreRoutesServer(server.GRPC(), s)
|
|
|
|
// Run the agreement sender process
|
|
asProcess, err := agreementsender.Initialize(s.DB, server.Identity())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
go func() {
|
|
if err := asProcess.Run(ctx); err != nil {
|
|
cancel()
|
|
}
|
|
}()
|
|
|
|
// Run the kademlia bucket refresher process
|
|
k := kademlia.LoadFromContext(ctx)
|
|
if k == nil {
|
|
return ServerError.New("Failed to load Kademlia from context")
|
|
}
|
|
|
|
rt, err := k.GetRoutingTable(ctx)
|
|
if err != nil {
|
|
return ServerError.Wrap(err)
|
|
}
|
|
|
|
krt, ok := rt.(*kademlia.RoutingTable)
|
|
if !ok {
|
|
return ServerError.New("Could not convert dht.RoutingTable to *kademlia.RoutingTable")
|
|
}
|
|
|
|
refreshProcess := newService(zap.L(), c.KBucketRefreshInterval, krt, s)
|
|
|
|
go func() {
|
|
if err := refreshProcess.Run(ctx); err != nil {
|
|
cancel()
|
|
}
|
|
}()
|
|
|
|
defer func() {
|
|
log.Fatal(s.Stop(ctx))
|
|
}()
|
|
|
|
s.log.Info("Started Node", zap.String("ID", fmt.Sprint(server.Identity().ID)))
|
|
return server.Run(ctx)
|
|
}
|