-
Notifications
You must be signed in to change notification settings - Fork 0
/
handlers.go
78 lines (73 loc) · 2.22 KB
/
handlers.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
package main
import (
"encoding/json"
"github.com/julienschmidt/httprouter"
"io"
"net/http"
)
func getFromQueueHandler(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
defer req.Body.Close()
queueName := ps.ByName("queue")
q, err := qm.GetQueue(queueName)
if err != nil {
io.WriteString(w, "Queue "+queueName+" not found")
logger.log(QK_LOG_LEVEL_WARNING, "Queue "+queueName+" not found")
http.NotFound(w, req)
return
}
msg, err := q.Get()
if err != nil {
http.Error(w, err.Error(), 400)
logger.log(QK_LOG_LEVEL_ERROR, "Get from queue error: "+err.Error())
return
}
logger.log(QK_LOG_LEVEL_TRACE, "Get from queue: "+msg.String())
io.WriteString(w, msg.String())
}
func putToQueueHandler(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
defer req.Body.Close()
queueName := ps.ByName("queue")
q, err := qm.GetQueue(queueName)
if err != nil {
io.WriteString(w, "Queue "+queueName+" not found")
logger.log(QK_LOG_LEVEL_WARNING, "Queue "+queueName+" not found")
http.NotFound(w, req)
return
}
body, err := extractBody(req)
if err != nil {
http.Error(w, err.Error(), 400)
return
}
preparedMessage := message{Msg: body, Delay: -1}
qi := makeQueueItem(preparedMessage)
msg, err := q.Put(qi)
if err != nil {
http.Error(w, err.Error(), 400)
logger.log(QK_LOG_LEVEL_ERROR, "Put in queue error: "+err.Error())
}
logger.log(QK_LOG_LEVEL_TRACE, "Put in queue: "+msg.String())
io.WriteString(w, msg.String())
}
func adminReloadQueueConfigHandler(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
defer req.Body.Close()
adminReloadQueueConfigMutex.Lock()
defer adminReloadQueueConfigMutex.Unlock()
logger.log(QK_LOG_LEVEL_INFO, "Reload queue config")
readQueuesConfigs(qm, conf)
io.WriteString(w, qm.String())
}
func healthRoute(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
defer req.Body.Close()
logger.log(QK_LOG_LEVEL_INFO, "Read health info")
h := health()
w.Header().Set("Content-Type", "application/json; charset=utf-8")
jsonBA, err := json.Marshal(h)
if err != nil {
http.Error(w, err.Error(), 400)
logger.log(QK_LOG_LEVEL_ERROR, "json error: "+err.Error())
return
}
jsonStr := string(jsonBA[:])
io.WriteString(w, jsonStr)
}