Commit 231eb8ee authored by vicotor's avatar vicotor

update

parent 4b607e05
...@@ -24,7 +24,7 @@ func (wm *WorkerManager) AddWorkerFirst(worker *Worker) error { ...@@ -24,7 +24,7 @@ func (wm *WorkerManager) AddWorkerFirst(worker *Worker) error {
} }
} }
// add worker to redis queue // add worker to redis queue
if err := wm.rdb.SAdd(context.Background(), config.WORKER_STATUS_PREFIX+worker.addr, config.GetConfig().LocalEndpoint()).Err(); err != nil { if err := wm.rdb.SAdd(context.Background(), config.WORKER_STATUS_PREFIX+worker.addr, config.GetConfig().PublicEndpoint()).Err(); err != nil {
return err return err
} }
return nil return nil
...@@ -40,16 +40,16 @@ func (wm *WorkerManager) AddWorkerSingle(worker *Worker) error { ...@@ -40,16 +40,16 @@ func (wm *WorkerManager) AddWorkerSingle(worker *Worker) error {
} }
} }
// add worker to redis queue // add worker to redis queue
if err := wm.rdb.SAdd(context.Background(), config.WORKER_STATUS_PREFIX+worker.addr, config.GetConfig().LocalEndpoint()).Err(); err != nil { if err := wm.rdb.SAdd(context.Background(), config.WORKER_STATUS_PREFIX+worker.addr, config.GetConfig().PublicEndpoint()).Err(); err != nil {
return err return err
} }
return nil return nil
} }
func (wm *WorkerManager) ActiveWorker(worker *Worker) { func (wm *WorkerManager) ActiveWorker(worker *Worker) {
wm.rdb.SAdd(context.Background(), config.WORKER_STATUS_PREFIX+worker.addr, config.GetConfig().LocalEndpoint()) wm.rdb.SAdd(context.Background(), config.WORKER_STATUS_PREFIX+worker.addr, config.GetConfig().PublicEndpoint())
} }
func (wm *WorkerManager) InActiveWorker(addr string) { func (wm *WorkerManager) InActiveWorker(addr string) {
wm.rdb.SRem(context.Background(), config.WORKER_STATUS_PREFIX+addr, config.GetConfig().LocalEndpoint()) wm.rdb.SRem(context.Background(), config.WORKER_STATUS_PREFIX+addr, config.GetConfig().PublicEndpoint())
} }
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