fix
All checks were successful
Deploy Dev / Build (pull_request) Successful in 38s
Deploy Dev / Push (pull_request) Successful in 24s
Deploy Dev / Deploy dev (pull_request) Successful in 7s

This commit is contained in:
Egor Matveev
2025-01-06 13:33:28 +03:00
parent 64bee3c59f
commit bc411ea644
2 changed files with 10 additions and 13 deletions

View File

@@ -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 {