Dev simd (#1146)

[bytom/bytom.git] / cmd / bytomd / commands / run_node.go

diff --git a/cmd/bytomd/commands/run_node.go b/cmd/bytomd/commands/run_node.go
index 60c0d1f..049cd88 100644 (file)
--- a/cmd/bytomd/commands/run_node.go
+++ b/cmd/bytomd/commands/run_node.go
@@ -7,6 +7,7 @@ import (
        "github.com/spf13/cobra"
 
        "github.com/bytom/node"
+       "strings"
 )
 
 var runNodeCmd = &cobra.Command{
@@ -19,13 +20,19 @@ func init() {
        runNodeCmd.Flags().String("prof_laddr", config.ProfListenAddress, "Use http to profile bytomd programs")
        runNodeCmd.Flags().Bool("mining", config.Mining, "Enable mining")
 
+       runNodeCmd.Flags().Bool("simd.enable", config.Simd.Enable, "Enable SIMD mechan for tensority")
+
        runNodeCmd.Flags().Bool("auth.disable", config.Auth.Disable, "Disable rpc access authenticate")
 
        runNodeCmd.Flags().Bool("wallet.disable", config.Wallet.Disable, "Disable wallet")
-
+       runNodeCmd.Flags().Bool("wallet.rescan", config.Wallet.Rescan, "Rescan wallet")
+       runNodeCmd.Flags().Bool("vault_mode", config.VaultMode, "Run in the offline enviroment")
        runNodeCmd.Flags().Bool("web.closed", config.Web.Closed, "Lanch web browser or not")
        runNodeCmd.Flags().String("chain_id", config.ChainID, "Select network type")
 
+       // log level
+       runNodeCmd.Flags().String("log_level", config.LogLevel, "Select log level(debug, info, warn, error or fatal")
+
        // p2p flags
        runNodeCmd.Flags().String("p2p.laddr", config.P2P.ListenAddress, "Node listen address. (0.0.0.0:0 means any interface, any port)")
        runNodeCmd.Flags().String("p2p.seeds", config.P2P.Seeds, "Comma delimited host:port seed nodes")
@@ -35,16 +42,39 @@ func init() {
        runNodeCmd.Flags().Int("p2p.handshake_timeout", config.P2P.HandshakeTimeout, "Set handshake timeout")
        runNodeCmd.Flags().Int("p2p.dial_timeout", config.P2P.DialTimeout, "Set dial timeout")
 
+       // log flags
+       runNodeCmd.Flags().String("log_file", config.LogFile, "Log output file")
+
        RootCmd.AddCommand(runNodeCmd)
 }
 
+func getLogLevel(level string) log.Level {
+       switch strings.ToLower(level) {
+       case "debug":
+               return log.DebugLevel
+       case "info":
+               return log.InfoLevel
+       case "warn":
+               return log.WarnLevel
+       case "error":
+               return log.ErrorLevel
+       case "fatal":
+               return log.FatalLevel
+       default:
+               return log.InfoLevel
+       }
+}
+
 func runNode(cmd *cobra.Command, args []string) error {
+       // Set log level by config.LogLevel
+       log.SetLevel(getLogLevel(config.LogLevel))
+
        // Create & start node
        n := node.NewNode(config)
        if _, err := n.Start(); err != nil {
                return fmt.Errorf("Failed to start node: %v", err)
        } else {
-               log.WithField("nodeInfo", n.SyncManager().Switch().NodeInfo()).Info("Started node")
+               log.Info("Start node ", n.SyncManager().NodeInfo())
        }
 
        // Trap signal, run forever.