fix

app/routers/take.go

@@ -1,7 +1,6 @@
 package routers
 
 import (
-	"encoding/json"
 	"net/http"
 	tasks "queues-go/app/storage/mongo/collections"
 )
@@ -16,13 +15,11 @@ type TakeResponse struct {
 	Task *TaskResponse `json:"task"`
 }
 
-func Take(w http.ResponseWriter, r *http.Request) {
+func Take(r *http.Request) (interface{}, int) {
 	queue := r.Header.Get("queue")
 	task, err := tasks.Take(queue)
 	if err != nil {
-		println("Error taking")
-		w.WriteHeader(http.StatusInternalServerError)
-		return
+		return nil, http.StatusInternalServerError
 	}
 
 	var response TakeResponse
@@ -36,12 +33,5 @@ func Take(w http.ResponseWriter, r *http.Request) {
 		}
 	}
 
-	data, err := json.Marshal(response)
-	if err != nil {
-		println("Error marshalling")
-		w.WriteHeader(http.StatusInternalServerError)
-		return
-	}
-
-	w.Write(data)
+	return response, http.StatusOK
 }