-
Notifications
You must be signed in to change notification settings - Fork 1
/
engine.go
94 lines (79 loc) · 1.82 KB
/
engine.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
package server
import (
"context"
"crypto/tls"
"fmt"
"net"
"net/http"
"github.com/chitoku-k/ejaculation-counter/reactor/service"
"github.com/gin-gonic/gin"
"github.com/prometheus/client_golang/prometheus/promhttp"
"golang.org/x/sync/errgroup"
)
type engine struct {
Port string
CertFile string
KeyFile string
Through service.Through
Doublet service.Doublet
}
type Engine interface {
Start(ctx context.Context) error
}
func NewEngine(
port string,
certFile string,
keyFile string,
through service.Through,
doublet service.Doublet,
) Engine {
return &engine{
Port: port,
CertFile: certFile,
KeyFile: keyFile,
Through: through,
Doublet: doublet,
}
}
func (e *engine) Start(ctx context.Context) error {
router := gin.New()
router.Use(gin.Recovery())
router.Use(gin.LoggerWithConfig(gin.LoggerConfig{
SkipPaths: []string{"/healthz", "/metrics"},
}))
router.Any("/healthz", func(c *gin.Context) {
c.String(http.StatusOK, "OK")
})
router.GET("/metrics", gin.WrapH(promhttp.Handler()))
router.GET("/through", e.HandleThrough)
router.GET("/doublet", e.HandleDoublet)
server := http.Server{
Addr: net.JoinHostPort("", e.Port),
Handler: router,
}
var eg errgroup.Group
eg.Go(func() error {
<-ctx.Done()
return server.Shutdown(context.Background())
})
var err error
if e.CertFile != "" && e.KeyFile != "" {
server.TLSConfig = &tls.Config{
GetCertificate: e.getCertificate,
}
err = server.ListenAndServeTLS("", "")
} else {
err = server.ListenAndServe()
}
if err == http.ErrServerClosed {
return eg.Wait()
}
return err
}
func (e *engine) getCertificate(*tls.ClientHelloInfo) (*tls.Certificate, error) {
cert, err := tls.LoadX509KeyPair(e.CertFile, e.KeyFile)
if err != nil {
return nil, fmt.Errorf("failed to get certificate: %w", err)
}
return &cert, nil
}