knowledge-base/traefik-whoami
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add access logs

Browse Source
This commit is contained in:
Pierre-Yves Aillet
2022-02-13 15:50:07 +01:00
committed by GitHub
parent 36f93ee1a6
commit c88402491b
2 changed files with 34 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
README.md
View File

@ -44,12 +44,13 @@ Heath check.
### Flags ### Flags
| Flag | Env var | Description | | Flag | Env var | Description |
|----------|----------------------|-----------------------------------------| |-----------|----------------------|-----------------------------------------|
| `cert` | | give me a certificate. | | `cert` | | Give me a certificate. |
| `key` | | give me a key. | | `key` | | Give me a key. |
| `cacert` | | give me a CA chain, enforces mutual TLS | | `cacert` | | Give me a CA chain, enforces mutual TLS |
| `port` | `WHOAMI_PORT_NUMBER` | give me a port number. (default: `80`) | | `port` | `WHOAMI_PORT_NUMBER` | Give me a port number. (default: `80`) |
| `name` | `WHOAMI_NAME` | give me a name. | | `name` | `WHOAMI_NAME` | Give me a name. |
| `verbose` | | Enable verbose logging. |
## Examples ## Examples

27
app.go
View File

@ -37,9 +37,11 @@ var (
ca string ca string
port string port string
name string name string
verbose bool
) )
func init() { func init() {
flag.BoolVar(&verbose, "verbose", false, "Enable verbose logging")
flag.StringVar(&cert, "cert", "", "give me a certificate") flag.StringVar(&cert, "cert", "", "give me a certificate")
flag.StringVar(&key, "key", "", "give me a key") flag.StringVar(&key, "key", "", "give me a key")
flag.StringVar(&ca, "cacert", "", "give me a CA chain, enforces mutual TLS") flag.StringVar(&ca, "cacert", "", "give me a CA chain, enforces mutual TLS")
@ -56,12 +58,12 @@ func main() {
flag.Parse() flag.Parse()
mux := http.NewServeMux() mux := http.NewServeMux()
mux.HandleFunc("/data", dataHandler) mux.Handle("/data", handle(dataHandler, verbose))
mux.HandleFunc("/echo", echoHandler) mux.Handle("/echo", handle(echoHandler, verbose))
mux.HandleFunc("/bench", benchHandler) mux.Handle("/bench", handle(benchHandler, verbose))
mux.HandleFunc("/", whoamiHandler) mux.Handle("/api", handle(apiHandler, verbose))
mux.HandleFunc("/api", apiHandler) mux.Handle("/health", handle(healthHandler, verbose))
mux.HandleFunc("/health", healthHandler) mux.Handle("/", handle(whoamiHandler, verbose))
if cert == "" || key == "" { if cert == "" || key == "" {
log.Printf("Starting up on port %s", port) log.Printf("Starting up on port %s", port)
@ -102,6 +104,19 @@ func setupMutualTLS(ca string) *tls.Config {
return tlsConfig return tlsConfig
} }
func handle(next http.HandlerFunc, verbose bool) http.Handler {
if !verbose {
return next
}
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
next(w, r)
// <remote_IP_address> - [<timestamp>] "<request_method> <request_path> <request_protocol>" -
log.Printf("%s - - [%s] \"%s %s %s\" - -", r.RemoteAddr, time.Now().Format("02/Jan/2006:15:04:05 -0700"), r.Method, r.URL.Path, r.Proto)
})
}
func benchHandler(w http.ResponseWriter, _ *http.Request) { func benchHandler(w http.ResponseWriter, _ *http.Request) {
w.Header().Set("Connection", "keep-alive") w.Header().Set("Connection", "keep-alive")
w.Header().Set("Content-Type", "text/plain") w.Header().Set("Content-Type", "text/plain")