improved logging

1 files changed, 12 insertions, 0 deletions

diff --git a/whois.go b/whois.go
index 3d75477..751a5d1 100644
--- a/whois.go
+++ b/whois.go
@@ -2,7 +2,9 @@ package main
 
 import (
 	"io/ioutil"
+	"log"
 	"net"
+	"os"
 	"strings"
 	"time"
 )
@@ -14,6 +16,14 @@ const (
 	RETRIES      = 2
 )
 
+var (
+	log *log.Logger
+)
+
+func init() {
+	logger = log.New(os.Stdout, "[ Whois ]", log.LstdFlags)
+}
+
 /*
 func main() {
 	pairs, err := findIPBlock("130.229.137.105", "192.88.99.1")
@@ -25,6 +35,7 @@ func main() {
 */
 
 func findIPBlock(domains ...string) (pairs map[string]string, err error) {
+	logger.Println("Querying for ip-blocks...")
 	if len(domains) == 0 {
 		return
 	}
@@ -50,6 +61,7 @@ func findIPBlock(domains ...string) (pairs map[string]string, err error) {
 		ipb := strings.TrimSpace(content[2])
 		pairs[ipaddr] = ipb
 	}
+	logger.Println("ip-blocks returned")
 	return
 }