storj/pkg/inspector/config.go
2019-01-02 12:45:47 +02:00

65 lines
1.4 KiB
Go

// Copyright (C) 2018 Storj Labs, Inc.
// See LICENSE for copying information.
package inspector
import (
"context"
"github.com/zeebo/errs"
"go.uber.org/zap"
monkit "gopkg.in/spacemonkeygo/monkit.v2"
"storj.io/storj/pkg/kademlia"
"storj.io/storj/pkg/overlay"
"storj.io/storj/pkg/pb"
"storj.io/storj/pkg/provider"
"storj.io/storj/pkg/statdb"
)
var (
mon = monkit.Package()
// Error is the main inspector error class for this package
Error = errs.Class("inspector server error:")
)
// Config is passed to CaptPlanet for bootup and configuration
type Config struct {
Enabled bool `help:"enable or disable the inspector" default:"true"`
}
// Run starts up the server and loads configs
func (c Config) Run(ctx context.Context, server *provider.Provider) (err error) {
defer mon.Task()(&ctx)(&err)
kad := kademlia.LoadFromContext(ctx)
if kad == nil {
return Error.New("programmer error: kademlia responsibility unstarted")
}
ol := overlay.LoadFromContext(ctx)
if ol == nil {
return Error.New("programmer error: overlay responsibility unstarted")
}
sdb, ok := ctx.Value("masterdb").(interface {
StatDB() statdb.DB
})
if !ok {
return Error.New("unable to get master db instance")
}
srv := &Server{
dht: kad,
identity: server.Identity(),
cache: ol,
statdb: sdb.StatDB(),
logger: zap.L(),
metrics: monkit.Default,
}
pb.RegisterInspectorServer(server.GRPC(), srv)
return server.Run(ctx)
}