-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathserver.go
More file actions
135 lines (125 loc) · 3.14 KB
/
server.go
File metadata and controls
135 lines (125 loc) · 3.14 KB
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
122
123
124
125
126
127
128
129
130
131
132
133
134
135
package main
import (
"encoding/json"
"fmt"
"github.com/googollee/go-socket.io"
"github.com/tatsushid/go-fastping"
"log"
"net"
"net/http"
"strings"
"time"
)
type Response struct {
Url string
Addr string
Rtt time.Duration
Epoch int64
}
type PingManger struct {
p *fastping.Pinger
results map[string]*Response
ipTourl map[string]string
}
func main() {
server, err := socketio.NewServer(nil)
if err != nil {
log.Fatal(err)
}
var pm *PingManger
pinglist := make([]string, 2)
pinglist[0] = "google.com"
pinglist[1] = "cogentco.com"
pm = StartPingEmitter(pinglist, server)
// p = StartPingEmitter("google.com", server)
server.On("connection", func(so socketio.Socket) {
// Store this by connection ID or something
log.Println("on connection")
so.Join("chat")
so.On("chat message", func(msg string) {
fmt.Println(msg)
switch {
case msg == "start ping":
fmt.Print("Start Ping (Already Started)")
case msg == "stop ping":
fmt.Print("Stop Ping")
pm.p.Stop()
}
log.Println("emit:", so.Emit("chat message", msg))
so.BroadcastTo("chat", "chat message", msg)
})
so.On("disconnection", func() {
log.Println("on disconnect")
})
})
server.On("error", func(so socketio.Socket, err error) {
log.Println("error:", err)
})
http.Handle("/socket.io/", server)
http.Handle("/", http.FileServer(http.Dir("./public")))
//http.Handle("/add/{url}", handler)
log.Println("Serving at localhost:5000...")
log.Fatal(http.ListenAndServe(":5000", nil))
}
func AddURL(rw http.ResponseWriter, req *http.Request, pm PingManger) {
}
func StartPingEmitter(urls []string, server *socketio.Server) *PingManger {
pm := PingManger{
p: fastping.NewPinger(),
results: make(map[string]*Response),
ipTourl: make(map[string]string),
}
netProto := "ip4:icmp"
if strings.Index(urls[0], ":") != -1 {
netProto = "ip6:ipv6-icmp"
}
for _, element := range urls {
ra, err := net.ResolveIPAddr(netProto, element)
if err != nil {
return &pm
}
pm.ipTourl[ra.String()] = element
pm.results[ra.String()] = nil
pm.p.AddIPAddr(ra)
}
onRecv, onIdle := make(chan *Response), make(chan bool)
pm.p.OnRecv = func(addr *net.IPAddr, t time.Duration) {
onRecv <- &Response{Url: pm.ipTourl[addr.String()], Addr: addr.String(), Rtt: t, Epoch: time.Now().Unix()}
}
pm.p.OnIdle = func() {
onIdle <- true
}
pm.p.MaxRTT = time.Millisecond * 250
pm.p.RunLoop()
go func() {
for {
select {
case res := <-onRecv:
if _, ok := pm.results[res.Addr]; ok {
pm.results[res.Addr] = res
}
case <-onIdle:
for host, r := range pm.results {
if r == nil {
fmt.Printf("%s : unreachable %v\n", host, time.Now())
server.BroadcastTo("chat", "ping message", "0")
} else {
// msg := fmt.Sprintf("%s : %v %v\n", host, r.rtt, time.Now())
// fmt.Print(msg)
b, _ := json.Marshal(pm.results[host])
// fmt.Println(results[host])
// fmt.Println(string(b))
server.BroadcastTo("chat", "ping message", string(b))
}
pm.results[host] = nil
}
// case <-done:
// fmt.Println("HERE")
// p.Stop()
case <-pm.p.Done():
break
}
}
}()
return &pm
}