Make the warnings for validating services more obvious

[govpp.git] / cmd / binapi-generator / main.go

diff --git a/cmd/binapi-generator/main.go b/cmd/binapi-generator/main.go
index 8045212..b73a699 100644 (file)
--- a/cmd/binapi-generator/main.go
+++ b/cmd/binapi-generator/main.go
@@ -20,13 +20,13 @@ import (
        "flag"
        "fmt"
        "io/ioutil"
-       "log"
        "os"
        "os/exec"
        "path/filepath"
        "strings"
 
        "github.com/bennyscetbun/jsongo"
+       "github.com/sirupsen/logrus"
 )
 
 var (
@@ -38,15 +38,26 @@ var (
        continueOnError = flag.Bool("continue-onerror", false, "Wheter to continue with next file on error.")
 )
 
+func init() {
+       flag.Parse()
+       if *debug {
+               logrus.SetLevel(logrus.DebugLevel)
+       }
+}
+
 func logf(f string, v ...interface{}) {
        if *debug {
-               log.Printf(f, v...)
+               logrus.Debugf(f, v...)
        }
 }
 
-func main() {
-       flag.Parse()
+var log = logrus.Logger{
+       Level:     logrus.InfoLevel,
+       Formatter: &logrus.TextFormatter{},
+       Out:       os.Stdout,
+}
 
+func main() {
        if *inputFile == "" && *inputDir == "" {
                fmt.Fprintln(os.Stderr, "ERROR: input-file or input-dir must be specified")
                os.Exit(1)
@@ -143,7 +154,7 @@ func generateFromFile(inputFile, outputDir string) error {
        // count number of lines in generated output file
        cmd = exec.Command("wc", "-l", ctx.outputFile)
        if output, err := cmd.CombinedOutput(); err != nil {
-               log.Printf("wc command failed: %v\n%s", err, string(output))
+               log.Warnf("wc command failed: %v\n%s", err, string(output))
        } else {
                logf("generated lines: %s", output)
        }