46 lines
834 B
Go
46 lines
834 B
Go
package routers
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
tasks "queues-go/app/storage/mongo/collections"
|
|
)
|
|
|
|
type TaskResponse struct {
|
|
Id string `json:"id"`
|
|
Attempt int `json:"attempt"`
|
|
Payload interface{} `json:"payload"`
|
|
}
|
|
|
|
type TakeResponse struct {
|
|
Task *TaskResponse `json:"task"`
|
|
}
|
|
|
|
func Take(w http.ResponseWriter, r *http.Request) {
|
|
queue := r.Header.Get("queue")
|
|
task, err := tasks.Take(queue)
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
var response TakeResponse
|
|
if task == nil {
|
|
response.Task = nil
|
|
} else {
|
|
response.Task = &TaskResponse{
|
|
Id: task.Id.Hex(),
|
|
Attempt: task.Attempts,
|
|
Payload: task.Payload,
|
|
}
|
|
}
|
|
|
|
data, err := json.Marshal(response)
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
w.Write(data)
|
|
}
|