queues-go/app/routers/take.go

package routers

import (
	"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(r *http.Request) (interface{}, int) {
	queue := r.Header.Get("queue")
	task, err := tasks.Take(queue)
	if err != nil {
		return nil, http.StatusInternalServerError
	}

	var response TakeResponse
	if task == nil {
		response.Task = nil
	} else {
		response.Task = &TaskResponse{
			Id:      task.Id.Hex(),
			Attempt: task.Attempts,
			Payload: task.Payload,
		}
	}

	return response, http.StatusOK
}