1ebd66d880
* wip post-demos cleanup
* wip
* wip
* wip
* wip
* wip
* wip
* wip
* wip
* wip
* wip
* Reorganize:
storage
├── common
│ └── common.go `package storage`
├── boltdb
│ └── ...
└── redis
└── ...
storage
├── common.go `package storage`
├── boltdb
│ └── ...
└── redis
└── ...
storage
├── common
│ ├── common.go `package storage`
│ └── testing.go `package storage` <--
└── ...
internal
├── app
│ └── cli
├── pkg
│ └── readcloser
└── test
└── util.go `package test` <--
* remove comment
* add and use goimports
* add test types & fix some lint errors
* better typing
* fixing linter issues/comments
* goimports
* goimports
* more linter fixes; replace panic with NoError assertions in tests
* fix typo/more linter errors
* moar better linter fixes
* even moar better linter fixes
* linter
* add localPort back
* fixing exports, imports, and add comments
78 lines
1.4 KiB
Go
78 lines
1.4 KiB
Go
// Copyright (C) 2018 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"fmt"
|
|
"net"
|
|
|
|
"github.com/spf13/viper"
|
|
"go.uber.org/zap"
|
|
"google.golang.org/grpc"
|
|
monkit "gopkg.in/spacemonkeygo/monkit.v2"
|
|
|
|
"storj.io/storj/pkg/netstate"
|
|
"storj.io/storj/pkg/process"
|
|
proto "storj.io/storj/protos/netstate"
|
|
"storj.io/storj/storage/boltdb"
|
|
)
|
|
|
|
var (
|
|
port = flag.Int("port", 8080, "port")
|
|
dbPath = flag.String("db", "netstate.db", "db path")
|
|
)
|
|
|
|
func (s *serv) Process(ctx context.Context) error {
|
|
bdb, err := boltdb.NewClient(s.logger, *dbPath, boltdb.PointerBucket)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer bdb.Close()
|
|
|
|
// start grpc server
|
|
lis, err := net.Listen("tcp", fmt.Sprintf(":%d", *port))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
grpcServer := grpc.NewServer()
|
|
|
|
proto.RegisterNetStateServer(grpcServer, netstate.NewServer(bdb, s.logger))
|
|
s.logger.Debug(fmt.Sprintf("server listening on port %d", *port))
|
|
|
|
defer grpcServer.GracefulStop()
|
|
return grpcServer.Serve(lis)
|
|
}
|
|
|
|
type serv struct {
|
|
logger *zap.Logger
|
|
metrics *monkit.Registry
|
|
}
|
|
|
|
func (s *serv) SetLogger(l *zap.Logger) error {
|
|
s.logger = l
|
|
return nil
|
|
}
|
|
|
|
func setEnv() error {
|
|
viper.SetEnvPrefix("API")
|
|
viper.AutomaticEnv()
|
|
return nil
|
|
}
|
|
|
|
func (s *serv) SetMetricHandler(m *monkit.Registry) error {
|
|
s.metrics = m
|
|
return nil
|
|
}
|
|
|
|
func (s *serv) InstanceID() string { return "" }
|
|
|
|
func main() {
|
|
setEnv()
|
|
process.Must(process.Main(&serv{}))
|
|
}
|