Remove memprofile file, add pprof HTTP endpoint instead.

This commit is contained in:
Julius Volz 2015-03-05 17:02:17 +01:00
parent 0c8bfba04d
commit a730cff002
1 changed files with 14 additions and 32 deletions

View File

@ -6,9 +6,9 @@ import (
"fmt" "fmt"
"io/ioutil" "io/ioutil"
"net/http" "net/http"
_ "net/http/pprof"
"os" "os"
"os/signal" "os/signal"
"runtime/pprof"
"strings" "strings"
"sync" "sync"
"syscall" "syscall"
@ -171,37 +171,19 @@ func main() {
password: *authPass, 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 { http.Handle(*metricsPath, handler)
select { http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
case <-sigUsr1: w.Write([]byte(`<html>
glog.Infof("got signal") <head><title>Node Exporter</title></head>
if *memProfile != "" { <body>
glog.Infof("Writing memory profile to %s", *memProfile) <h1>Node Exporter</h1>
f, err := os.Create(*memProfile) <p><a href="` + *metricsPath + `">Metrics</a></p>
if err != nil { </body>
glog.Fatal(err) </html>`))
} })
pprof.WriteHeapProfile(f) err = http.ListenAndServe(*listenAddress, nil)
f.Close() if err != nil {
} glog.Fatal(err)
}
} }
} }