/
logs.go
111 lines (102 loc) · 2.6 KB
/
logs.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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
package daemon
import (
"bytes"
"net/http"
"os"
"strconv"
"strings"
docker "github.com/docker/docker/client"
"github.com/go-chi/render"
"github.com/ubclaunchpad/inertia/api"
"github.com/ubclaunchpad/inertia/daemon/inertiad/containers"
"github.com/ubclaunchpad/inertia/daemon/inertiad/log"
"github.com/ubclaunchpad/inertia/daemon/inertiad/res"
)
// logHandler handles requests for container logs
func (s *Server) logHandler(w http.ResponseWriter, r *http.Request) {
var (
shouldStream bool
err error
)
// Get container name and stream from request query params
params := r.URL.Query()
container := params.Get(api.Container)
streamParam := params.Get(api.Stream)
if streamParam != "" {
s, err := strconv.ParseBool(streamParam)
if err != nil {
println(err.Error())
render.Render(w, r, res.ErrBadRequest(err.Error()))
return
}
shouldStream = s
} else {
shouldStream = false
}
// Determine number of entries to fetch
entriesParam := params.Get(api.Entries)
var entries int
if entriesParam != "" {
if entries, err = strconv.Atoi(entriesParam); err != nil {
render.Render(w, r, res.ErrBadRequest("invalid number of entries",
"error", err))
return
}
}
if entries == 0 {
entries = 500
}
// Upgrade to websocket connection if required, otherwise just set up a
// standard streamer
var stream *log.Streamer
if shouldStream {
socket, err := s.websocket.Upgrade(w, r, nil)
if err != nil {
render.Render(w, r,
res.ErrInternalServer("failed to esablish websocket connection", err))
return
}
stream = log.NewStreamer(log.StreamerOptions{
Request: r,
Stdout: os.Stdout,
Socket: socket,
HTTPWriter: w,
})
} else {
stream = log.NewStreamer(log.StreamerOptions{
Request: r,
Stdout: os.Stdout,
HTTPWriter: w,
})
}
logs, err := containers.ContainerLogs(s.docker, containers.LogOptions{
Container: container,
Stream: shouldStream,
Entries: entries,
})
if err != nil {
if docker.IsErrNotFound(err) {
stream.Error(res.ErrNotFound(err.Error()))
} else {
stream.Error(res.ErrInternalServer("failed to find logs for container", err))
}
return
}
defer logs.Close()
if shouldStream {
var stop = make(chan struct{})
socket, err := stream.GetSocketWriter()
if err != nil {
stream.Error(res.ErrInternalServer("failed to write to socket", err))
return
}
log.FlushRoutine(socket, logs, stop)
defer stream.Close()
defer close(stop)
} else {
buf := new(bytes.Buffer)
buf.ReadFrom(logs)
render.Render(w, r, res.MsgOK("logs retrieved",
"logs", strings.Split(buf.String(), "\n")))
}
}