From 5e97b47197b432a66843ceaf43b7a311de2123e2 Mon Sep 17 00:00:00 2001
From: George Shammas <george@shamm.as>
Date: Thu, 10 Jul 2014 18:34:34 +0000
Subject: [PATCH] Only lookup hostname if the results are going to need it.

---
 main.go | 24 ++++++++++++++++++------
 1 file changed, 18 insertions(+), 6 deletions(-)

diff --git a/main.go b/main.go
index 105f9ac..6455e08 100644
--- a/main.go
+++ b/main.go
@@ -40,6 +40,15 @@ func Logger() gin.HandlerFunc {
 	}
 }
 
+func stringInSlice(a string, list []string) bool {
+	for _, b := range list {
+		if b == a {
+			return true
+		}
+	}
+	return false
+}
+
 func mainHandler(c *gin.Context) {
 	fields := strings.Split(c.Params.ByName("field"), ".")
 	ip, err := net.ResolveTCPAddr("tcp", c.Req.RemoteAddr)
@@ -51,16 +60,19 @@ func mainHandler(c *gin.Context) {
 	c.Set("ua", c.Req.UserAgent())
 	c.Set("lang", c.Req.Header.Get("Accept-Language"))
 	c.Set("encoding", c.Req.Header.Get("Accept-Encoding"))
-	c.Set("method",  c.Req.Method)
+	c.Set("method", c.Req.Method)
 	c.Set("mime", c.Req.Header.Get("Accept"))
 	c.Set("referer", c.Req.Header.Get("Referer"))
 	c.Set("forwarded", c.Req.Header.Get("X-Forwarded-For"))
 
-	hostnames, err := net.LookupAddr(ip.IP.String())
-	if err != nil {
-		c.Set("host", "")
-	} else {
-		c.Set("host", hostnames[0])
+	// Only lookup hostname if the results are going to need it.
+	if stringInSlice(fields[0], []string{"", "all", "host"}) {
+		hostnames, err := net.LookupAddr(ip.IP.String())
+		if err != nil {
+			c.Set("host", "")
+		} else {
+			c.Set("host", hostnames[0])
+		}
 	}
 
 	wantsJSON := false