fix panic when containerd-stress density --count 0

Signed-off-by: yanggang <gang.yang@daocloud.io>
This commit is contained in:
yanggang 2022-12-02 17:32:21 +08:00
parent c0a89fbbdd
commit b0dd506553
No known key found for this signature in database
GPG Key ID: 7F6E0CAD1556A240

View File

@ -20,6 +20,7 @@ import (
"bufio" "bufio"
"context" "context"
"encoding/json" "encoding/json"
"errors"
"fmt" "fmt"
"os" "os"
"os/signal" "os/signal"
@ -47,6 +48,15 @@ var densityCommand = cli.Command{
}, },
}, },
Action: func(cliContext *cli.Context) error { Action: func(cliContext *cli.Context) error {
var (
pids []uint32
count = cliContext.Int("count")
)
if count < 1 {
return errors.New("count cannot be less than one")
}
config := config{ config := config{
Address: cliContext.GlobalString("address"), Address: cliContext.GlobalString("address"),
Duration: cliContext.GlobalDuration("duration"), Duration: cliContext.GlobalDuration("duration"),
@ -76,10 +86,6 @@ var densityCommand = cli.Command{
s := make(chan os.Signal, 1) s := make(chan os.Signal, 1)
signal.Notify(s, syscall.SIGTERM, syscall.SIGINT) signal.Notify(s, syscall.SIGTERM, syscall.SIGINT)
var (
pids []uint32
count = cliContext.Int("count")
)
loop: loop:
for i := 0; i < count+1; i++ { for i := 0; i < count+1; i++ {
select { select {