1a06c2b0c7
* adds proto files for netstate crud * moves netstate grpc client lib into pkg/netstate where grpc netstate service is defined * starts adding grpc client and server tests * moves creation of grpc server into cmd/netstate/main.go, removes pkg/netstate/service.go, adds more client testing * changed all 'Path' and 'Value' fields from strings to bytes, updated tests * changes Get and Delete in proto file to receive 'requests' instead of 'file paths', adds tests for Get, List, and Delete * changes netstate-routes to get 'fileValue' bytes not 'fileInfo' * adds example rpc client in 'examples' and adds more specific debug logs * adds readmes for netstate rpc services and updates netstate-routes
65 lines
1.3 KiB
Go
65 lines
1.3 KiB
Go
// Copyright (C) 2018 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"net"
|
|
|
|
"go.uber.org/zap"
|
|
"google.golang.org/grpc"
|
|
|
|
"storj.io/storj/pkg/netstate"
|
|
proto "storj.io/storj/protos/netstate"
|
|
"storj.io/storj/storage/boltdb"
|
|
)
|
|
|
|
var (
|
|
port int
|
|
dbPath string
|
|
prod bool
|
|
)
|
|
|
|
func initializeFlags() {
|
|
flag.IntVar(&port, "port", 8080, "port")
|
|
flag.StringVar(&dbPath, "db", "netstate.db", "db path")
|
|
flag.BoolVar(&prod, "prod", false, "type of environment where this service runs")
|
|
flag.Parse()
|
|
}
|
|
|
|
func main() {
|
|
initializeFlags()
|
|
|
|
// No err here because no vars passed into NewDevelopment().
|
|
// The default won't return an error, but if args are passed in,
|
|
// then there will need to be error handling.
|
|
logger, _ := zap.NewDevelopment()
|
|
if prod {
|
|
logger, _ = zap.NewProduction()
|
|
}
|
|
defer logger.Sync()
|
|
|
|
bdb, err := boltdb.New(logger, dbPath)
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer bdb.Close()
|
|
|
|
// start grpc server
|
|
lis, err := net.Listen("tcp", fmt.Sprintf(":%d", port))
|
|
if err != nil {
|
|
logger.Fatal("failed to listen", zap.Error(err))
|
|
}
|
|
|
|
grpcServer := grpc.NewServer()
|
|
proto.RegisterNetStateServer(grpcServer, netstate.NewServer(bdb, logger))
|
|
|
|
defer grpcServer.GracefulStop()
|
|
err = grpcServer.Serve(lis)
|
|
if err != nil {
|
|
logger.Error("Failed to serve:", zap.Error(err))
|
|
}
|
|
}
|