mirror of https://github.com/linkerd/linkerd2.git
69 lines
2.0 KiB
Go
69 lines
2.0 KiB
Go
package webhook
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/linkerd/linkerd2/controller/k8s"
|
|
"github.com/linkerd/linkerd2/pkg/admin"
|
|
"github.com/linkerd/linkerd2/pkg/flags"
|
|
"github.com/linkerd/linkerd2/pkg/tls"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// Launch sets up and starts the webhook and metrics servers
|
|
func Launch(config *Config, APIResources []k8s.APIResource, metricsPort uint32, serviceName string, handler handlerFunc) {
|
|
metricsAddr := flag.String("metrics-addr", fmt.Sprintf(":%d", metricsPort), "address to serve scrapable metrics on")
|
|
addr := flag.String("addr", ":8443", "address to serve on")
|
|
kubeconfig := flag.String("kubeconfig", "", "path to kubeconfig")
|
|
controllerNamespace := flag.String("controller-namespace", "linkerd", "namespace in which Linkerd is installed")
|
|
flags.ConfigureAndParse()
|
|
|
|
stop := make(chan os.Signal, 1)
|
|
defer close(stop)
|
|
signal.Notify(stop, os.Interrupt, syscall.SIGTERM)
|
|
|
|
k8sAPI, err := k8s.InitializeAPI(*kubeconfig, APIResources...)
|
|
if err != nil {
|
|
log.Fatalf("failed to initialize Kubernetes API: %s", err)
|
|
}
|
|
|
|
rootCA, err := tls.GenerateRootCAWithDefaults(serviceName)
|
|
if err != nil {
|
|
log.Fatalf("failed to create root CA: %s", err)
|
|
}
|
|
|
|
config.client = k8sAPI.Client.AdmissionregistrationV1beta1()
|
|
config.controllerNamespace = *controllerNamespace
|
|
config.rootCA = rootCA
|
|
|
|
selfLink, err := config.Create()
|
|
if err != nil {
|
|
log.Fatalf("failed to create the webhook configurations resource: %s", err)
|
|
}
|
|
log.Infof("created webhook configuration: %s", selfLink)
|
|
|
|
s, err := NewServer(k8sAPI, *addr, serviceName, *controllerNamespace, rootCA, handler)
|
|
if err != nil {
|
|
log.Fatalf("failed to initialize the webhook server: %s", err)
|
|
}
|
|
|
|
k8sAPI.Sync()
|
|
|
|
go s.Start()
|
|
go admin.StartServer(*metricsAddr)
|
|
|
|
<-stop
|
|
log.Info("shutting down webhook server")
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
if err := s.Shutdown(ctx); err != nil {
|
|
log.Error(err)
|
|
}
|
|
}
|