Skip to content

Commit

Permalink
feat: always output config reloader file even if application is not r…
Browse files Browse the repository at this point in the history
…eady
  • Loading branch information
TheSpiritXIII committed Jul 15, 2024
1 parent d5dab39 commit 9487d00
Showing 1 changed file with 53 additions and 26 deletions.
79 changes: 53 additions & 26 deletions cmd/config-reloader/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@ import (
"context"
"flag"
"fmt"
"net"
"net/http"
"net/url"
"os"
Expand Down Expand Up @@ -73,6 +74,35 @@ func main() {
collectors.NewProcessCollector(collectors.ProcessCollectorOpts{}),
)

var cfgDirs []reloader.CfgDirOption
if *configDir != "" {
cfgDirs = append(cfgDirs, reloader.CfgDirOption{
Dir: *configDir,
OutputDir: *configDirOutput,
})
}

newReloader := func(reg prometheus.Registerer, reloadURL *url.URL) *reloader.Reloader {
return reloader.New(
logger,
reg,
&reloader.Options{
ReloadURL: reloadURL,
CfgFile: *configFile,
CfgOutputFile: *configFileOutput,
CfgDirs: cfgDirs,
WatchedDirs: watchedDirs,
// There are some reliability issues with fsnotify picking up file changes.
// Configure a very aggress refresh for now. The reloader will only send reload signals
// to Prometheus if the contents actually changed. So this should not have any practical
// drawbacks.
WatchInterval: 10 * time.Second,
RetryInterval: 5 * time.Second,
DelayInterval: 3 * time.Second,
},
)
}

reloadURL, err := url.Parse(*reloadURLStr)
if err != nil {
//nolint:errcheck
Expand Down Expand Up @@ -109,6 +139,26 @@ func main() {
serverErr <- server.ListenAndServe()
}()

// We want to create an output file regardless of whether we startup successfully.
listenAddr, err := net.ResolveTCPAddr("tcp", *listenAddress)
if err != nil {
//nolint:errcheck
level.Error(logger).Log("msg", "parsing listen URL failed", "err", err)
os.Exit(1)
}
// Don't pass a metrics registry, because config-reloader panics when we create our second.
tempReloader := newReloader(nil, &url.URL{
Scheme: "http",
// Since Prometheus is not ready, we won't be able to hit the reload URL so hit itself.
Host: listenAddr.String(),
Path: "/-/healthy",
})
tempReloaderCtx, tempReloaderCancel := context.WithCancel(context.Background())
go func() {
// Ignore errors because we'll re-watch after the application is ready.
_ = tempReloader.Watch(tempReloaderCtx)
}()

// Poll ready endpoint indefinitely until it's up and running.
req, err := http.NewRequest(http.MethodGet, *readyURLStr, nil)
if err != nil {
Expand Down Expand Up @@ -157,35 +207,12 @@ func main() {
}
}
}()

<-done
isReady.Store(true)
tempReloaderCancel()

var cfgDirs []reloader.CfgDirOption
if *configDir != "" {
cfgDirs = append(cfgDirs, reloader.CfgDirOption{
Dir: *configDir,
OutputDir: *configDirOutput,
})
}

rel := reloader.New(
logger,
metrics,
&reloader.Options{
ReloadURL: reloadURL,
CfgFile: *configFile,
CfgOutputFile: *configFileOutput,
CfgDirs: cfgDirs,
WatchedDirs: watchedDirs,
// There are some reliability issues with fsnotify picking up file changes.
// Configure a very aggress refresh for now. The reloader will only send reload signals
// to Prometheus if the contents actually changed. So this should not have any practical
// drawbacks.
WatchInterval: 10 * time.Second,
RetryInterval: 5 * time.Second,
DelayInterval: 3 * time.Second,
},
)
rel := newReloader(metrics, reloadURL)

var g run.Group
{
Expand Down

0 comments on commit 9487d00

Please sign in to comment.