mirror of https://github.com/linkerd/linkerd2.git
95 lines
2.5 KiB
Go
95 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"os"
|
|
"os/signal"
|
|
"strings"
|
|
"syscall"
|
|
|
|
"github.com/linkerd/linkerd2/controller/k8s"
|
|
"github.com/linkerd/linkerd2/pkg/admin"
|
|
"github.com/linkerd/linkerd2/pkg/flags"
|
|
"github.com/linkerd/linkerd2/pkg/trace"
|
|
api "github.com/linkerd/linkerd2/viz/metrics-api"
|
|
promApi "github.com/prometheus/client_golang/api"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func main() {
|
|
cmd := flag.NewFlagSet("metrics-api", flag.ExitOnError)
|
|
|
|
addr := cmd.String("addr", ":8085", "address to serve on")
|
|
kubeConfigPath := cmd.String("kubeconfig", "", "path to kube config")
|
|
prometheusURL := cmd.String("prometheus-url", "", "prometheus url")
|
|
metricsAddr := cmd.String("metrics-addr", ":9995", "address to serve scrapable metrics on")
|
|
controllerNamespace := cmd.String("controller-namespace", "linkerd", "namespace in which Linkerd is installed")
|
|
ignoredNamespaces := cmd.String("ignore-namespaces", "kube-system", "comma separated list of namespaces to not list pods from")
|
|
clusterDomain := cmd.String("cluster-domain", "cluster.local", "kubernetes cluster domain")
|
|
|
|
traceCollector := flags.AddTraceFlags(cmd)
|
|
|
|
flags.ConfigureAndParse(cmd, os.Args[1:])
|
|
ctx := context.Background()
|
|
|
|
stop := make(chan os.Signal, 1)
|
|
signal.Notify(stop, os.Interrupt, syscall.SIGTERM)
|
|
|
|
k8sAPI, err := k8s.InitializeAPI(
|
|
ctx,
|
|
*kubeConfigPath,
|
|
true,
|
|
k8s.CJ, k8s.DS, k8s.Deploy, k8s.Job, k8s.NS, k8s.Pod, k8s.RC, k8s.RS, k8s.Svc, k8s.SS, k8s.SP,
|
|
)
|
|
if err != nil {
|
|
log.Fatalf("Failed to initialize K8s API: %s", err)
|
|
}
|
|
|
|
var prometheusClient promApi.Client
|
|
if *prometheusURL != "" {
|
|
prometheusClient, err = promApi.NewClient(promApi.Config{Address: *prometheusURL})
|
|
if err != nil {
|
|
log.Fatal(err.Error())
|
|
}
|
|
}
|
|
|
|
log.Infof("prometheusClient: %#v", prometheusClient)
|
|
log.Info("Using cluster domain: ", *clusterDomain)
|
|
|
|
if *traceCollector != "" {
|
|
if err := trace.InitializeTracing("linkerd-public-api", *traceCollector); err != nil {
|
|
log.Warnf("failed to initialize tracing: %s", err)
|
|
}
|
|
}
|
|
|
|
server := api.NewServer(
|
|
*addr,
|
|
prometheusClient,
|
|
k8sAPI,
|
|
*controllerNamespace,
|
|
*clusterDomain,
|
|
strings.Split(*ignoredNamespaces, ","),
|
|
)
|
|
|
|
k8sAPI.Sync(nil) // blocks until caches are synced
|
|
|
|
go func() {
|
|
log.Infof("starting HTTP server on %+v", *addr)
|
|
server.ListenAndServe()
|
|
}()
|
|
|
|
adminServer := admin.NewServer(*metricsAddr)
|
|
|
|
go func() {
|
|
log.Infof("starting admin server on %s", *metricsAddr)
|
|
adminServer.ListenAndServe()
|
|
}()
|
|
|
|
<-stop
|
|
|
|
log.Infof("shutting down HTTP server on %+v", *addr)
|
|
server.Shutdown(ctx)
|
|
adminServer.Shutdown(ctx)
|
|
}
|