Support CPU and memory profiling; log number of goroutines every minute

geodns.go

@@ -5,6 +5,7 @@ import (
 	"log"
 	"os"
 	"os/signal"
+	"runtime/pprof"
 	"time"
 )
 
@@ -17,6 +18,9 @@ var (
 	listen  = flag.String("listen", ":8053", "set the listener address")
 	flaglog = flag.Bool("log", false, "be more verbose")
 	flagrun = flag.Bool("run", false, "run server")
+
+	cpuprofile = flag.String("cpuprofile", "", "write cpu profile to file")
+	memprofile = flag.String("memprofile", "", "write memory profile to this file")
 )
 
 func main() {
@@ -27,6 +31,25 @@ func main() {
 
 	log.Printf("Starting geodns/%s\n", VERSION)
 
+	if *cpuprofile != "" {
+		prof, err := os.Create(*cpuprofile)
+		if err != nil {
+			panic(err.Error())
+		}
+
+		pprof.StartCPUProfile(prof)
+		defer func() {
+			log.Println("closing file")
+			prof.Close()
+		}()
+		defer func() {
+			log.Println("stopping profile")
+			pprof.StopCPUProfile()
+		}()
+	}
+
+	go monitor()
+
 	dirName := "dns"
 
 	Zones := make(Zones)
@@ -37,11 +60,21 @@ func main() {
 	go listenAndServe(&Zones)
 
 	if *flagrun {
-		sig := make(chan os.Signal)
-		signal.Notify(sig, os.Interrupt)
+		terminate := make(chan os.Signal)
+		signal.Notify(terminate, os.Interrupt)
 
-		<-sig
+		<-terminate
 		log.Printf("geodns: signal received, stopping")
-		os.Exit(0)
+
+		if *memprofile != "" {
+			f, err := os.Create(*memprofile)
+			if err != nil {
+				log.Fatal(err)
+			}
+			pprof.WriteHeapProfile(f)
+			f.Close()
+		}
+
+		//os.Exit(0)
 	}
 }

monitor.go

@@ -0,0 +1,13 @@
+package main
+
+import (
+	"runtime"
+	"time"
+)
+
+func monitor() {
+	for {
+		logPrintln("goroutines", runtime.NumGoroutine())
+		time.Sleep( 60 * time.Second)
+	}
+}