/
socketserver.go
58 lines (49 loc) · 1.05 KB
/
socketserver.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
package main
import (
"log"
"net"
"os"
"os/signal"
"syscall"
)
//https://gist.github.com/hakobe/6f70d69b8c5243117787fd488ae7fbf2
func RequestSocketServer(c net.Conn) {
for {
buf := make([]byte, 512)
nr, err := c.Read(buf)
if err != nil {
return
}
data := buf[0:nr]
log.Println("Server got:", string(data))
if string(data) == "reload" {
ReloadBackendServers(flagConfigFile)
data = []byte("server was reloaded")
}
_, err = c.Write(data)
if err != nil {
log.Println("Writing client error: ", err)
}
}
}
func StartSocketServer() {
syscall.Unlink(conf.SocketPath)
ln, err := net.Listen("unix", conf.SocketPath)
if err != nil {
log.Fatal("Listen error: ", conf.SocketPath, err.Error())
}
sigc := make(chan os.Signal, 1)
signal.Notify(sigc, os.Interrupt, os.Kill, syscall.SIGTERM)
go func(ln net.Listener, c chan os.Signal) {
sig := <-c
log.Printf("Caught signal %s: shutting down.", sig)
ln.Close()
os.Exit(0)
}(ln, sigc)
for {
fd, err := ln.Accept()
if err == nil {
go RequestSocketServer(fd)
}
}
}