-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
101 lines (85 loc) · 2 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
package main
import (
"context"
"github.com/caarlos0/env"
"github.com/gin-gonic/gin"
"github.com/h3adex/guardgress/pkg/healthmetrics"
"github.com/h3adex/guardgress/pkg/server"
"github.com/h3adex/guardgress/pkg/watcher"
log "github.com/sirupsen/logrus"
"golang.org/x/sync/errgroup"
"k8s.io/client-go/kubernetes"
"k8s.io/client-go/rest"
"k8s.io/client-go/tools/clientcmd"
"os"
"path/filepath"
)
func init() {
logLevel, ok := os.LookupEnv("LOG_LEVEL")
// LOG_LEVEL not set, let's default to info
if !ok {
logLevel = "info"
}
level, err := log.ParseLevel(logLevel)
if err != nil {
log.Error("failed to parse LOG_LEVEL, defaulting to info")
log.SetLevel(log.InfoLevel)
return
}
if log.GetLevel() != log.DebugLevel {
gin.SetMode(gin.ReleaseMode)
}
log.SetLevel(level)
}
func main() {
k8sClient, err := kubernetes.NewForConfig(getKubernetesConfig())
if err != nil {
log.Panic(err)
}
config := &server.Config{}
if err = env.Parse(config); err != nil {
log.Panic(err)
}
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
eg, ctx := errgroup.WithContext(ctx)
srv := server.New(config)
// Start reverse proxy port 80/443
eg.Go(func() error {
return srv.Run(ctx)
})
// Start watcher for k8s resources
eg.Go(func() error {
return watcher.New(
k8sClient,
func(routingTable watcher.Payload) {
srv.UpdateRoutingTable(routingTable)
},
).Run(ctx)
})
// start health/metric server
eg.Go(func() error {
return healthmetrics.New().Run(ctx)
})
if err = eg.Wait(); err != nil {
log.Error("Error on watcher or server goroutine: ", err.Error())
log.Panic(err)
}
}
func getKubernetesConfig() *rest.Config {
config, err := rest.InClusterConfig()
if err != nil {
config, err = clientcmd.BuildConfigFromFlags("", filepath.Join(getHomeDir(), ".kube", "config"))
}
if err != nil {
log.Fatalln(err)
}
return config
}
func getHomeDir() string {
if home := os.Getenv("HOME"); home != "" {
return home
}
// windows
return os.Getenv("USERPROFILE")
}