/
main.go
61 lines (49 loc) · 1.44 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
package main
import (
"flag"
"os"
"os/signal"
"syscall"
"github.com/linkerd/linkerd2/controller/destination"
"github.com/linkerd/linkerd2/controller/k8s"
"github.com/linkerd/linkerd2/pkg/admin"
"github.com/linkerd/linkerd2/pkg/flags"
log "github.com/sirupsen/logrus"
)
func main() {
addr := flag.String("addr", "127.0.0.1:8089", "address to serve on")
metricsAddr := flag.String("metrics-addr", ":9999", "address to serve scrapable metrics on")
kubeConfigPath := flag.String("kubeconfig", "", "path to kube config")
k8sDNSZone := flag.String("kubernetes-dns-zone", "", "The DNS suffix for the local Kubernetes zone.")
enableTLS := flag.Bool("enable-tls", false, "Enable TLS connections among pods in the service mesh")
flags.ConfigureAndParse()
stop := make(chan os.Signal, 1)
signal.Notify(stop, os.Interrupt, syscall.SIGTERM)
k8sClient, err := k8s.NewClientSet(*kubeConfigPath)
if err != nil {
log.Fatal(err.Error())
}
k8sAPI := k8s.NewAPI(
k8sClient,
k8s.Endpoint,
k8s.Pod,
k8s.RS,
k8s.Svc,
)
done := make(chan struct{})
ready := make(chan struct{})
server, lis, err := destination.NewServer(*addr, *k8sDNSZone, *enableTLS, k8sAPI, done)
if err != nil {
log.Fatal(err)
}
go k8sAPI.Sync(ready)
go func() {
log.Infof("starting gRPC server on %s", *addr)
server.Serve(lis)
}()
go admin.StartServer(*metricsAddr, ready)
<-stop
log.Infof("shutting down gRPC server on %s", *addr)
close(done)
server.GracefulStop()
}