/
main.go
121 lines (99 loc) · 2.71 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
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
package main
import (
"context"
"net/http"
"os"
"os/signal"
"sync"
"syscall"
"time"
"github.com/jsmit257/userservice/internal/data/mysql"
"github.com/jsmit257/userservice/internal/metrics"
"github.com/jsmit257/userservice/internal/router"
"github.com/prometheus/client_golang/prometheus"
log "github.com/sirupsen/logrus"
)
const APP_NAME = "serve-mysql"
var traps = []os.Signal{
os.Interrupt,
syscall.SIGTERM,
syscall.SIGHUP,
syscall.SIGQUIT}
func main() {
cfg := NewConfig()
// TODO:
// - os.MkdirAll(cfg.LogDir)
// - logFileBasePath := cfg.LogDir + os.PathSeparator + cfg.LogFileName
// - logFilePath := logFileBasePath + os.Process.Pid
// - var logWriter *io.Writer
// - if logFile, err := os.OpenFile(logFilePath, os.O_CREATE+os.???, 644); err != nil {
// - panic(fmt.Errorf("can't create logfile: '%s'", logFilePath))
// - } else {
// - logWriter = bufio.NewWriter(logFile)
// - defer logFile.Close()
// - }
// - os.Link(logFilePath, logFileBasePath)
// - log.SetOutput(logWriter)
log.SetOutput(os.Stdout)
// TODO: log.SetLevel(log.Level(cfg.LogLevel))
log.SetLevel(log.DebugLevel)
// log.SetReportCaller(true) // this seems expensive, maybe nice to have, check it out before enabling
log.SetFormatter(&log.JSONFormatter{})
logger := log.WithField("app", APP_NAME)
logger.Debug("loaded config")
mtrcs := metrics.NewHandler(prometheus.NewRegistry())
mysql, err := mysql.NewInstance(
cfg.MySQLUser,
cfg.MySQLPwd,
cfg.MySQLHost,
cfg.MySQLPort,
logger)
cfg.MySQLPwd = "*****" // kinda rude
if err != nil {
logger.
WithFields(log.Fields{"cfg": cfg}).
WithError(err).
Fatal("failed to connect mysql client")
return
}
logger.Debug("configured mysql client")
srv := router.NewInstance(
&router.UserService{
User: mysql,
Address: mysql,
Contact: mysql,
},
cfg.ServerHost,
cfg.ServerPort,
mtrcs,
logger)
wg := &sync.WaitGroup{}
wg.Add(1)
go func(srv *http.Server, wg *sync.WaitGroup) {
defer wg.Done()
if err := srv.ListenAndServe(); err != http.ErrServerClosed {
logger.
WithFields(log.Fields{"cfg": cfg}).
WithError(err).
Fatal("http server didn't start properly")
panic(err)
}
logger.Debug("http server shutdown complete")
}(srv, wg)
trap(logger)
timeout, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
if err := srv.Shutdown(timeout); err != nil {
logger.
WithFields(log.Fields{"cfg": cfg}).
WithError(err).
Error("error stopping server")
}
wg.Wait()
logger.Debug("done")
}
func trap(logger *log.Entry) {
trapped := make(chan os.Signal, len(traps))
signal.Notify(trapped, traps...)
logger.WithField("sig", <-trapped).Info("stopping app with signal")
}