|
|
|
@ -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)
|
|
|
|
|