fbonalair/traefik-crowdsec-bouncer

package main

import (
    "fmt"
    "log"
    "net/http"
    "os"
    "time"
)

/**
  Simple binary to query bouncer health check route and allow use of docker container health check
  For more information, see issue https://github.com/fbonalair/traefik-crowdsec-bouncer/issues/6
*/
func main() {
    port := os.Getenv("PORT")
    timeoutString := os.Getenv("HEALTH_CHECKER_TIMEOUT_DURATION")
    if port == "" {
        port = "8080"
    }
    if timeoutString == "" {
        timeoutString = "2s"
    }

    // Calling bouncer health check
    healthCheckUrl := fmt.Sprintf("http://127.0.0.1:%s/api/v1/ping", port)
    duration, err := time.ParseDuration(timeoutString)
    if err != nil {
        log.Fatal("error while parsing HEALTH_CHECKER_TIMEOUT_DURATION value to duration: ", err)
    }
    httpClient := http.Client{Timeout: duration}
    resp, err := httpClient.Get(healthCheckUrl)
    if err != nil {
        log.Fatal("error while requesting bouncer's health check route: ", err)
    }

    if resp.StatusCode == http.StatusOK {
        os.Exit(0)
    }

    os.Exit(1)
}