From 2c5a5e76ec40fd526d8cd56953cac50fe79f2014 Mon Sep 17 00:00:00 2001
From: Parker Moore <parkrmoore@gmail.com>
Date: Sun, 13 Dec 2015 12:21:32 -0800
Subject: [PATCH] script/stackprof: allow CLI to set stackprof mode [ci skip]

---
 script/stackprof | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/script/stackprof b/script/stackprof
index 3399616f..1c833142 100755
--- a/script/stackprof
+++ b/script/stackprof
@@ -2,15 +2,21 @@
 
 set -e
 
+case "$1" in
+  cpu|object) STACKPROF_MODE="$1"; shift ;;
+  *) STACKPROF_MODE="cpu" ;;
+esac
+
 export BENCHMARK=true
 command -v stackprof > /dev/null || script/bootstrap
 
 TEST_SCRIPT="Jekyll::Commands::Build.process({'source' => 'site'})"
-PROF_OUTPUT_FILE=tmp/stackprof-$(date +%Y%m%d%H%M).dump
+PROF_OUTPUT_FILE=tmp/stackprof-${STACKPROF_MODE}-$(date +%Y%m%d%H%M).dump
 
+echo Stackprof Mode: $STACKPROF_MODE
 test -f "$PROF_OUTPUT_FILE" || {
   bundle exec ruby -r./lib/jekyll -rstackprof \
-    -e "StackProf.run(mode: :cpu, interval: 100, out: '${PROF_OUTPUT_FILE}') { ${TEST_SCRIPT} }"
+    -e "StackProf.run(mode: :${STACKPROF_MODE}, interval: 100, out: '${PROF_OUTPUT_FILE}') { ${TEST_SCRIPT} }"
 }
 
 bundle exec stackprof $PROF_OUTPUT_FILE $@