forked from tomasen/fast-table-tunnel
/
main.go
71 lines (61 loc) · 1.48 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
package main
import (
gozd "bitbucket.org/PinIdea/zero-downtime-daemon"
ftunnel "bitbucket.org/Tomasen/fast-table-tunnel/src"
"flag"
"log"
"net"
"os"
"syscall"
)
var (
optAccept = flag.String("s", "", "listen to ip:port")
optConnect = flag.String("c", "", "connect to ip:port")
optServerID = flag.String("id", "fasttunnel", "connect to ip:port")
optLogfile = flag.String("log", "", "log filepath")
optHelp = flag.Bool("h", false, "this help")
)
func usage() {
log.Println("[command] -conf=[config file]")
flag.PrintDefaults()
}
func main() {
// parse arguments
flag.Parse()
if *optHelp || len(*optServerID) <= 0 || len(*optAccept) <= 0 || len(*optConnect) <= 0 {
usage()
return
}
log.Println(os.TempDir())
ctx := gozd.Context{
Hash: *optServerID,
Command: "start",
Maxfds: syscall.Rlimit{Cur: 32677, Max: 32677},
User: "nobody",
Group: "nobody",
Logfile: "ftunnel.log",
Directives: map[string]gozd.Server{
"client": gozd.Server{
Network: "tcp",
Address: *optAccept,
},
},
}
cl := make(chan net.Listener, 1)
go ftunnel.HandleListners(cl, *optConnect)
sig, err := gozd.Daemonize(ctx, cl) // returns channel that connects with daemon
if err != nil {
log.Println("error: ", err)
return
}
// other initializations or config setting
for s := range sig {
switch s {
case syscall.SIGHUP, syscall.SIGUSR2:
// do some custom jobs while reload/hotupdate
case syscall.SIGTERM:
// do some clean up and exit
return
}
}
}