mirror of
https://github.com/hibiken/asynq.git
synced 2025-04-22 16:50:18 +08:00
Simplify ListWorkers implementation
This commit is contained in:
parent
ea456741b6
commit
d95f487bef
@ -381,7 +381,6 @@ type WorkerInfo struct {
|
|||||||
Host string
|
Host string
|
||||||
PID int
|
PID int
|
||||||
ServerID string
|
ServerID string
|
||||||
// TODO: Can we have Task field (*TaskMessage)?
|
|
||||||
ID string
|
ID string
|
||||||
Type string
|
Type string
|
||||||
Payload map[string]interface{}
|
Payload map[string]interface{}
|
||||||
|
@ -1022,36 +1022,37 @@ func (r *RDB) ListServers() ([]*base.ServerInfo, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Note: Script also removes stale keys.
|
// Note: Script also removes stale keys.
|
||||||
var listWorkerKeysCmd = redis.NewScript(`
|
var listWorkersCmd = redis.NewScript(`
|
||||||
local now = tonumber(ARGV[1])
|
local now = tonumber(ARGV[1])
|
||||||
local keys = redis.call("ZRANGEBYSCORE", KEYS[1], now, "+inf")
|
local keys = redis.call("ZRANGEBYSCORE", KEYS[1], now, "+inf")
|
||||||
redis.call("ZREMRANGEBYSCORE", KEYS[1], "-inf", now-1)
|
redis.call("ZREMRANGEBYSCORE", KEYS[1], "-inf", now-1)
|
||||||
return keys`)
|
local res = {}
|
||||||
|
for _, key in ipairs(keys) do
|
||||||
|
local vals = redis.call("HVALS", key)
|
||||||
|
for _, v in ipairs(vals) do
|
||||||
|
table.insert(res, v)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
return res`)
|
||||||
|
|
||||||
// ListWorkers returns the list of worker stats.
|
// ListWorkers returns the list of worker stats.
|
||||||
func (r *RDB) ListWorkers() ([]*base.WorkerInfo, error) {
|
func (r *RDB) ListWorkers() ([]*base.WorkerInfo, error) {
|
||||||
now := time.Now()
|
now := time.Now()
|
||||||
res, err := listWorkerKeysCmd.Run(r.client, []string{base.AllWorkers}, now.Unix()).Result()
|
res, err := listWorkersCmd.Run(r.client, []string{base.AllWorkers}, now.Unix()).Result()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
keys, err := cast.ToStringSliceE(res)
|
data, err := cast.ToStringSliceE(res)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
var workers []*base.WorkerInfo
|
var workers []*base.WorkerInfo
|
||||||
for _, key := range keys {
|
for _, s := range data {
|
||||||
data, err := r.client.HVals(key).Result()
|
w, err := base.DecodeWorkerInfo([]byte(s))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
continue // skip bad data
|
continue // skip bad data
|
||||||
}
|
}
|
||||||
for _, s := range data {
|
workers = append(workers, w)
|
||||||
w, err := base.DecodeWorkerInfo([]byte(s))
|
|
||||||
if err != nil {
|
|
||||||
continue // skip bad data
|
|
||||||
}
|
|
||||||
workers = append(workers, w)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
return workers, nil
|
return workers, nil
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user