add debug level switch (devstream-io#175)

* add opt for debug swith * add opt for debug swith * update for pr
keepauto · Feb 8, 2022 · b253e6d · b253e6d
1 parent 928fe8b
commit b253e6d
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 4 deletions.
diff --git a/cmd/devstream/apply.go b/cmd/devstream/apply.go
@@ -20,7 +20,6 @@ DevStream will generate and execute a new plan based on the config file and the
 
 func applyCMDFunc(cmd *cobra.Command, args []string) {
  log.Info("Apply started.")
-
  if err := pluginengine.Apply(configFile, continueDirectly); err != nil {
  log.Errorf("Apply error: %s.", err)
  os.Exit(1)

diff --git a/cmd/devstream/main.go b/cmd/devstream/main.go
@@ -3,18 +3,19 @@ package main
 import (
  "strings"
 
- "github.com/merico-dev/stream/internal/pkg/log"
-
+ "github.com/sirupsen/logrus"
  "github.com/spf13/cobra"
  "github.com/spf13/viper"
 
+ "github.com/merico-dev/stream/internal/pkg/log"
  "github.com/merico-dev/stream/internal/pkg/pluginengine"
 )
 
 var (
  configFile string
  pluginDir string
  continueDirectly bool
+ isDebug bool
 
  rootCMD = &cobra.Command{
  Use: "dtm",
@@ -29,6 +30,9 @@ var (
 # # # # # # # # # # # # # # # 
 ###### ###### ## ##### # # # ###### # # # # 
 `,
+ PersistentPreRun: func(cmd *cobra.Command, args []string) {
+ initLog()
+ },
  }
 )
 
@@ -38,7 +42,7 @@ func init() {
  rootCMD.PersistentFlags().StringVarP(&configFile, "config-file", "f", "config.yaml", "config file")
  rootCMD.PersistentFlags().StringVarP(&pluginDir, "plugin-dir", "p", pluginengine.DefaultPluginDir, "plugins directory")
  rootCMD.PersistentFlags().BoolVarP(&continueDirectly, "yes", "y", false, "apply/delete directly without confirmation")
-
+ rootCMD.PersistentFlags().BoolVarP(&isDebug, "debug", "d", false, "debug level log")
  rootCMD.AddCommand(versionCMD)
  rootCMD.AddCommand(initCMD)
  rootCMD.AddCommand(applyCMD)
@@ -68,6 +72,15 @@ func initConfig() {
  }
 }
 
+func initLog() {
+ if isDebug {
+ logrus.SetLevel(logrus.DebugLevel)
+ } else {
+ logrus.SetLevel(logrus.InfoLevel)
+ }
+ log.Info("Log level is: ", logrus.GetLevel())
+}
+
 func main() {
  err := rootCMD.Execute()
  if err != nil {