mirror of
https://github.com/httprunner/httprunner.git
synced 2026-05-12 02:21:29 +08:00
82 lines
3.3 KiB
Go
82 lines
3.3 KiB
Go
package cmd
|
|
|
|
import (
|
|
"os"
|
|
"runtime"
|
|
"strings"
|
|
|
|
"github.com/rs/zerolog"
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/httprunner/httprunner/v4/hrp/internal/version"
|
|
)
|
|
|
|
// rootCmd represents the base command when called without any subcommands
|
|
var rootCmd = &cobra.Command{
|
|
Use: "hrp",
|
|
Short: "Next-Generation API Testing Solution.",
|
|
Long: `
|
|
██╗ ██╗████████╗████████╗██████╗ ██████╗ ██╗ ██╗███╗ ██╗███╗ ██╗███████╗██████╗
|
|
██║ ██║╚══██╔══╝╚══██╔══╝██╔══██╗██╔══██╗██║ ██║████╗ ██║████╗ ██║██╔════╝██╔══██╗
|
|
███████║ ██║ ██║ ██████╔╝██████╔╝██║ ██║██╔██╗ ██║██╔██╗ ██║█████╗ ██████╔╝
|
|
██╔══██║ ██║ ██║ ██╔═══╝ ██╔══██╗██║ ██║██║╚██╗██║██║╚██╗██║██╔══╝ ██╔══██╗
|
|
██║ ██║ ██║ ██║ ██║ ██║ ██║╚██████╔╝██║ ╚████║██║ ╚████║███████╗██║ ██║
|
|
╚═╝ ╚═╝ ╚═╝ ╚═╝ ╚═╝ ╚═╝ ╚═╝ ╚═════╝ ╚═╝ ╚═══╝╚═╝ ╚═══╝╚══════╝╚═╝ ╚═╝
|
|
|
|
HttpRunner is an open source API testing tool that supports HTTP(S)/HTTP2/WebSocket/RPC
|
|
network protocols, covering API testing, performance testing and digital experience
|
|
monitoring (DEM) test types. Enjoy! ✨ 🚀 ✨
|
|
|
|
License: Apache-2.0
|
|
Website: https://httprunner.com
|
|
Github: https://github.com/httprunner/httprunner
|
|
Copyright 2017 debugtalk`,
|
|
PersistentPreRun: func(cmd *cobra.Command, args []string) {
|
|
noColor := false
|
|
if runtime.GOOS == "windows" {
|
|
noColor = true
|
|
}
|
|
if !logJSON {
|
|
log.Logger = zerolog.New(zerolog.ConsoleWriter{NoColor: noColor, Out: os.Stderr}).With().Timestamp().Logger()
|
|
log.Info().Msg("Set log to color console other than JSON format.")
|
|
}
|
|
},
|
|
Version: version.VERSION,
|
|
}
|
|
|
|
var (
|
|
logLevel string
|
|
logJSON bool
|
|
)
|
|
|
|
// Execute adds all child commands to the root command and sets flags appropriately.
|
|
// This is called by main.main(). It only needs to happen once to the rootCmd.
|
|
func Execute() {
|
|
rootCmd.PersistentFlags().StringVarP(&logLevel, "log-level", "l", "INFO", "set log level")
|
|
rootCmd.PersistentFlags().BoolVar(&logJSON, "log-json", false, "set log to json format")
|
|
|
|
if err := rootCmd.Execute(); err != nil {
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func setLogLevel(level string) {
|
|
level = strings.ToUpper(level)
|
|
log.Info().Str("level", level).Msg("Set log level")
|
|
switch level {
|
|
case "DEBUG":
|
|
zerolog.SetGlobalLevel(zerolog.DebugLevel)
|
|
case "INFO":
|
|
zerolog.SetGlobalLevel(zerolog.InfoLevel)
|
|
case "WARN":
|
|
zerolog.SetGlobalLevel(zerolog.WarnLevel)
|
|
case "ERROR":
|
|
zerolog.SetGlobalLevel(zerolog.ErrorLevel)
|
|
case "FATAL":
|
|
zerolog.SetGlobalLevel(zerolog.FatalLevel)
|
|
case "PANIC":
|
|
zerolog.SetGlobalLevel(zerolog.PanicLevel)
|
|
}
|
|
}
|