Merge pull request #1623 from OpenBazaar/profile

Add profiling option to mobile Node

mobile/config.go

@@ -28,4 +28,7 @@ type NodeConfig struct {
 
 	// The coin to use
 	CoinType wallet.CoinType
+
+	// Run the pprof profiler on port 6060
+	Profile bool
 }

mobile/node.go

@@ -5,6 +5,7 @@ import (
 	"errors"
 	"fmt"
 	"io/ioutil"
+	"net"
 	"net/http"
 	"os"
 	"path"
@@ -53,6 +54,7 @@ import (
 	"github.com/natefinch/lumberjack"
 	"github.com/op/go-logging"
 	"github.com/tyler-smith/go-bip39"
+	_ "net/http/pprof"
 )
 
 var log = logging.MustGetLogger("mobile")
@@ -75,14 +77,15 @@ var (
 )
 
 // NewNode create the configuration file for a new node
-func NewNode(repoPath string, authenticationToken string, testnet bool, userAgent string, walletTrustedPeer string, password string, mnemonic string) *Node {
+func NewNode(repoPath string, authenticationToken string, testnet bool, userAgent string, walletTrustedPeer string, password string, mnemonic string, profile bool) *Node {
 	// Node config
 	nodeconfig := &NodeConfig{
 		RepoPath:            repoPath,
 		AuthenticationToken: authenticationToken,
 		Testnet:             testnet,
 		UserAgent:           userAgent,
 		WalletTrustedPeer:   walletTrustedPeer,
+		Profile:             profile,
 	}
 
 	// Use Mobile struct to carry config data
@@ -339,6 +342,15 @@ func (n *Node) startIPFSNode(repoPath string, config *ipfscore.BuildCfg) (*ipfsc
 
 // Start start openbazaard (OpenBazaar daemon)
 func (n *Node) Start() error {
+	if n.config.Profile {
+		go func() {
+			listenAddr := net.JoinHostPort("", "6060")
+			profileRedirect := http.RedirectHandler("/debug/pprof",
+				http.StatusSeeOther)
+			http.Handle("/", profileRedirect)
+			log.Errorf("%v", http.ListenAndServe(listenAddr, nil))
+		}()
+	}
 	nd, ctx, err := n.startIPFSNode(n.config.RepoPath, n.ipfsConfig)
 	if err != nil {
 		return err