mirror of
https://github.com/hibiken/asynq.git
synced 2024-12-26 15:52:18 +08:00
55 lines
1.4 KiB
Go
55 lines
1.4 KiB
Go
// Copyright 2020 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"
|
|
"os"
|
|
|
|
"github.com/go-redis/redis/v7"
|
|
"github.com/hibiken/asynq/internal/rdb"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
// rmqCmd represents the rmq command
|
|
var rmqCmd = &cobra.Command{
|
|
Use: "rmq [queue name]",
|
|
Short: "Removes the specified queue",
|
|
Long: `Rmq (asynqmon rmq) will remove the specified queue.
|
|
By default, it will remove the queue only if it's empty.
|
|
Use --force option to override this behavior.
|
|
|
|
Example: asynqmon rmq low -> Removes "low" queue`,
|
|
Args: cobra.ExactValidArgs(1),
|
|
Run: rmq,
|
|
}
|
|
|
|
var rmqForce bool
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(rmqCmd)
|
|
rmqCmd.Flags().BoolVarP(&rmqForce, "force", "f", false, "remove the queue regardless of its size")
|
|
}
|
|
|
|
func rmq(cmd *cobra.Command, args []string) {
|
|
c := redis.NewClient(&redis.Options{
|
|
Addr: viper.GetString("uri"),
|
|
DB: viper.GetInt("db"),
|
|
Password: viper.GetString("password"),
|
|
})
|
|
r := rdb.NewRDB(c)
|
|
err := r.RemoveQueue(args[0], rmqForce)
|
|
if err != nil {
|
|
if _, ok := err.(*rdb.ErrQueueNotEmpty); ok {
|
|
fmt.Printf("error: %v\nIf you are sure you want to delete it, run 'asynqmon rmq --force %s'\n", err, args[0])
|
|
os.Exit(1)
|
|
}
|
|
fmt.Printf("error: %v", err)
|
|
os.Exit(1)
|
|
}
|
|
fmt.Printf("Successfully removed queue %q\n", args[0])
|
|
}
|