From 9f5ef21dda86cad40c79897bb5b1cdb03ea99aef Mon Sep 17 00:00:00 2001
From: c9s <yoanlin93@gmail.com>
Date: Tue, 6 Jun 2023 16:57:38 +0800
Subject: [PATCH] types: Add TradeWith helper

---
 pkg/types/kline.go | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/pkg/types/kline.go b/pkg/types/kline.go
index 8df06d297..a17e1e94e 100644
--- a/pkg/types/kline.go
+++ b/pkg/types/kline.go
@@ -627,6 +627,16 @@ func (k *KLineSeries) Length() int {
 
 var _ Series = &KLineSeries{}
 
+func TradeWith(symbol string, f func(trade Trade)) func(trade Trade) {
+	return func(trade Trade) {
+		if symbol != "" && trade.Symbol != symbol {
+			return
+		}
+
+		f(trade)
+	}
+}
+
 func KLineWith(symbol string, interval Interval, callback KLineCallback) KLineCallback {
 	return func(k KLine) {
 		if k.Symbol != symbol || (k.Interval != "" && k.Interval != interval) {