commit 73fd73f5b922bacc2a0f245edfe807e83a9d80bc
parent fcc0be24867a6f9ba3e6472de00aedb6e68bc299
Author: Frederic Cambus <fred@statdns.com>
Date: Fri, 3 Feb 2017 22:05:18 +0100
Display mode (HTTP or FastCGI) and refactor host + port concatenation
Diffstat:
1 file changed, 9 insertions(+), 3 deletions(-)
diff --git a/rrda.go b/rrda.go
@@ -167,6 +167,7 @@ func main() {
fastcgi := flag.Bool("fastcgi", false, "Enable FastCGI mode")
host := flag.String("host", "127.0.0.1", "Set the server host")
port := flag.String("port", "8080", "Set the server port")
+ mode := "HTTP"
flag.Usage = func() {
fmt.Println("\nUSAGE:")
@@ -174,22 +175,27 @@ func main() {
}
flag.Parse()
+ address := *host + ":" + *port;
- fmt.Println("\nListening on :", *host+":"+*port)
+ if *fastcgi {
+ mode = "FastCGI"
+ }
+
+ fmt.Println("Listening on (" + mode + " mode):", address)
m := pat.New()
m.Get("/:server/x/:ip", http.HandlerFunc(ptr))
m.Get("/:server/:domain/:querytype", http.HandlerFunc(query))
if *fastcgi {
- listener, _ := net.Listen("tcp", *host+":"+*port)
+ listener, _ := net.Listen("tcp", address)
if err := fcgi.Serve(listener, m); err != nil {
fmt.Println("\nERROR:", err)
os.Exit(1)
}
} else {
- if err := http.ListenAndServe(*host+":"+*port, m); err != nil {
+ if err := http.ListenAndServe(address, m); err != nil {
fmt.Println("\nERROR:", err)
os.Exit(1)
}