netbird/netbird-kubeapi-proxy
0
0
Fork 0
mirror of https://github.com/netbirdio/netbird-kubeapi-proxy.git synced 2026-09-23 09:34:58 -07:00
Code Issues Packages Projects Releases Wiki Activity

Add readyz handler for probe server

Browse Source
This commit is contained in:
Philip Laine
2026-06-15 13:47:05 +02:00
parent ea2939c5b0
commit bef808c21a
1 changed files with 25 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main.go
+25 -3
View File
@@ -31,13 +31,15 @@ func main() {
kubeAPIServer string kubeAPIServer string
instanceName string instanceName string
clusterName string clusterName string
probeAddr string
) )
flag.StringVar(&mgmtURL, "management-url", "https://api.netbird.io", "NetBird management URL") flag.StringVar(&mgmtURL, "management-url", "https://api.netbird.io", "NetBird management URL")
flag.StringVar(&apiKey, "api-key", "", "NetBird API key") flag.StringVar(&apiKey, "api-key", "", "NetBird API key")
flag.StringVar(&setupKey, "setup-key", "", "NetBird setup key") flag.StringVar(&setupKey, "setup-key", "", "NetBird setup key")
flag.StringVar(&kubeAPIServer, "kubernetes-api-server", "https://kubernetes.default.svc.cluster.local", "Target Kubernetes API server URL") flag.StringVar(&kubeAPIServer, "kubernetes-api-server", "https://kubernetes.default.svc.cluster.local/", "Target Kubernetes API server URL")
flag.StringVar(&instanceName, "instance-name", "", "Name of the instance") flag.StringVar(&instanceName, "instance-name", "", "Name of the instance")
flag.StringVar(&clusterName, "cluster-name", "", "Name of the cluster") flag.StringVar(&clusterName, "cluster-name", "", "Name of the cluster")
flag.StringVar(&probeAddr, "probe-addr", ":8081", "Address probe server listens to")
flag.Parse() flag.Parse()
logger := slog.New(slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{ logger := slog.New(slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{
@@ -46,14 +48,14 @@ func main() {
})) }))
slog.SetDefault(logger) slog.SetDefault(logger)
err := run(context.Background(), kubeAPIServer, mgmtURL, apiKey, setupKey, instanceName, clusterName) err := run(context.Background(), kubeAPIServer, mgmtURL, apiKey, setupKey, instanceName, clusterName, probeAddr)
if err != nil { if err != nil {
slog.Default().Error("exit due to error", "error", err) slog.Default().Error("exit due to error", "error", err)
os.Exit(1) os.Exit(1)
} }
} }
func run(ctx context.Context, kubeAPIServer, mgmtURL, apiKey, setupKey, instanceName, clusterName string) error { func run(ctx context.Context, kubeAPIServer, mgmtURL, apiKey, setupKey, instanceName, clusterName, probeAddr string) error {
ctx, cancel := signal.NotifyContext(ctx, syscall.SIGTERM) ctx, cancel := signal.NotifyContext(ctx, syscall.SIGTERM)
defer cancel() defer cancel()
g, gCtx := errgroup.WithContext(ctx) g, gCtx := errgroup.WithContext(ctx)
@@ -112,6 +114,26 @@ func run(ctx context.Context, kubeAPIServer, mgmtURL, apiKey, setupKey, instance
return proxySrv.Shutdown(context.Background()) return proxySrv.Shutdown(context.Background())
}) })
probeMux := http.NewServeMux()
probeMux.HandleFunc("/readyz", func(rw http.ResponseWriter, req *http.Request) {
rw.WriteHeader(http.StatusOK)
})
probeSrv := http.Server{
Addr: probeAddr,
Handler: probeMux,
}
g.Go(func() error {
err := probeSrv.ListenAndServe()
if err != nil && !errors.Is(err, http.ErrServerClosed) {
return err
}
return nil
})
g.Go(func() error {
<-gCtx.Done()
return probeSrv.Shutdown(context.Background())
})
slog.Default().Info("running API server proxy") slog.Default().Info("running API server proxy")
err = g.Wait() err = g.Wait()
if err != nil { if err != nil {