-
Notifications
You must be signed in to change notification settings - Fork 0
/
broker.go
121 lines (106 loc) · 2.6 KB
/
broker.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
112
113
114
115
116
117
118
119
120
121
/**
* Copyright (c) 2015 Intel Corporation
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package main
import (
"errors"
"log"
"net/http"
"code.google.com/p/go.net/websocket"
)
type Authenticator interface {
Validate(*http.Request) bool
}
func newBroker() *broker {
clients := make(map[int64]*handler, 5)
addCh := make(chan *handler, 5)
delCh := make(chan *handler)
doneCh := make(chan bool)
errCh := make(chan error)
var authV Authenticator
switch args.Server.AuthMethod {
case "none":
log.Println("Using no authentication")
authV = NewNoAuth()
case "simple":
log.Println("Using simple authentication")
authV = NewSimpleAuth(args.Server.Token)
case "jwt":
log.Println("Using JWT authentication")
authV = NewJwtAuth()
}
return &broker{
clients,
addCh,
delCh,
doneCh,
errCh,
authV,
}
}
type broker struct {
clients map[int64]*handler
addCh chan *handler
delCh chan *handler
doneCh chan bool
errCh chan error
authVal Authenticator
}
func (s *broker) add(c *handler) { s.addCh <- c }
func (s *broker) del(c *handler) { s.delCh <- c }
func (s *broker) err(err error) { s.errCh <- err }
func (s *broker) listen() {
onConnected := func(ws *websocket.Conn) {
// make sure closes cleanly
defer func() {
err := ws.Close()
if err != nil {
s.errCh <- err
}
}()
// create a new producer client per connection
if s.authVal.Validate(ws.Request()) {
handler := newClient(ws, s)
s.add(handler)
handler.listen()
} else {
log.Println("Invalid token")
s.errCh <- errors.New("Invalid token")
}
}
onRequest := func(w http.ResponseWriter, req *http.Request) {
s := websocket.Server{
Handler: websocket.Handler(onConnected),
}
s.ServeHTTP(w, req)
}
http.HandleFunc(args.Server.Root, onRequest)
for {
select {
case c := <-s.addCh:
s.clients[c.id] = c
if args.Trace {
log.Printf("app:%d handler:%d clients:%d", args.Index, c.id, len(s.clients))
}
case c := <-s.delCh:
delete(s.clients, c.id)
if args.Trace {
Trace("handler deleted", c.id)
}
case err := <-s.errCh:
log.Println("error:", err.Error())
}
}
}