feat: allow using pprof on FDW when STEAMPIPE_FDW_PPROF environment variable is set (#368)

Author: pdecat

fdw.go

@@ -13,6 +13,9 @@ import (
 	"fmt"
 	"io/ioutil"
 	"log"
+	"net"
+	"net/http"
+	"os"
 	"time"
 	"unsafe"
 
@@ -57,6 +60,18 @@ func init() {
 	log.Printf("[INFO] Version:   v%s\n", version.FdwVersion.String())
 	log.Printf("[INFO] Log level: %s\n", level)
 
+	if _, found := os.LookupEnv("STEAMPIPE_FDW_PPROF"); found {
+		log.Printf("[INFO] PROFILING!!!!")
+		go func() {
+			listener, err := net.Listen("tcp", "localhost:0")
+			if err != nil {
+				log.Println(err)
+				return
+			}
+			log.Printf("[INFO] Check http://localhost:%d/debug/pprof/", listener.Addr().(*net.TCPAddr).Port)
+			log.Println(http.Serve(listener, nil))
+		}()
+	}
 }
 
 //export goFdwGetRelSize