Commit 7211cda0 authored by vicotor's avatar vicotor

fix bug

parent 2e964852
...@@ -30,10 +30,11 @@ type Worker struct { ...@@ -30,10 +30,11 @@ type Worker struct {
func PopWorker(rdb *redis.Client) (Worker, error) { func PopWorker(rdb *redis.Client) (Worker, error) {
for i := 0; i < maxPriority; i++ { for i := 0; i < maxPriority; i++ {
for {
elem, err := rdb.LPop(context.Background(), config.WORKER_QUEUE_PREFIX+strconv.Itoa(i)).Result() elem, err := rdb.LPop(context.Background(), config.WORKER_QUEUE_PREFIX+strconv.Itoa(i)).Result()
if err != nil { if err != nil {
log.WithError(err).Error("lPop worker failed") log.WithError(err).Error("lPop worker failed")
continue break
} }
log.WithField("elem", elem).Debug("lPop worker") log.WithField("elem", elem).Debug("lPop worker")
managerList, err := rdb.SMembers(context.Background(), config.WORKER_STATUS_PREFIX+elem).Result() managerList, err := rdb.SMembers(context.Background(), config.WORKER_STATUS_PREFIX+elem).Result()
...@@ -50,6 +51,7 @@ func PopWorker(rdb *redis.Client) (Worker, error) { ...@@ -50,6 +51,7 @@ func PopWorker(rdb *redis.Client) (Worker, error) {
priority: i, priority: i,
managers: managerList, managers: managerList,
}, nil }, nil
}
} }
return Worker{}, ErrNoWorker return Worker{}, ErrNoWorker
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment