-
Notifications
You must be signed in to change notification settings - Fork 13
/
main.go
97 lines (79 loc) · 1.85 KB
/
main.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
package main
import (
"fmt"
"log"
"sync"
"time"
"github.com/dgrr/websocket"
"github.com/fasthttp/router"
"github.com/valyala/fasthttp"
)
type Broadcaster struct {
cs sync.Map
}
func (b *Broadcaster) OnOpen(c *websocket.Conn) {
b.cs.Store(c.ID(), c)
log.Printf("%s connected\n", c.RemoteAddr())
}
func (b *Broadcaster) OnClose(c *websocket.Conn, err error) {
if err != nil {
log.Printf("%d closed with error: %s\n", c.ID(), err)
} else {
log.Printf("%d closed the connection\n", c.ID())
}
b.cs.Delete(c.ID())
}
func (b *Broadcaster) Start(i int) {
time.AfterFunc(time.Second, b.sendData(i))
}
func (b *Broadcaster) sendData(i int) func() {
return func() {
b.cs.Range(func(_, v interface{}) bool {
nc := v.(*websocket.Conn)
fmt.Fprintf(nc, "Sending message number %d\n", i)
return true
})
b.Start(i 1)
}
}
func main() {
b := &Broadcaster{}
wServer := websocket.Server{}
wServer.HandleOpen(b.OnOpen)
wServer.HandleClose(b.OnClose)
router := router.New()
router.GET("/", rootHandler)
router.GET("/ws", wServer.Upgrade)
server := fasthttp.Server{
Handler: router.Handler,
}
b.Start(0)
server.ListenAndServe(":8080")
}
func rootHandler(ctx *fasthttp.RequestCtx) {
ctx.SetContentType("text/html")
fmt.Fprintln(ctx, `<!DOCTYPE html>
<html>
<head>
<meta charset="UTF-8"/>
<title>Sample of websocket with Golang</title>
</head>
<body>
<div id="text"></div>
<script>
var ws = new WebSocket("ws://localhost:8080/ws");
ws.onmessage = function(e) {
var d = document.createElement("div");
d.innerHTML = e.data;
ws.send(e.data);
document.getElementById("text").appendChild(d);
}
ws.onclose = function(e){
var d = document.createElement("div");
d.innerHTML = "CLOSED";
document.getElementById("text").appendChild(d);
}
</script>
</body>
</html>`)
}