Trade stack synchronization added

AndreyMashukov · Mar 13, 2024 · ac48437 · ac48437
1 parent b19d77a
commit ac48437
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/config/container.go b/src/config/container.go
@@ -216,6 +216,7 @@ func InitServiceContainer() Container {
 		PriceCalculator:    &priceCalculator,
 		RDB:                rdb,
 		Ctx:                &ctx,
+		Lock:               &sync.Mutex{},
 	}
 
 	orderExecutor := exchange.OrderExecutor{

diff --git a/src/service/exchange/trade_stack.go b/src/service/exchange/trade_stack.go
@@ -11,6 +11,7 @@ import (
 	"gitlab.com/open-soft/go-crypto-bot/src/service"
 	"gitlab.com/open-soft/go-crypto-bot/src/utils"
 	"sort"
+	"sync"
 	"time"
 )
 
@@ -28,6 +29,7 @@ type TradeStack struct {
 	PriceCalculator    PriceCalculatorInterface
 	RDB                *redis.Client
 	Ctx                *context.Context
+	Lock               *sync.Mutex
 }
 
 type TradeStackParams struct {
@@ -69,6 +71,9 @@ func (t *TradeStack) CanBuy(limit model.TradeLimit) bool {
 }
 
 func (t *TradeStack) GetTradeStack(params TradeStackParams) []model.TradeStackItem {
+	t.Lock.Lock()
+	defer t.Lock.Unlock()
+
 	balanceUsdt, err := t.BalanceService.GetAssetBalance("USDT", true)
 	stack := make([]model.TradeStackItem, 0)