mirror of
https://github.com/hibiken/asynq.git
synced 2026-06-10 15:04:03 +08:00
f8d6677814
Migrate all CLI command handlers from cobra's Run to RunE, replacing fmt.Println(err) + os.Exit(1) patterns with idiomatic error returns. This improves testability and lets cobra handle errors consistently via the existing SilenceUsage/SilenceErrors configuration. Also refactor getDuration() and getTime() helpers to return errors instead of calling os.Exit(1). Fix a bug in group.go where the error from inspector.Groups() was never checked, causing silent failures (e.g. on Redis connection errors) to be misreported as "No groups found".
45 lines
1.0 KiB
Go
45 lines
1.0 KiB
Go
// Copyright 2022 Kentaro Hibino. All rights reserved.
|
|
// Use of this source code is governed by a MIT license
|
|
// that can be found in the LICENSE file.
|
|
|
|
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/MakeNowJust/heredoc/v2"
|
|
"github.com/hibiken/asynq/tools/asynq/cmd/dash"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var (
|
|
flagPollInterval = 8 * time.Second
|
|
)
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(dashCmd)
|
|
dashCmd.Flags().DurationVar(&flagPollInterval, "refresh", 8*time.Second, "Interval between data refresh (default: 8s, min allowed: 1s)")
|
|
}
|
|
|
|
var dashCmd = &cobra.Command{
|
|
Use: "dash",
|
|
Short: "View dashboard",
|
|
Long: heredoc.Doc(`
|
|
Display interactive dashboard.`),
|
|
Args: cobra.NoArgs,
|
|
Example: heredoc.Doc(`
|
|
$ asynq dash
|
|
$ asynq dash --refresh=3s`),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
if flagPollInterval < 1*time.Second {
|
|
return fmt.Errorf("--refresh cannot be less than 1s")
|
|
}
|
|
dash.Run(dash.Options{
|
|
PollInterval: flagPollInterval,
|
|
RedisConnOpt: getRedisConnOpt(),
|
|
})
|
|
return nil
|
|
},
|
|
}
|