mirror of
https://github.com/httprunner/httprunner.git
synced 2026-05-07 04:42:42 +08:00
94 lines
2.6 KiB
Go
94 lines
2.6 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/rs/zerolog"
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/httprunner/httprunner/v4/hrp/pkg/uixt"
|
|
)
|
|
|
|
var rootCmd = &cobra.Command{
|
|
Use: "wcl",
|
|
Short: "Monitor FIFA World Cup Live",
|
|
Version: "2022.11.27.2240",
|
|
PreRun: func(cmd *cobra.Command, args []string) {
|
|
log.Logger = zerolog.New(
|
|
zerolog.ConsoleWriter{NoColor: false, Out: os.Stderr},
|
|
).With().Timestamp().Logger()
|
|
zerolog.SetGlobalLevel(zerolog.InfoLevel)
|
|
setLogLevel(logLevel)
|
|
},
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
var device uixt.Device
|
|
var bundleID string
|
|
if iosApp != "" {
|
|
log.Info().Str("bundleID", iosApp).Msg("init ios device")
|
|
device = initIOSDevice(uuid)
|
|
bundleID = iosApp
|
|
} else if androidApp != "" {
|
|
log.Info().Str("bundleID", androidApp).Msg("init android device")
|
|
device = initAndroidDevice(uuid)
|
|
bundleID = androidApp
|
|
} else {
|
|
return errors.New("android or ios app bundldID is required")
|
|
}
|
|
|
|
wc := NewWorldCupLive(device, matchName, bundleID, duration, interval)
|
|
wc.Start()
|
|
wc.DumpResult()
|
|
return nil
|
|
},
|
|
}
|
|
|
|
var (
|
|
uuid string
|
|
iosApp string
|
|
androidApp string
|
|
duration int
|
|
interval int
|
|
logLevel string
|
|
matchName string
|
|
perf []string
|
|
)
|
|
|
|
func main() {
|
|
rootCmd.PersistentFlags().StringVarP(&logLevel, "log-level", "l", "INFO", "set log level")
|
|
rootCmd.PersistentFlags().StringVarP(&uuid, "uuid", "u", "", "specify device serial or udid")
|
|
rootCmd.PersistentFlags().StringVar(&iosApp, "ios", "", "run ios app")
|
|
rootCmd.PersistentFlags().StringVar(&androidApp, "android", "", "run android app")
|
|
rootCmd.PersistentFlags().IntVarP(&duration, "duration", "d", 30, "set duration in seconds")
|
|
rootCmd.PersistentFlags().IntVarP(&interval, "interval", "i", 15, "set interval in seconds")
|
|
rootCmd.PersistentFlags().StringVarP(&matchName, "match-name", "n", "", "specify match name")
|
|
rootCmd.PersistentFlags().StringSliceVarP(&perf, "perf", "p", nil,
|
|
"specify performance monitor, e.g. sys_cpu,sys_mem,sys_net,sys_disk,fps,network,gpu")
|
|
|
|
err := rootCmd.Execute()
|
|
if 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)
|
|
}
|
|
}
|