mirror of
https://github.com/httprunner/httprunner.git
synced 2026-05-12 19:39:44 +08:00
99 lines
2.7 KiB
Go
99 lines
2.7 KiB
Go
package ios
|
|
|
|
import (
|
|
"os"
|
|
"os/signal"
|
|
"strings"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/httprunner/httprunner/v4/hrp/internal/builtin"
|
|
"github.com/httprunner/httprunner/v4/hrp/internal/env"
|
|
"github.com/httprunner/httprunner/v4/hrp/internal/sdk"
|
|
"github.com/httprunner/httprunner/v4/hrp/pkg/uixt"
|
|
)
|
|
|
|
var perfCmd = &cobra.Command{
|
|
Use: "perf",
|
|
Short: "capture ios performance data (cpu,mem,disk,net,fps,etc.)",
|
|
RunE: func(cmd *cobra.Command, args []string) (err error) {
|
|
startTime := time.Now()
|
|
defer func() {
|
|
sdk.SendGA4Event("hrp_ios_perf", map[string]interface{}{
|
|
"args": strings.Join(args, "-"),
|
|
"success": err == nil,
|
|
"engagement_time_msec": time.Since(startTime).Milliseconds(),
|
|
})
|
|
}()
|
|
|
|
perfOptions := []uixt.IOSPerfOption{}
|
|
for _, p := range indicators {
|
|
switch p {
|
|
case "sys_cpu":
|
|
perfOptions = append(perfOptions, uixt.WithIOSPerfSystemCPU(true))
|
|
case "sys_mem":
|
|
perfOptions = append(perfOptions, uixt.WithIOSPerfSystemMem(true))
|
|
case "sys_net":
|
|
perfOptions = append(perfOptions, uixt.WithIOSPerfSystemNetwork(true))
|
|
case "sys_disk":
|
|
perfOptions = append(perfOptions, uixt.WithIOSPerfSystemDisk(true))
|
|
case "network":
|
|
perfOptions = append(perfOptions, uixt.WithIOSPerfNetwork(true))
|
|
case "fps":
|
|
perfOptions = append(perfOptions, uixt.WithIOSPerfFPS(true))
|
|
case "gpu":
|
|
perfOptions = append(perfOptions, uixt.WithIOSPerfGPU(true))
|
|
}
|
|
}
|
|
perfOptions = append(perfOptions, uixt.WithIOSPerfOutputInterval(interval*1000))
|
|
|
|
device, err := uixt.NewIOSDevice(
|
|
uixt.WithUDID(udid),
|
|
uixt.WithIOSPerfOptions(perfOptions...),
|
|
)
|
|
if err != nil {
|
|
log.Fatal().Err(err).Msg("failed to init ios device")
|
|
}
|
|
|
|
err = builtin.EnsureFolderExists(env.ResultsPath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if err = device.StartPerf(); err != nil {
|
|
return err
|
|
}
|
|
defer device.StopPerf()
|
|
|
|
c := make(chan os.Signal, 1)
|
|
signal.Notify(c, syscall.SIGTERM, syscall.SIGINT)
|
|
timer := time.NewTimer(time.Duration(timeDuration) * time.Second)
|
|
for {
|
|
select {
|
|
case <-timer.C:
|
|
return nil
|
|
case <-c:
|
|
log.Warn().Msg("received signal, stop perf")
|
|
return nil
|
|
}
|
|
}
|
|
},
|
|
}
|
|
|
|
var (
|
|
interval int
|
|
indicators []string
|
|
)
|
|
|
|
func init() {
|
|
perfCmd.Flags().StringVarP(&udid, "udid", "u", "", "specify device by udid")
|
|
perfCmd.Flags().StringSliceVarP(&indicators, "indicators", "p", []string{"sys_cpu", "sys_mem"},
|
|
"specify performance monitor, e.g. sys_cpu,sys_mem,sys_net,sys_disk,fps,network,gpu")
|
|
perfCmd.Flags().IntVarP(&timeDuration, "duration", "t", 10, "specify time duraion in seconds")
|
|
perfCmd.Flags().IntVarP(&interval, "interval", "i", 3, "set interval in seconds")
|
|
iosRootCmd.AddCommand(perfCmd)
|
|
}
|