/
main.go
43 lines (35 loc) · 1.07 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
package main
import (
"flag"
"log"
"net/http"
"os"
"strings"
"github.com/prometheus/client_golang/prometheus/promhttp"
"go.uber.org/zap/zapcore"
)
var addr = flag.String("addr", ":8080", "listen address(default :8080)")
var username = flag.String("username", "", "Basic Auth Username")
var password = flag.String("password", "", "Basic Auth Passowrd")
func main() {
// parse environment variable args.
flag.VisitAll(func(f *flag.Flag) {
if s := os.Getenv(strings.ToUpper(f.Name)); s != "" {
f.Value.Set(s)
}
})
flag.Parse()
h := NewHandler(os.Stdout, zapcore.InfoLevel)
defer h.Logger.Sync()
mux := http.NewServeMux()
mux.Handle("/metrics", promhttp.Handler())
mux.HandleFunc("/healthz", h.GetStatus)
mux.HandleFunc("/time", h.GetTime)
mux.HandleFunc("/sleep", h.GetSleep)
// Required BasicAuth Handler
if *username != "" || *password != "" {
mux.HandleFunc("/ip", h.BasicAuthMiddleware(h.GetRemoteIP, *username, *password))
mux.HandleFunc("/env", h.BasicAuthMiddleware(h.GetEnv, *username, *password))
}
log.Fatal(http.ListenAndServe(*addr, mux))
}