From 0f7e13222d087d85c1afd67d22d36f544a39801d Mon Sep 17 00:00:00 2001 From: Julien Dessaux Date: Sat, 28 Sep 2024 09:45:03 +0200 Subject: feat(tfstate): bootstrap an http server that answers /healthz --- cmd/tfstate/healthz.go | 12 +++++++ cmd/tfstate/main.go | 93 ++++++++++++++++++++++++++++++++++++++++++++++++++ cmd/tfstate/routes.go | 7 ++++ 3 files changed, 112 insertions(+) create mode 100644 cmd/tfstate/healthz.go create mode 100644 cmd/tfstate/main.go create mode 100644 cmd/tfstate/routes.go (limited to 'cmd') diff --git a/cmd/tfstate/healthz.go b/cmd/tfstate/healthz.go new file mode 100644 index 0000000..20c72c9 --- /dev/null +++ b/cmd/tfstate/healthz.go @@ -0,0 +1,12 @@ +package main + +import "net/http" + +func handleHealthz() http.Handler { + return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { + w.Header().Set("Cache-Control", "no-store, no-cache") + w.Header().Set("Content-Type", "application/json") + w.WriteHeader(http.StatusOK) + _, _ = w.Write([]byte("{}")) + }) +} diff --git a/cmd/tfstate/main.go b/cmd/tfstate/main.go new file mode 100644 index 0000000..e0b33a5 --- /dev/null +++ b/cmd/tfstate/main.go @@ -0,0 +1,93 @@ +package main + +import ( + "context" + "fmt" + "io" + "log" + "log/slog" + "net" + "net/http" + "os" + "os/signal" + "sync" + "time" +) + +type Config struct { + Host string + Port string +} + +func run( + ctx context.Context, + config *Config, + args []string, + getenv func(string) string, + stdin io.Reader, + stdout, stderr io.Writer, +) error { + ctx, cancel := signal.NotifyContext(ctx, os.Interrupt) + defer cancel() + + mux := http.NewServeMux() + addRoutes( + mux, + ) + httpServer := &http.Server{ + Addr: net.JoinHostPort(config.Host, config.Port), + Handler: mux, + } + go func() { + log.Printf("listening on %s\n", httpServer.Addr) + if err := httpServer.ListenAndServe(); err != nil && err != http.ErrServerClosed { + fmt.Fprintf(os.Stderr, "error listening and serving: %+v\n", err) + } + }() + var wg sync.WaitGroup + wg.Add(1) + go func() { + defer wg.Done() + <-ctx.Done() + shutdownCtx := context.Background() + shutdownCtx, cancel := context.WithTimeout(shutdownCtx, 10*time.Second) + defer cancel() + if err := httpServer.Shutdown(shutdownCtx); err != nil { + fmt.Fprintf(os.Stderr, "error shutting down http server: %+v\n", err) + } + }() + wg.Wait() + + return nil +} + +func main() { + ctx := context.Background() + + var opts *slog.HandlerOptions + if os.Getenv("TFSTATE_DEBUG") != "" { + opts = &slog.HandlerOptions{ + AddSource: true, + Level: slog.LevelDebug, + } + } + logger := slog.New(slog.NewJSONHandler(os.Stdout, opts)) + slog.SetDefault(logger) + + config := Config{ + Host: "0.0.0.0", + Port: "8080", + } + + if err := run( + ctx, + &config, + os.Args, + os.Getenv, + os.Stdin, + os.Stdout, os.Stderr, + ); err != nil { + fmt.Fprintf(os.Stderr, "%+v\n", err) + os.Exit(1) + } +} diff --git a/cmd/tfstate/routes.go b/cmd/tfstate/routes.go new file mode 100644 index 0000000..f583fc7 --- /dev/null +++ b/cmd/tfstate/routes.go @@ -0,0 +1,7 @@ +package main + +import "net/http" + +func addRoutes(mux *http.ServeMux) { + mux.Handle("GET /healthz", handleHealthz()) +} -- cgit v1.2.3