GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-07-07 15:31:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: global set log with pretty console

Browse Source
This commit is contained in:
debugtalk
2021-10-18 10:04:45 +08:00
parent 368ed9e14b
commit 268c4fd0af
4 changed files with 17 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
hrp/cmd/boom.go
View File

@@ -19,6 +19,10 @@ var boomCmd = &cobra.Command{
Args: cobra.MinimumNArgs(1), Args: cobra.MinimumNArgs(1),
Run: func(cmd *cobra.Command, args []string) { Run: func(cmd *cobra.Command, args []string) {
hrp.SetLogLevel(logLevel) hrp.SetLogLevel(logLevel)
if !logJSON {
hrp.SetLogPretty()
}
var paths []hrp.ITestCase var paths []hrp.ITestCase
for _, arg := range args { for _, arg := range args {
paths = append(paths, &hrp.TestCasePath{Path: arg}) paths = append(paths, &hrp.TestCasePath{Path: arg})

4
hrp/cmd/har2case.go
View File

@@ -16,6 +16,10 @@ var har2caseCmd = &cobra.Command{
Args: cobra.MinimumNArgs(1), Args: cobra.MinimumNArgs(1),
RunE: func(cmd *cobra.Command, args []string) error { RunE: func(cmd *cobra.Command, args []string) error {
hrp.SetLogLevel(logLevel) hrp.SetLogLevel(logLevel)
if !logJSON {
hrp.SetLogPretty()
}
var outputFiles []string var outputFiles []string
for _, arg := range args { for _, arg := range args {
var outputPath string var outputPath string

4
hrp/cmd/run.go
View File

@@ -17,6 +17,10 @@ var runCmd = &cobra.Command{
Args: cobra.MinimumNArgs(1), Args: cobra.MinimumNArgs(1),
RunE: func(cmd *cobra.Command, args []string) error { RunE: func(cmd *cobra.Command, args []string) error {
hrp.SetLogLevel(logLevel) hrp.SetLogLevel(logLevel)
if !logJSON {
hrp.SetLogPretty()
}
var paths []hrp.ITestCase var paths []hrp.ITestCase
for _, arg := range args { for _, arg := range args {
paths = append(paths, &hrp.TestCasePath{Path: arg}) paths = append(paths, &hrp.TestCasePath{Path: arg})

9
log.go
View File

@@ -10,10 +10,6 @@ import (
var log = zlog.Logger var log = zlog.Logger
func init() {
log = log.Output(zerolog.ConsoleWriter{Out: os.Stderr})
}
func SetLogLevel(level string) { func SetLogLevel(level string) {
level = strings.ToUpper(level) level = strings.ToUpper(level)
log.Info().Msgf("Set log level to %s", level) log.Info().Msgf("Set log level to %s", level)
@@ -33,6 +29,11 @@ func SetLogLevel(level string) {
} }
} }
func SetLogPretty() {
log.Info().Msg("Set log to pretty console")
log = log.Output(zerolog.ConsoleWriter{Out: os.Stderr})
}
func GetLogger() zerolog.Logger { func GetLogger() zerolog.Logger {
return log return log
} }