-
Notifications
You must be signed in to change notification settings - Fork 0
/
index.go
84 lines (72 loc) · 2.18 KB
/
index.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
package handlers
import (
"context"
"log"
"net/http"
"github.com/gorilla/websocket"
"github.com/lmika/broadtail/middleware/errhandler"
"github.com/lmika/broadtail/models"
"github.com/lmika/broadtail/providers/jobs"
"github.com/pkg/errors"
"github.com/lmika/broadtail/services/feedsmanager"
"github.com/lmika/broadtail/services/jobsmanager"
"github.com/lmika/gopkgs/http/middleware/render"
)
type indexHandlers struct {
jobsManager *jobsmanager.JobsManager
feedsManager *feedsmanager.FeedsManager
upgrader websocket.Upgrader
}
func (ih *indexHandlers) Index() http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
recentFeedItems, err := ih.feedsManager.RecentFeedItemsFromAllFeeds(r.Context(), models.FeedItemFilter{
Ordering: models.ChronologicalFeedItemOrdering,
}, 0, 10)
if err != nil {
log.Printf("warn: cannot get list of recent feed items: %v", err)
}
render.Set(r, "recentFeedItems", recentFeedItems)
render.Set(r, "jobs", ih.jobsManager.RecentJobs())
render.HTML(r, w, http.StatusOK, "index.html")
})
}
func (ih *indexHandlers) StatusUpdateWebsocket() http.Handler {
type wbJobUpdateMessage struct {
ID string `json:"id"`
Type string `json:"type"`
State string `json:"state,omitempty"`
Summary string `json:"summary,omitempty"`
Percent float64 `json:"percent"`
}
return errhandler.HandlerFunc(func(ctx context.Context, w http.ResponseWriter, r *http.Request) error {
c, err := ih.upgrader.Upgrade(w, r, nil)
if err != nil {
return errors.Wrap(err, "cannot update socket")
}
defer c.Close()
sub := ih.jobsManager.Dispatcher().Subscribe()
defer sub.Close()
for msg := range sub.Chan() {
var err error = nil
switch m := msg.(type) {
case jobs.UpdateSubscriptionEvent:
err = c.WriteJSON(wbJobUpdateMessage{
ID: m.Job.ID().String(),
Type: "update",
Summary: m.Update.Summary,
Percent: m.Update.Percent,
})
case jobs.StateTransitionSubscriptionEvent:
err = c.WriteJSON(wbJobUpdateMessage{
ID: m.Job.ID().String(),
Type: "newstate",
State: m.ToState.String(),
})
}
if err != nil {
break
}
}
return nil
})
}