e74cac52ab
* change BindSetup to be an option to Bind * add process.Bind to allow composite structures * hack fix for noprefix flags * used tagged version of structs Before this PR, some flags were created by calling `cfgstruct.Bind` and having their fields create a flag. Once the flags were parsed, `viper` was used to acquire all the values from them and config files, and the fields in the struct were set through the flag interface. This doesn't work for slices of things on config structs very well, since it can only set strings, and for a string slice, it turns out that the implementation in `pflag` appends an entry rather than setting it. This changes three things: 1. Only have a `Bind` call instead of `Bind` and `BindSetup`, and make `BindSetup` an option instead. 2. Add a `process.Bind` call that takes in a `*cobra.Cmd`, binds the struct to the command's flags, and keeps track of that struct in a global map keyed by the command. 3. Use `viper` to get the values and load them into the bound configuration structs instead of using the flags to propagate the changes. In this way, we can support whatever rich configuration we want in the config yaml files, while still getting command like flags when important.
97 lines
2.4 KiB
Go
97 lines
2.4 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/zeebo/errs"
|
|
|
|
"storj.io/storj/cmd/statreceiver/luacfg"
|
|
"storj.io/storj/internal/fpath"
|
|
"storj.io/storj/pkg/cfgstruct"
|
|
"storj.io/storj/pkg/process"
|
|
)
|
|
|
|
// Config is the set of configuration values we care about
|
|
var Config struct {
|
|
Input string `default:"" help:"path to configuration file"`
|
|
}
|
|
|
|
func main() {
|
|
defaultConfDir := fpath.ApplicationDir("storj", "statreceiver")
|
|
|
|
cmd := &cobra.Command{
|
|
Use: os.Args[0],
|
|
Short: "stat receiving",
|
|
RunE: Main,
|
|
}
|
|
defaults := cfgstruct.DefaultsFlag(cmd)
|
|
process.Bind(cmd, &Config, defaults, cfgstruct.ConfDir(defaultConfDir))
|
|
cmd.Flags().String("config", filepath.Join(defaultConfDir, "config.yaml"), "path to configuration")
|
|
process.Exec(cmd)
|
|
}
|
|
|
|
// Main is the real main method
|
|
func Main(cmd *cobra.Command, args []string) error {
|
|
var input io.Reader
|
|
switch Config.Input {
|
|
case "":
|
|
return fmt.Errorf("--input path to script is required")
|
|
case "stdin":
|
|
input = os.Stdin
|
|
default:
|
|
inputFile, err := os.Open(Config.Input)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer func() {
|
|
if err := inputFile.Close(); err != nil {
|
|
log.Printf("Failed to close input: %scope", err)
|
|
}
|
|
}()
|
|
|
|
input = inputFile
|
|
}
|
|
|
|
scope := luacfg.NewScope()
|
|
err := errs.Combine(
|
|
scope.RegisterVal("deliver", Deliver),
|
|
scope.RegisterVal("filein", NewFileSource),
|
|
scope.RegisterVal("fileout", NewFileDest),
|
|
scope.RegisterVal("udpin", NewUDPSource),
|
|
scope.RegisterVal("udpout", NewUDPDest),
|
|
scope.RegisterVal("parse", NewParser),
|
|
scope.RegisterVal("print", NewPrinter),
|
|
scope.RegisterVal("pcopy", NewPacketCopier),
|
|
scope.RegisterVal("mcopy", NewMetricCopier),
|
|
scope.RegisterVal("pbuf", NewPacketBuffer),
|
|
scope.RegisterVal("mbuf", NewMetricBuffer),
|
|
scope.RegisterVal("packetfilter", NewPacketFilter),
|
|
scope.RegisterVal("appfilter", NewApplicationFilter),
|
|
scope.RegisterVal("instfilter", NewInstanceFilter),
|
|
scope.RegisterVal("keyfilter", NewKeyFilter),
|
|
scope.RegisterVal("sanitize", NewSanitizer),
|
|
scope.RegisterVal("graphite", NewGraphiteDest),
|
|
scope.RegisterVal("db", NewDBDest),
|
|
scope.RegisterVal("pbufprep", NewPacketBufPrep),
|
|
scope.RegisterVal("mbufprep", NewMetricBufPrep),
|
|
)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = scope.Run(input)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
select {}
|
|
}
|