linkerd2/controller/cmd/proxy-api/main.go

71 lines
1.8 KiB
Go

package main
import (
"flag"
"net/http"
"os"
"os/signal"
"syscall"
"github.com/prometheus/client_golang/prometheus/promhttp"
"github.com/runconduit/conduit/controller/api/proxy"
"github.com/runconduit/conduit/controller/destination"
"github.com/runconduit/conduit/pkg/version"
log "github.com/sirupsen/logrus"
)
func main() {
addr := flag.String("addr", ":8086", "address to serve on")
metricsAddr := flag.String("metrics-addr", ":9996", "address to serve scrapable metrics on")
telemetryAddr := flag.String("telemetry-addr", ":8087", "address of telemetry service")
destinationAddr := flag.String("destination-addr", ":8089", "address of destination service")
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)
telemetryClient, conn, err := proxy.NewTelemetryClient(*telemetryAddr)
if err != nil {
log.Fatal(err)
}
defer conn.Close()
destinationClient, conn, err := destination.NewClient(*destinationAddr)
if err != nil {
log.Fatal(err)
}
defer conn.Close()
server, lis, err := proxy.NewServer(*addr, telemetryClient, destinationClient)
if err != nil {
log.Fatal(err)
}
go func() {
log.Infof("starting gRPC server on %s", *addr)
server.Serve(lis)
}()
go func() {
log.Infof("serving scrapable metrics on %s", *metricsAddr)
http.Handle("/metrics", promhttp.Handler())
http.ListenAndServe(*metricsAddr, nil)
}()
<-stop
log.Infof("shutting down gRPC server on %s", *addr)
server.GracefulStop()
}