frontend: implement GetLatestBlock, GetBlock; refactoring

2018-12-11 02:31:23 -05:00
parent 80b063fe8e
commit f6def7cda8
4 changed files with 166 additions and 81 deletions
--- a/cmd/server/main.go
+++ b/cmd/server/main.go
@@ -0,0 +1,123 @@
+package main
+
+import (
+	"flag"
+	"net"
+	"os"
+	"os/signal"
+	"syscall"
+
+	"github.com/gtank/ctxd/frontend"
+	"github.com/gtank/ctxd/rpc"
+	"github.com/sirupsen/logrus"
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/credentials"
+	"google.golang.org/grpc/reflection"
+)
+
+var log *logrus.Entry
+var logger = logrus.New()
+
+func init() {
+	logger.SetFormatter(&logrus.TextFormatter{
+		//DisableColors:          true,
+		FullTimestamp:          true,
+		DisableLevelTruncation: true,
+	})
+
+	log = logger.WithFields(logrus.Fields{
+		"app": "frontend-grpc",
+	})
+}
+
+type Options struct {
+	bindAddr    string `json:"bind_address,omitempty"`
+	dbPath      string `json:"db_path"`
+	tlsCertPath string `json:"tls_cert_path,omitempty"`
+	tlsKeyPath  string `json:"tls_cert_key,omitempty"`
+	logLevel    uint64 `json:"log_level,omitempty"`
+	logPath     string `json:"log_file,omitempty"`
+}
+
+func main() {
+	opts := &Options{}
+	flag.StringVar(&opts.bindAddr, "bind-addr", "127.0.0.1:9067", "the address to listen on")
+	flag.StringVar(&opts.dbPath, "db-path", "", "the path to a sqlite database file")
+	flag.StringVar(&opts.tlsCertPath, "tls-cert", "", "the path to a TLS certificate (optional)")
+	flag.StringVar(&opts.tlsKeyPath, "tls-key", "", "the path to a TLS key file (optional)")
+	flag.Uint64Var(&opts.logLevel, "log-level", uint64(logrus.InfoLevel), "log level (logrus 1-7)")
+	// TODO prod logging flag.StringVar(&opts.logPath, "log-file", "", "log file to write to")
+	// TODO prod metrics
+	// TODO support config from file and env vars
+	flag.Parse()
+
+	if opts.dbPath == "" {
+		flag.Usage()
+		os.Exit(1)
+	}
+
+	logger.SetLevel(logrus.Level(opts.logLevel))
+
+	// gRPC initialization
+	var server *grpc.Server
+
+	if opts.tlsCertPath != "" && opts.tlsKeyPath != "" {
+		transportCreds, err := credentials.NewServerTLSFromFile(opts.tlsCertPath, opts.tlsKeyPath)
+		if err != nil {
+			log.WithFields(logrus.Fields{
+				"cert_file": opts.tlsCertPath,
+				"key_path":  opts.tlsKeyPath,
+				"error":     err,
+			}).Fatal("couldn't load TLS credentials")
+		}
+		server = grpc.NewServer(grpc.Creds(transportCreds), frontend.LoggingInterceptor())
+	} else {
+		server = grpc.NewServer(frontend.LoggingInterceptor())
+	}
+
+	// Enable reflection for debugging
+	if opts.logLevel >= uint64(logrus.WarnLevel) {
+		reflection.Register(server)
+	}
+
+	// Compact transaction service initialization
+	service, err := frontend.NewSQLiteStreamer(opts.dbPath)
+	if err != nil {
+		log.WithFields(logrus.Fields{
+			"db_path": opts.dbPath,
+			"error":   err,
+		}).Fatal("couldn't create SQL streamer")
+	}
+
+	// Register service
+	rpc.RegisterCompactTxStreamerServer(server, service)
+
+	// Start listening
+	listener, err := net.Listen("tcp", opts.bindAddr)
+	if err != nil {
+		log.WithFields(logrus.Fields{
+			"bind_addr": opts.bindAddr,
+			"error":     err,
+		}).Fatal("couldn't create listener")
+	}
+
+	// Signal handler for graceful stops
+	signals := make(chan os.Signal, 1)
+	signal.Notify(signals, syscall.SIGINT, syscall.SIGTERM)
+	go func() {
+		s := <-signals
+		log.WithFields(logrus.Fields{
+			"signal": s.String(),
+		}).Info("caught signal, stopping gRPC server")
+		server.GracefulStop()
+	}()
+
+	log.Infof("Starting gRPC server on %s", opts.bindAddr)
+
+	err = server.Serve(listener)
+	if err != nil {
+		log.WithFields(logrus.Fields{
+			"error": err,
+		}).Fatal("gRPC server exited")
+	}
+}