fix #1
1
.gitignore
vendored
Normal file
1
.gitignore
vendored
Normal file
@ -0,0 +1 @@
|
||||
queues-go
|
@ -21,6 +21,7 @@ func Put(w http.ResponseWriter, r *http.Request) {
|
||||
w.WriteHeader(http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
queue := r.Header.Get("queue")
|
||||
var availableFrom time.Time
|
||||
if body.Delay == nil {
|
||||
@ -28,6 +29,7 @@ func Put(w http.ResponseWriter, r *http.Request) {
|
||||
} else {
|
||||
availableFrom = time.Now().Add(time.Second + time.Duration(*body.Delay))
|
||||
}
|
||||
|
||||
task := tasks.InsertedTask{
|
||||
Queue: queue,
|
||||
Payload: body.Payload,
|
||||
@ -37,10 +39,12 @@ func Put(w http.ResponseWriter, r *http.Request) {
|
||||
TakenAt: nil,
|
||||
Attempts: 0,
|
||||
}
|
||||
|
||||
err = tasks.Add(task)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
w.WriteHeader(http.StatusAccepted)
|
||||
}
|
||||
|
@ -23,6 +23,7 @@ func Take(w http.ResponseWriter, r *http.Request) {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
var response TakeResponse
|
||||
if task == nil {
|
||||
response.Task = nil
|
||||
@ -33,10 +34,12 @@ func Take(w http.ResponseWriter, r *http.Request) {
|
||||
Payload: task.Payload,
|
||||
}
|
||||
}
|
||||
|
||||
data, err := json.Marshal(response)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
w.Write(data)
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user