From abaedd4f39952ea27ebb2937a2e7e7cbff182a27 Mon Sep 17 00:00:00 2001
From: Jan Pochyla <jpochyla@gmail.com>
Date: Wed, 13 Sep 2017 02:50:34 +0200
Subject: [PATCH] add profiling

---
 Gopkg.lock   | 8 +++++++-
 Gopkg.toml   | 4 ++++
 blockbook.go | 8 +++++++-
 3 files changed, 18 insertions(+), 2 deletions(-)

diff --git a/Gopkg.lock b/Gopkg.lock
index 4704879b..1f5d72a3 100644
--- a/Gopkg.lock
+++ b/Gopkg.lock
@@ -25,6 +25,12 @@
   packages = [".","base58","bech32"]
   revision = "501929d3d046174c3d39f0ea54ece471aa17238c"
 
+[[projects]]
+  name = "github.com/pkg/profile"
+  packages = ["."]
+  revision = "5b67d428864e92711fcbd2f8629456121a56d91f"
+  version = "v1.2.1"
+
 [[projects]]
   branch = "master"
   name = "github.com/tecbot/gorocksdb"
@@ -40,6 +46,6 @@
 [solve-meta]
   analyzer-name = "dep"
   analyzer-version = 1
-  inputs-digest = "07980ae9893c6e67b266e97e50329139ccf005f5b425117cd5d0d5590e8ca541"
+  inputs-digest = "119cb8683257687e541ae944d0f3a29aa916bb81dcedd6a69a0f765fd8f1be9d"
   solver-name = "gps-cdcl"
   solver-version = 1
diff --git a/Gopkg.toml b/Gopkg.toml
index f92435a9..dcf099cc 100644
--- a/Gopkg.toml
+++ b/Gopkg.toml
@@ -40,3 +40,7 @@
 [[constraint]]
   branch = "master"
   name = "github.com/tecbot/gorocksdb"
+
+[[constraint]]
+  name = "github.com/pkg/profile"
+  version = "1.2.1"
diff --git a/blockbook.go b/blockbook.go
index 7b741985..f29d02ac 100644
--- a/blockbook.go
+++ b/blockbook.go
@@ -4,6 +4,8 @@ import (
 	"flag"
 	"log"
 	"time"
+
+	"github.com/pkg/profile"
 )
 
 type BlockParser interface {
@@ -124,13 +126,17 @@ var (
 	queryAddress = flag.String("address", "", "query contents of this address")
 
 	resync = flag.Bool("resync", false, "resync until tip")
-
 	repair = flag.Bool("repair", false, "repair the database")
+	prof   = flag.Bool("prof", false, "profile program execution")
 )
 
 func main() {
 	flag.Parse()
 
+	if *prof {
+		defer profile.Start().Stop()
+	}
+
 	if *repair {
 		if err := RepairRocksDB(*dbPath); err != nil {
 			log.Fatal(err)