Add pprof

2 files changed, 17 insertions, 0 deletions

diff --git a/README.md b/README.md
index 4f286fb..fa09904 100644
--- a/README.md
+++ b/README.md
@@ -72,6 +72,10 @@ It is good idea to obtain Telegram API ID from [**https://my.telegram.org**](htt
     ...
 ```
 
+### Arguments ###
+
+* `--profiling-port=xxxx`: starts the pprof server on port `xxxx`. Access is limited to localhost.
+
 ### How to receive files from Telegram ###
 
 First of all, you need to set up web server that will serve some directory in your filesystem.
diff --git a/telegabber.go b/telegabber.go
index 2752163..14b3b3d 100644
--- a/telegabber.go
+++ b/telegabber.go
@@ -1,6 +1,10 @@
 package main
 
 import (
+	"flag"
+	"fmt"
+	"net/http"
+	_ "net/http/pprof"
 	"os"
 	"os/signal"
 
@@ -25,6 +29,15 @@ var cleanupDone chan struct{}
 var sigintChannel chan os.Signal
 
 func main() {
+	var profilingPort = flag.Int("profiling-port", 0, "The port for pprof server")
+	flag.Parse()
+
+	if *profilingPort > 0 {
+		go func() {
+			log.Println(http.ListenAndServe(fmt.Sprintf("localhost:%v", *profilingPort), nil))
+		}()
+	}
+
 	cleanupDone = make(chan struct{})
 	sigintChannel = make(chan os.Signal, 1)
 	signal.Notify(sigintChannel, os.Interrupt)