-
Notifications
You must be signed in to change notification settings - Fork 143
/
Copy pathmain.go
75 lines (61 loc) · 1.5 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
package main
import (
"github.com/rcrowley/goagain"
"fmt"
"log"
"net"
"syscall"
"time"
)
func init() {
log.SetFlags(log.Lmicroseconds | log.Lshortfile)
log.SetPrefix(fmt.Sprintf("pid:%d ", syscall.Getpid()))
}
func main() {
// Inherit a net.Listener from our parent process or listen anew.
l, err := goagain.Listener()
if nil != err {
// Listen on a TCP or a UNIX domain socket (TCP here).
l, err = net.Listen("tcp", "127.0.0.1:48879")
if nil != err {
log.Fatalln(err)
}
log.Println("listening on", l.Addr())
// Accept connections in a new goroutine.
go serve(l)
} else {
// Resume accepting connections in a new goroutine.
log.Println("resuming listening on", l.Addr())
go serve(l)
// Kill the parent, now that the child has started successfully.
if err := goagain.Kill(); nil != err {
log.Fatalln(err)
}
}
// Block the main goroutine awaiting signals.
if _, err := goagain.Wait(l); nil != err {
log.Fatalln(err)
}
// Do whatever's necessary to ensure a graceful exit like waiting for
// goroutines to terminate or a channel to become closed.
//
// In this case, we'll simply stop listening and wait one second.
if err := l.Close(); nil != err {
log.Fatalln(err)
}
time.Sleep(1e9)
}
// A very rude server that says hello and then closes your connection.
func serve(l net.Listener) {
for {
c, err := l.Accept()
if nil != err {
if goagain.IsErrClosing(err) {
break
}
log.Fatalln(err)
}
c.Write([]byte("Hello, world!\n"))
c.Close()
}
}