Jason Kulatunga
266c95f857
Make sure that the collector attempts to correctly communicate with webapp container, even when running in dedicated container (and triggered manually). fixes cron schedule to run daily. added instructions for dedicated containers.
124 lines
2.7 KiB
Go
124 lines
2.7 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/analogj/scrutiny/collector/pkg/collector"
|
|
"github.com/analogj/scrutiny/webapp/backend/pkg/version"
|
|
"github.com/sirupsen/logrus"
|
|
"log"
|
|
"os"
|
|
"time"
|
|
|
|
utils "github.com/analogj/go-util/utils"
|
|
"github.com/fatih/color"
|
|
"github.com/urfave/cli/v2"
|
|
)
|
|
|
|
var goos string
|
|
var goarch string
|
|
|
|
func main() {
|
|
|
|
cli.CommandHelpTemplate = `NAME:
|
|
{{.HelpName}} - {{.Usage}}
|
|
USAGE:
|
|
{{if .UsageText}}{{.UsageText}}{{else}}{{.HelpName}} {{if .ArgsUsage}}{{.ArgsUsage}}{{else}}[arguments...]{{end}}{{end}}{{if .Category}}
|
|
CATEGORY:
|
|
{{.Category}}{{end}}{{if .Description}}
|
|
DESCRIPTION:
|
|
{{.Description}}{{end}}{{if .VisibleFlags}}
|
|
OPTIONS:
|
|
{{range .VisibleFlags}}{{.}}
|
|
{{end}}{{end}}
|
|
`
|
|
|
|
app := &cli.App{
|
|
Name: "scrutiny-collector-metrics",
|
|
Usage: "smartctl data collector for scrutiny",
|
|
Version: version.VERSION,
|
|
Compiled: time.Now(),
|
|
Authors: []*cli.Author{
|
|
{
|
|
Name: "Jason Kulatunga",
|
|
Email: "jason@thesparktree.com",
|
|
},
|
|
},
|
|
Before: func(c *cli.Context) error {
|
|
|
|
collectorMetrics := "AnalogJ/scrutiny/metrics"
|
|
|
|
var versionInfo string
|
|
if len(goos) > 0 && len(goarch) > 0 {
|
|
versionInfo = fmt.Sprintf("%s.%s-%s", goos, goarch, version.VERSION)
|
|
} else {
|
|
versionInfo = fmt.Sprintf("dev-%s", version.VERSION)
|
|
}
|
|
|
|
subtitle := collectorMetrics + utils.LeftPad2Len(versionInfo, " ", 65-len(collectorMetrics))
|
|
|
|
color.New(color.FgGreen).Fprintf(c.App.Writer, fmt.Sprintf(utils.StripIndent(
|
|
`
|
|
___ ___ ____ __ __ ____ ____ _ _ _ _
|
|
/ __) / __)( _ \( )( )(_ _)(_ _)( \( )( \/ )
|
|
\__ \( (__ ) / )(__)( )( _)(_ ) ( \ /
|
|
(___/ \___)(_)\_)(______) (__) (____)(_)\_) (__)
|
|
%s
|
|
|
|
`), subtitle))
|
|
|
|
return nil
|
|
},
|
|
|
|
Commands: []*cli.Command{
|
|
{
|
|
Name: "run",
|
|
Usage: "Run the scrutiny smartctl metrics collector",
|
|
Action: func(c *cli.Context) error {
|
|
|
|
collectorLogger := logrus.WithFields(logrus.Fields{
|
|
"type": "metrics",
|
|
})
|
|
|
|
if c.Bool("debug") {
|
|
logrus.SetLevel(logrus.DebugLevel)
|
|
} else {
|
|
logrus.SetLevel(logrus.InfoLevel)
|
|
}
|
|
|
|
metricCollector, err := collector.CreateMetricsCollector(
|
|
collectorLogger,
|
|
c.String("api-endpoint"),
|
|
)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return metricCollector.Run()
|
|
},
|
|
|
|
Flags: []cli.Flag{
|
|
&cli.StringFlag{
|
|
Name: "api-endpoint",
|
|
Usage: "The api server endpoint",
|
|
Value: "http://localhost:8080",
|
|
EnvVars: []string{"SCRUTINY_API_ENDPOINT"},
|
|
},
|
|
|
|
&cli.BoolFlag{
|
|
Name: "debug",
|
|
Usage: "Enable debug logging",
|
|
EnvVars: []string{"DEBUG"},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
err := app.Run(os.Args)
|
|
if err != nil {
|
|
log.Fatal(color.HiRedString("ERROR: %v", err))
|
|
}
|
|
|
|
}
|