From d763a3c4150ac3fb76acfb94b45a6b0524e26e61 Mon Sep 17 00:00:00 2001
From: c9s <yoan@maicoin.com>
Date: Mon, 18 Oct 2021 17:26:03 +0800
Subject: [PATCH] bbgo: add debug ewma and sma

---
 pkg/bbgo/session.go | 13 ++++---------
 pkg/util/envvars.go |  9 +++++++++
 2 files changed, 13 insertions(+), 9 deletions(-)

diff --git a/pkg/bbgo/session.go b/pkg/bbgo/session.go
index fce711e46..9a9e66153 100644
--- a/pkg/bbgo/session.go
+++ b/pkg/bbgo/session.go
@@ -19,17 +19,12 @@ import (
 
 var (
 	debugEWMA = false
-	debugSMA = false
+	debugSMA  = false
 )
 
 func init() {
-	if v, ok := util.GetEnvVarBool("DEBUG_EWMA"); ok {
-		debugEWMA = v
-	}
-
-	if v, ok := util.GetEnvVarBool("DEBUG_SMA"); ok {
-		debugSMA = v
-	}
+	util.SetEnvVarBool("DEBUG_EWMA", &debugEWMA)
+	util.SetEnvVarBool("DEBUG_SMA", &debugSMA)
 }
 
 type StandardIndicatorSet struct {
@@ -69,7 +64,7 @@ func NewStandardIndicatorSet(symbol string, store *MarketDataStore) *StandardInd
 			set.ewma[iw] = &indicator.EWMA{IntervalWindow: iw}
 			set.ewma[iw].Bind(store)
 
-			// if debug ewma is enabled, we add the debug handler
+			// if debug EWMA is enabled, we add the debug handler
 			if debugEWMA {
 				set.ewma[iw].OnUpdate(func(value float64) {
 					log.Infof("%s EWMA %s: %f", symbol, iw.String(), value)
diff --git a/pkg/util/envvars.go b/pkg/util/envvars.go
index c673ae019..9ad3a714e 100644
--- a/pkg/util/envvars.go
+++ b/pkg/util/envvars.go
@@ -38,6 +38,15 @@ func GetEnvVarInt(n string) (int, bool) {
 	return num, true
 }
 
+func SetEnvVarBool(n string, v *bool) bool {
+	b, ok := GetEnvVarBool(n)
+	if ok {
+		*v = b
+	}
+
+	return ok
+}
+
 func GetEnvVarBool(n string) (bool, bool) {
 	str, ok := os.LookupEnv(n)
 	if !ok {