Add profiling (and flag for it)

.gitignore

@@ -1,2 +1,3 @@
 *sublime*
-/gomatrix
+/gomatrix
+/gomatrix.prof

main.go

@@ -9,6 +9,7 @@ import (
 	"math/rand"
 	"os"
 	"os/signal"
+	"runtime/pprof"
 	"time"
 )
 
@@ -19,6 +20,9 @@ var opts struct {
 
 	// enable logging
 	Logging bool `short:"l" long:"log" description:"Enable logging debug messages to ~/.gomatrix-log."`
+
+	// enable profiling
+	Profile string `short:"p" long:"profile" description:"Write profile to given file path"`
 }
 
 // array with half width kanas as Go runes
@@ -85,6 +89,17 @@ func main() {
 		return
 	}
 
+	// Start profiling (if required)
+	if len(opts.Profile) > 0 {
+		f, err := os.Create(opts.Profile)
+		if err != nil {
+			fmt.Printf("Error opening profiling file: %s\n", err)
+			os.Exit(1)
+		}
+		pprof.StartCPUProfile(f)
+		defer pprof.StopCPUProfile()
+	}
+
 	// Juse a println for fun..
 	fmt.Println("Opening connection to The Matrix.. Please stand by..")
 
@@ -245,5 +260,4 @@ func main() {
 	termbox.Close()
 	log.Println("stopping gomatrix")
 	fmt.Println("Thank you for connecting with Morpheus' Matrix API v4.2. Have a nice day!")
-	os.Exit(0)
 }