2019-12-04 12:02:29 +08:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"flag"
|
|
|
|
"fmt"
|
|
|
|
"log"
|
|
|
|
"os"
|
|
|
|
"strings"
|
|
|
|
"text/tabwriter"
|
|
|
|
"time"
|
|
|
|
|
2019-12-04 13:01:26 +08:00
|
|
|
"github.com/go-redis/redis/v7"
|
|
|
|
"github.com/hibiken/asynq/internal/rdb"
|
2019-12-04 12:02:29 +08:00
|
|
|
)
|
|
|
|
|
|
|
|
var pollInterval = flag.Duration("interval", 3*time.Second, "polling interval")
|
|
|
|
|
|
|
|
func main() {
|
2019-12-04 13:01:26 +08:00
|
|
|
c := redis.NewClient(&redis.Options{
|
2019-12-04 12:02:29 +08:00
|
|
|
Addr: "localhost:6379",
|
|
|
|
DB: 2,
|
|
|
|
})
|
2019-12-04 13:01:26 +08:00
|
|
|
r := rdb.NewClient(c)
|
2019-12-04 12:02:29 +08:00
|
|
|
|
|
|
|
for {
|
2019-12-04 13:01:26 +08:00
|
|
|
stats, err := r.CurrentStats()
|
2019-12-04 12:02:29 +08:00
|
|
|
if err != nil {
|
|
|
|
log.Fatal(err)
|
|
|
|
}
|
|
|
|
printStats(stats)
|
|
|
|
fmt.Println()
|
|
|
|
time.Sleep(*pollInterval)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-12-04 13:01:26 +08:00
|
|
|
func printStats(s *rdb.Stats) {
|
2019-12-04 12:02:29 +08:00
|
|
|
format := strings.Repeat("%v\t", 5) + "\n"
|
|
|
|
tw := new(tabwriter.Writer).Init(os.Stdout, 0, 8, 2, ' ', 0)
|
|
|
|
fmt.Fprintf(tw, format, "Enqueued", "InProgress", "Scheduled", "Retry", "Dead")
|
|
|
|
fmt.Fprintf(tw, format, "--------", "----------", "---------", "-----", "----")
|
|
|
|
fmt.Fprintf(tw, format, s.Queued, s.InProgress, s.Scheduled, s.Retry, s.Dead)
|
|
|
|
tw.Flush()
|
|
|
|
}
|