DevFW-CICD/ingress-nginx-helm
16
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Refactor and rename custom error pages vars

Browse source 
Signed-off-by: Ricardo Lopes <ricardoapl.dev@gmail.com>
This commit is contained in:
Ricardo Lopes 2024-02-14 11:21:14 +00:00
parent b502b5fcdc
commit 20e86ed27a
1 changed files with 22 additions and 31 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

45
images/custom-error-pages/rootfs/main.go
View file

@ -69,13 +69,13 @@ const (
// cannot be mapped to a file extension.
DefaultFormatVar = "DEFAULT_RESPONSE_FORMAT"
// IsExposeSignalsVar is the name of the environment variable indicating
// whether or not to expose signals such as /metrics and /healthz.
IsExposeSignalsVar = "IS_EXPOSE_SIGNALS"
// IsExportMetricsVar is the name of the environment variable indicating
// whether or not to export /metrics and /healthz.
IsExportMetricsVar = "IS_EXPORT_METRICS"
// ExposeSignalsPortVar is the name of the environment variable indicating
// the port on which to expose signals such as /metrics and /healthz.
ExposeSignalsPortVar = "EXPOSE_SIGNALS_PORT"
// MetricsPortVar is the name of the environment variable indicating
// the port on which to export /metrics and /healthz.
MetricsPortVar = "METRICS_PORT"
// CustomErrorPagesPort is the port on which to listen to serve custom error pages.
CustomErrorPagesPort = "8080"
@ -97,60 +97,51 @@ func main() {
defaultFormat = os.Getenv(DefaultFormatVar)
}
isExposeSignals := true
if os.Getenv(IsExposeSignalsVar) != "" {
val, err := strconv.ParseBool(os.Getenv(IsExposeSignalsVar))
isExportMetrics := true
if os.Getenv(IsExportMetricsVar) != "" {
val, err := strconv.ParseBool(os.Getenv(IsExportMetricsVar))
if err == nil {
isExposeSignals = val
isExportMetrics = val
}
}
exposeSignalsPort := "8080"
if os.Getenv(ExposeSignalsPortVar) != "" {
exposeSignalsPort = os.Getenv(ExposeSignalsPortVar)
metricsPort := "8080"
if os.Getenv(MetricsPortVar) != "" {
metricsPort = os.Getenv(MetricsPortVar)
}
var mux *http.ServeMux
if isExposeSignals {
if exposeSignalsPort == CustomErrorPagesPort {
if isExportMetrics {
if metricsPort == CustomErrorPagesPort {
mux = http.DefaultServeMux
mux.HandleFunc("/", errorHandler(errFilesPath, defaultFormat))
mux.Handle("/metrics", promhttp.Handler())
mux.HandleFunc("/healthz", func(w http.ResponseWriter, r *http.Request) {
w.WriteHeader(http.StatusOK)
})
err := http.ListenAndServe(fmt.Sprintf(":%s", CustomErrorPagesPort), mux)
if err != nil {
panic(err)
}
} else {
mux = http.NewServeMux()
mux.HandleFunc("/", errorHandler(errFilesPath, defaultFormat))
http.Handle("/metrics", promhttp.Handler())
http.HandleFunc("/healthz", func(w http.ResponseWriter, r *http.Request) {
w.WriteHeader(http.StatusOK)
})
go func() {
err := http.ListenAndServe(fmt.Sprintf(":%s", CustomErrorPagesPort), mux)
err := http.ListenAndServe(fmt.Sprintf(":%s", metricsPort), nil)
if err != nil {
panic(err)
}
}()
err := http.ListenAndServe(fmt.Sprintf(":%s", exposeSignalsPort), nil)
if err != nil {
panic(err)
}
}
} else {
// Use a new ServerMux because expvar HTTP handler registers itself against DefaultServerMux
// as a consequence of importing it in client_golang/prometheus.
mux = http.NewServeMux()
}
mux.HandleFunc("/", errorHandler(errFilesPath, defaultFormat))
err := http.ListenAndServe(fmt.Sprintf(":%s", CustomErrorPagesPort), mux)
if err != nil {
panic(err)
}
}
}
func errorHandler(path, defaultFormat string) func(http.ResponseWriter, *http.Request) {