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

Merge pull request #18 from netbirdio/feature/readyz

Browse Source 
Add readyz handler and only expose specific paths from API server
This commit is contained in:
Philip Laine
2026-06-15 13:52:35 +02:00
committed by GitHub
parent ea2939c5b0 bef808c21a
commit 6e83692ad8
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
instanceName string
clusterName string
probeAddr string
)
flag.StringVar(&mgmtURL, "management-url", "https://api.netbird.io", "NetBird management URL")
flag.StringVar(&apiKey, "api-key", "", "NetBird API 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(&clusterName, "cluster-name", "", "Name of the cluster")
flag.StringVar(&probeAddr, "probe-addr", ":8081", "Address probe server listens to")
flag.Parse()
logger := slog.New(slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{
@@ -46,14 +48,14 @@ func main() {
}))
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 {
slog.Default().Error("exit due to error", "error", err)
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)
defer cancel()
g, gCtx := errgroup.WithContext(ctx)
@@ -112,6 +114,26 @@ func run(ctx context.Context, kubeAPIServer, mgmtURL, apiKey, setupKey, instance
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")
err = g.Wait()
if err != nil {