Remove memprofile file, add pprof HTTP endpoint instead.

10 years ago · a730cff002
parent 0c8bfba04d
commit 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,37 +171,19 @@ func main() {
 			password: *authPass,
 		}
 	}
 	go func() {
 		http.Handle(*metricsPath, handler)
 		http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
 			w.Write([]byte(`<html>
                       <head><title>Node Exporter</title></head>
                       <body>
                       <h1>Node Exporter</h1>
                       <p><a href="` + *metricsPath + `">Metrics</a></p>
                       </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)
 				if err != nil {
 					glog.Fatal(err)
 				}
 				pprof.WriteHeapProfile(f)
 				f.Close()
 			}
 		}
 	}
 	http.Handle(*metricsPath, handler)
 	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
 		w.Write([]byte(`<html>
 			<head><title>Node Exporter</title></head>
 			<body>
 			<h1>Node Exporter</h1>
 			<p><a href="` + *metricsPath + `">Metrics</a></p>
 			</body>
 			</html>`))
 	})
 	err = http.ListenAndServe(*listenAddress, nil)
 	if err != nil {
 		glog.Fatal(err)
 	}
 }