linkerd2/controller/cmd/tap/main.go

77 lines
2.1 KiB
Go

package main
import (
"context"
"crypto/tls"
"flag"
"os"
"os/signal"
"syscall"
"github.com/linkerd/linkerd2/controller/k8s"
"github.com/linkerd/linkerd2/controller/tap"
"github.com/linkerd/linkerd2/pkg/admin"
"github.com/linkerd/linkerd2/pkg/flags"
pkgK8s "github.com/linkerd/linkerd2/pkg/k8s"
log "github.com/sirupsen/logrus"
)
func main() {
apiServerAddr := flag.String("apiserver-addr", ":8089", "address to serve the apiserver on")
metricsAddr := flag.String("metrics-addr", ":9998", "address to serve scrapable metrics on")
kubeConfigPath := flag.String("kubeconfig", "", "path to kube config")
controllerNamespace := flag.String("controller-namespace", "linkerd", "namespace in which Linkerd is installed")
tapPort := flag.Uint("tap-port", 4190, "proxy tap port to connect to")
tlsCertPath := flag.String("tls-cert", pkgK8s.MountPathTLSCrtPEM, "path to TLS Cert PEM")
tlsKeyPath := flag.String("tls-key", pkgK8s.MountPathTLSKeyPEM, "path to TLS Key PEM")
disableCommonNames := flag.Bool("disable-common-names", false, "disable checks for Common Names (for development)")
flags.ConfigureAndParse()
stop := make(chan os.Signal, 1)
signal.Notify(stop, os.Interrupt, syscall.SIGTERM)
k8sAPI, err := k8s.InitializeAPI(
*kubeConfigPath,
k8s.DS,
k8s.SS,
k8s.Deploy,
k8s.Job,
k8s.NS,
k8s.Pod,
k8s.RC,
k8s.Svc,
k8s.RS,
)
if err != nil {
log.Fatalf("Failed to initialize K8s API: %s", err)
}
grpcTapServer := tap.NewGrpcTapServer(*tapPort, *controllerNamespace, k8sAPI)
// TODO: make this configurable for local development
cert, err := tls.LoadX509KeyPair(*tlsCertPath, *tlsKeyPath)
if err != nil {
log.Fatal(err.Error())
}
apiServer, apiLis, err := tap.NewAPIServer(*apiServerAddr, cert, k8sAPI, grpcTapServer, *disableCommonNames)
if err != nil {
log.Fatal(err.Error())
}
k8sAPI.Sync() // blocks until caches are synced
go func() {
log.Infof("starting APIServer on %s", *apiServerAddr)
apiServer.ServeTLS(apiLis, "", "")
}()
go admin.StartServer(*metricsAddr)
<-stop
log.Infof("shutting down APIServer on %s", *apiServerAddr)
apiServer.Shutdown(context.Background())
}