362f447d9f
* cmd/statreceiver: lua-scriptable stat receiver Change-Id: I3ce0fe3f1ef4b1f4f27eed90bac0e91cfecf22d7 * some updates Change-Id: I7c3485adcda1278fce01ae077b4761b3ddb9fb7a * more comments Change-Id: I0bb22993cd934c3d40fc1da80d07e49e686b80dd * linter fixes Change-Id: Ied014304ecb9aadcf00a6b66ad28f856a428d150 * catch errors Change-Id: I6e1920f1fd941e66199b30bc427285c19769fc70 * review feedback Change-Id: I9d4051851eab18970c5f5ddcf4ff265508e541d3 * errorgroup improvements Change-Id: I4699dda3022f0485fbb50c9dafe692d3921734ff * too tricky the previous thing was better for memory with lots of errors at a time but https://play.golang.org/p/RweTMRjoSCt is too much of a foot gun Change-Id: I23f0b3d77dd4288fcc20b3756a7110359576bf44
89 lines
2.1 KiB
Go
89 lines
2.1 KiB
Go
// Copyright (C) 2018 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
"storj.io/storj/pkg/cfgstruct"
|
|
"storj.io/storj/pkg/luacfg"
|
|
"storj.io/storj/pkg/process"
|
|
"storj.io/storj/pkg/utils"
|
|
)
|
|
|
|
// Config is the set of configuration values we care about
|
|
var Config struct {
|
|
Input string `default:"" help:"path to configuration file"`
|
|
}
|
|
|
|
const defaultConfDir = "$HOME/.storj/statreceiver"
|
|
|
|
func main() {
|
|
cmd := &cobra.Command{
|
|
Use: os.Args[0],
|
|
Short: "stat receiving",
|
|
RunE: Main,
|
|
}
|
|
cfgstruct.Bind(cmd.Flags(), &Config, 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:
|
|
fh, err := os.Open(Config.Input)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer utils.LogClose(fh)
|
|
input = fh
|
|
}
|
|
|
|
s := luacfg.NewScope()
|
|
err := utils.CombineErrors(
|
|
s.RegisterVal("deliver", Deliver),
|
|
s.RegisterVal("filein", NewFileSource),
|
|
s.RegisterVal("fileout", NewFileDest),
|
|
s.RegisterVal("udpin", NewUDPSource),
|
|
s.RegisterVal("udpout", NewUDPDest),
|
|
s.RegisterVal("parse", NewParser),
|
|
s.RegisterVal("print", NewPrinter),
|
|
s.RegisterVal("pcopy", NewPacketCopier),
|
|
s.RegisterVal("mcopy", NewMetricCopier),
|
|
s.RegisterVal("pbuf", NewPacketBuffer),
|
|
s.RegisterVal("mbuf", NewMetricBuffer),
|
|
s.RegisterVal("packetfilter", NewPacketFilter),
|
|
s.RegisterVal("appfilter", NewApplicationFilter),
|
|
s.RegisterVal("instfilter", NewInstanceFilter),
|
|
s.RegisterVal("keyfilter", NewKeyFilter),
|
|
s.RegisterVal("sanitize", NewSanitizer),
|
|
s.RegisterVal("graphite", NewGraphiteDest),
|
|
s.RegisterVal("db", NewDBDest),
|
|
s.RegisterVal("pbufprep", NewPacketBufPrep),
|
|
s.RegisterVal("mbufprep", NewMetricBufPrep),
|
|
)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = s.Run(input)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
select {}
|
|
}
|