mirror of https://github.com/linkerd/linkerd2.git
147 lines
4.0 KiB
Go
147 lines
4.0 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/runconduit/conduit/controller/k8s"
|
|
"github.com/runconduit/conduit/controller/tap"
|
|
"github.com/runconduit/conduit/controller/util"
|
|
"github.com/runconduit/conduit/pkg/version"
|
|
log "github.com/sirupsen/logrus"
|
|
"k8s.io/api/core/v1"
|
|
"k8s.io/client-go/informers"
|
|
"k8s.io/client-go/tools/cache"
|
|
)
|
|
|
|
func main() {
|
|
addr := flag.String("addr", "127.0.0.1:8088", "address to serve on")
|
|
metricsAddr := flag.String("metrics-addr", ":9998", "address to serve scrapable metrics on")
|
|
kubeConfigPath := flag.String("kubeconfig", "", "path to kube config")
|
|
tapPort := flag.Uint("tap-port", 4190, "proxy tap port to connect to")
|
|
logLevel := flag.String("log-level", log.InfoLevel.String(), "log level, must be one of: panic, fatal, error, warn, info, debug")
|
|
printVersion := version.VersionFlag()
|
|
flag.Parse()
|
|
|
|
// set global log level
|
|
level, err := log.ParseLevel(*logLevel)
|
|
if err != nil {
|
|
log.Fatalf("invalid log-level: %s", *logLevel)
|
|
}
|
|
log.SetLevel(level)
|
|
|
|
version.MaybePrintVersionAndExit(*printVersion)
|
|
|
|
stop := make(chan os.Signal, 1)
|
|
signal.Notify(stop, os.Interrupt, syscall.SIGTERM)
|
|
|
|
clientSet, err := k8s.NewClientSet(*kubeConfigPath)
|
|
if err != nil {
|
|
log.Fatalf("failed to create Kubernetes client: %s", err)
|
|
}
|
|
|
|
replicaSets, err := k8s.NewReplicaSetStore(clientSet)
|
|
if err != nil {
|
|
log.Fatalf("NewReplicaSetStore failed: %s", err)
|
|
}
|
|
err = replicaSets.Run()
|
|
if err != nil {
|
|
log.Fatalf("replicaSets.Run() failed: %s", err)
|
|
}
|
|
|
|
// index pods by deployment
|
|
deploymentIndex := func(obj interface{}) ([]string, error) {
|
|
pod, ok := obj.(*v1.Pod)
|
|
if !ok {
|
|
return nil, fmt.Errorf("object is not a Pod")
|
|
}
|
|
deployment, err := replicaSets.GetDeploymentForPod(pod)
|
|
if err != nil {
|
|
log.Debugf("Cannot get deployment for pod %s: %s", pod.Name, err)
|
|
return []string{}, nil
|
|
}
|
|
return []string{deployment}, nil
|
|
}
|
|
|
|
pods, err := k8s.NewPodIndex(clientSet, deploymentIndex)
|
|
if err != nil {
|
|
log.Fatalf("NewPodIndex failed: %s", err)
|
|
}
|
|
err = pods.Run()
|
|
if err != nil {
|
|
log.Fatalf("pods.Run() failed: %s", err)
|
|
}
|
|
|
|
// TODO: factor out with public-api
|
|
sharedInformers := informers.NewSharedInformerFactory(clientSet, 10*time.Minute)
|
|
|
|
namespaceInformer := sharedInformers.Core().V1().Namespaces()
|
|
namespaceInformerSynced := namespaceInformer.Informer().HasSynced
|
|
|
|
deployInformer := sharedInformers.Apps().V1beta2().Deployments()
|
|
deployInformerSynced := deployInformer.Informer().HasSynced
|
|
|
|
replicaSetInformer := sharedInformers.Apps().V1beta2().ReplicaSets()
|
|
replicaSetInformerSynced := replicaSetInformer.Informer().HasSynced
|
|
|
|
podInformer := sharedInformers.Core().V1().Pods()
|
|
podInformerSynced := podInformer.Informer().HasSynced
|
|
|
|
replicationControllerInformer := sharedInformers.Core().V1().ReplicationControllers()
|
|
replicationControllerInformerSynced := replicationControllerInformer.Informer().HasSynced
|
|
|
|
serviceInformer := sharedInformers.Core().V1().Services()
|
|
serviceInformerSynced := serviceInformer.Informer().HasSynced
|
|
|
|
sharedInformers.Start(nil)
|
|
|
|
server, lis, err := tap.NewServer(
|
|
*addr, *tapPort, replicaSets, pods,
|
|
namespaceInformer.Lister(),
|
|
deployInformer.Lister(),
|
|
replicaSetInformer.Lister(),
|
|
podInformer.Lister(),
|
|
replicationControllerInformer.Lister(),
|
|
serviceInformer.Lister(),
|
|
)
|
|
if err != nil {
|
|
log.Fatal(err.Error())
|
|
}
|
|
|
|
go func() {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)
|
|
defer cancel()
|
|
|
|
log.Infof("waiting for caches to sync")
|
|
if !cache.WaitForCacheSync(
|
|
ctx.Done(),
|
|
namespaceInformerSynced,
|
|
deployInformerSynced,
|
|
replicaSetInformerSynced,
|
|
podInformerSynced,
|
|
replicationControllerInformerSynced,
|
|
serviceInformerSynced,
|
|
) {
|
|
log.Fatalf("timed out wait for caches to sync")
|
|
}
|
|
log.Infof("caches synced")
|
|
}()
|
|
|
|
go func() {
|
|
log.Println("starting gRPC server on", *addr)
|
|
server.Serve(lis)
|
|
}()
|
|
|
|
go util.NewMetricsServer(*metricsAddr)
|
|
|
|
<-stop
|
|
|
|
log.Println("shutting down gRPC server on", *addr)
|
|
server.GracefulStop()
|
|
}
|