garionion
/
gstreamer-graphix
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add reflection to gRPC server 

only log to file if configured
This commit is contained in:
Garionion 2022-02-07 11:21:59 +01:00
parent bcdf42b277
commit aa6d0ed864
1 changed files with 13 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
main.go
View File

@ -9,6 +9,7 @@ import (
"github.com/rs/zerolog/log" "github.com/rs/zerolog/log"
"github.com/rs/zerolog/pkgerrors" "github.com/rs/zerolog/pkgerrors"
"google.golang.org/grpc" "google.golang.org/grpc"
"google.golang.org/grpc/reflection"
"net" "net"
"os" "os"
"time" "time"
@ -16,7 +17,7 @@ import (
type Config struct { type Config struct {
Outputs []string `yaml:"outputs"` Outputs []string `yaml:"outputs"`
Address string `yaml:"address" env:"ADDRESS" env-default:":3000"` Address string `yaml:"address" env:"ADDRESS" env-default:"localhost:3000"`
LogFile string `yaml:"logfile" env:"LOGFILE" env-default:"./gstreamer-graphix.log"` LogFile string `yaml:"logfile" env:"LOGFILE" env-default:"./gstreamer-graphix.log"`
} }
@ -28,19 +29,24 @@ func main() {
log.Logger = log.Output(consoleWriter).With().Timestamp().Caller().Logger() log.Logger = log.Output(consoleWriter).With().Timestamp().Caller().Logger()
if err := cleanenv.ReadConfig("config.toml", &cfg); err != nil { if err := cleanenv.ReadConfig("config.toml", &cfg); err != nil {
log.Fatal().Msgf("No configfile: ", err) log.Info().Msgf("No configfile, continuing without: ", err)
} }
fset := flag.NewFlagSet("config", flag.ContinueOnError) fset := flag.NewFlagSet("config", flag.ContinueOnError)
fset.Usage = cleanenv.FUsage(fset.Output(), &cfg, nil, fset.Usage) fset.Usage = cleanenv.FUsage(fset.Output(), &cfg, nil, fset.Usage)
fset.Parse(os.Args[1:]) fset.Parse(os.Args[1:])
l, err := os.OpenFile(cfg.LogFile, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0664) if cfg.LogFile != "" {
multi := zerolog.MultiLevelWriter(consoleWriter, l) l, _ := os.OpenFile(cfg.LogFile, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0664)
log.Logger = zerolog.New(multi).With().Timestamp().Caller().Logger() multi := zerolog.MultiLevelWriter(consoleWriter, l)
log.Logger = zerolog.New(multi).With().Timestamp().Caller().Logger()
}
zerolog.ErrorStackMarshaler = pkgerrors.MarshalStack if log.Logger.GetLevel() == zerolog.DebugLevel {
zerolog.ErrorStackMarshaler = pkgerrors.MarshalStack
}
log.Info().Msgf("listening on %s", cfg.Address)
ln, err := net.Listen("tcp", cfg.Address) ln, err := net.Listen("tcp", cfg.Address)
if err != nil { if err != nil {
log.Fatal().Err(err) log.Fatal().Err(err)
@ -51,6 +57,7 @@ func main() {
pipelineService := &api.PipelineService{Gstreamer: gst} pipelineService := &api.PipelineService{Gstreamer: gst}
g := grpc.NewServer() g := grpc.NewServer()
reflection.Register(g)
api.RegisterPipelineServiceServer(g, pipelineService) api.RegisterPipelineServiceServer(g, pipelineService)
log.Fatal().Msgf("Failed to serve: %v", g.Serve(ln)) log.Fatal().Msgf("Failed to serve: %v", g.Serve(ln))