Remove memprofile file, add pprof HTTP endpoint instead.

10 years ago · a730cff002
1 changed files with 15 additions and 33 deletions
--- a/node_exporter.go
+++ b/node_exporter.go
@ -6,9 +6,9 @@ import (
 	"fmt"
 	"io/ioutil"
 	"net/http"
+	_ "net/http/pprof"
 	"os"
 	"os/signal"
-	"runtime/pprof"
 	"strings"
 	"sync"
 	"syscall"
@ -171,7 +171,7 @@ func main() {
 			password: *authPass,
 		}
 	}
-	go func() {
+
 	http.Handle(*metricsPath, handler)
 	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
 		w.Write([]byte(`<html>
@ -182,26 +182,8 @@ func main() {
 			</body>
 			</html>`))
 	})
-		err := http.ListenAndServe(*listenAddress, nil)
-		if err != nil {
-			glog.Fatal(err)
-		}
-	}()
-
-	for {
-		select {
-		case <-sigUsr1:
-			glog.Infof("got signal")
-			if *memProfile != "" {
-				glog.Infof("Writing memory profile to %s", *memProfile)
-				f, err := os.Create(*memProfile)
+	err = http.ListenAndServe(*listenAddress, nil)
 	if err != nil {
 		glog.Fatal(err)
 	}
-				pprof.WriteHeapProfile(f)
-				f.Close()
-			}
-		}
-	}
-
 }