fix
This commit is contained in:
@@ -1,11 +1,9 @@
|
||||
package routers
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
tasks "queues-go/app/storage/mongo/collections"
|
||||
"queues-go/app/storage/redis"
|
||||
"time"
|
||||
"sync"
|
||||
)
|
||||
|
||||
type TaskResponse struct {
|
||||
@@ -18,17 +16,16 @@ type TakeResponse struct {
|
||||
Task *TaskResponse `json:"task"`
|
||||
}
|
||||
|
||||
var MutexMap map[string]*sync.Mutex
|
||||
|
||||
func Take(r *http.Request) (interface{}, int) {
|
||||
queue := r.Header.Get("queue")
|
||||
mutex := redis.Sync.NewMutex(fmt.Sprintf("lock_queues_%s", queue))
|
||||
for {
|
||||
err := mutex.Lock()
|
||||
if err != nil {
|
||||
time.Sleep(time.Millisecond * 5)
|
||||
} else {
|
||||
break
|
||||
}
|
||||
mutex, ok := MutexMap[queue]
|
||||
if !ok {
|
||||
mutex = &sync.Mutex{}
|
||||
MutexMap[queue] = mutex
|
||||
}
|
||||
mutex.Lock()
|
||||
task, err := tasks.Take(queue)
|
||||
mutex.Unlock()
|
||||
if err != nil {
|
||||
|
||||
Reference in New Issue
Block a user