From 6cb2a92b7aef10fb8269050bf5d57b2f356c3a16 Mon Sep 17 00:00:00 2001
From: debugtalk <mail@debugtalk.com>
Date: Tue, 28 Sep 2021 15:48:53 +0800
Subject: [PATCH] feat: support set debug for boomer

---
 boomer.go | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/boomer.go b/boomer.go
index 9fc7e5ab..dce01639 100644
--- a/boomer.go
+++ b/boomer.go
@@ -13,23 +13,32 @@ func Run(testcases ...*TestCase) {
 }
 
 func NewBoomer() *Boomer {
-	return &Boomer{}
+	return &Boomer{
+		debug: false,
+	}
 }
 
 type Boomer struct {
+	debug bool
+}
+
+func (b *Boomer) SetDebug(debug bool) *Boomer {
+	b.debug = debug
+	return b
 }
 
 func (b *Boomer) Run(testcases ...*TestCase) {
 	var taskSlice []*boomer.Task
 	for _, testcase := range testcases {
-		task := convertBoomerTask(testcase)
+		task := b.convertBoomerTask(testcase)
 		taskSlice = append(taskSlice, task)
 	}
 	boomer.Run(taskSlice...)
 }
 
-func convertBoomerTask(testcase *TestCase) *boomer.Task {
+func (b *Boomer) convertBoomerTask(testcase *TestCase) *boomer.Task {
 	runner := NewRunner()
+	runner = runner.SetDebug(b.debug)
 	return &boomer.Task{
 		Name:   testcase.Config.Name,
 		Weight: testcase.Config.Weight,