-
Notifications
You must be signed in to change notification settings - Fork 16
/
main.go
67 lines (60 loc) · 1.41 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
package main
import (
"flag"
"log"
"os"
"os/signal"
"strings"
"syscall"
)
func runServer(config *Config) {
server := createServer(config)
log.Printf("Starting server on %s", config.ServerAddress)
done := make(chan os.Signal, 1)
signal.Notify(done, os.Interrupt, syscall.SIGINT, syscall.SIGTERM)
go func() {
var err error
if strings.HasPrefix(config.ServerAddress, "unix:") {
socketPath := strings.Replace(config.ServerAddress, "unix:", "", -1)
defer os.Remove(socketPath)
err = server.ListenAndServeUNIX(socketPath, os.ModeSocket|0666)
} else {
err = server.ListenAndServe(config.ServerAddress)
}
if err != nil {
log.Fatalf("Listen error: %v", err)
}
}()
<-done
log.Print("Graceful Shutdown")
if err := server.Shutdown(); err != nil {
log.Fatal(err)
}
}
func main() {
configPath := flag.String("config", "", "Path of config file in yml format")
flag.Parse()
if *configPath == "" {
log.Fatal("Set config.yml path via -config flag.")
}
file, err := os.Open(*configPath)
if err != nil {
log.Fatal(err)
}
config, err := parseConfig(file)
file.Close()
if err != nil {
log.Fatal(err)
}
if config.LogPath != "" {
logFile, err := os.OpenFile(config.LogPath, os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0666)
if err != nil {
log.Fatalf("Could not open log file: %v", err)
}
defer logFile.Close()
log.SetOutput(logFile)
} else {
log.SetOutput(os.Stdout)
}
runServer(config)
}