From 9a2b792ed16c69294983480ccc3ccbdb3506824f Mon Sep 17 00:00:00 2001
From: c9s <yoanlin93@gmail.com>
Date: Tue, 20 Aug 2024 17:49:18 +0800
Subject: [PATCH] twap: split doneSignal into a single file

---
 pkg/twap/v2/done.go            | 39 ++++++++++++++++++++++++++++++++++
 pkg/twap/v2/stream_executor.go | 36 -------------------------------
 2 files changed, 39 insertions(+), 36 deletions(-)
 create mode 100644 pkg/twap/v2/done.go

diff --git a/pkg/twap/v2/done.go b/pkg/twap/v2/done.go
new file mode 100644
index 000000000..9d498b939
--- /dev/null
+++ b/pkg/twap/v2/done.go
@@ -0,0 +1,39 @@
+package twap
+
+import "sync"
+
+type DoneSignal struct {
+	doneC chan struct{}
+	mu    sync.Mutex
+}
+
+func NewDoneSignal() *DoneSignal {
+	return &DoneSignal{
+		doneC: make(chan struct{}),
+	}
+}
+
+func (e *DoneSignal) Emit() {
+	e.mu.Lock()
+	if e.doneC == nil {
+		e.doneC = make(chan struct{})
+	}
+
+	close(e.doneC)
+	e.mu.Unlock()
+}
+
+// Chan returns a channel that emits a signal when the execution is done.
+func (e *DoneSignal) Chan() (c <-chan struct{}) {
+	// if the channel is not allocated, it means it's not started yet, we need to return a closed channel
+	e.mu.Lock()
+	if e.doneC == nil {
+		e.doneC = make(chan struct{})
+		c = e.doneC
+	} else {
+		c = e.doneC
+	}
+	e.mu.Unlock()
+
+	return c
+}
diff --git a/pkg/twap/v2/stream_executor.go b/pkg/twap/v2/stream_executor.go
index 51e56b8d6..1b41889bb 100644
--- a/pkg/twap/v2/stream_executor.go
+++ b/pkg/twap/v2/stream_executor.go
@@ -18,42 +18,6 @@ import (
 
 var defaultUpdateInterval = time.Minute
 
-type DoneSignal struct {
-	doneC chan struct{}
-	mu    sync.Mutex
-}
-
-func NewDoneSignal() *DoneSignal {
-	return &DoneSignal{
-		doneC: make(chan struct{}),
-	}
-}
-
-func (e *DoneSignal) Emit() {
-	e.mu.Lock()
-	if e.doneC == nil {
-		e.doneC = make(chan struct{})
-	}
-
-	close(e.doneC)
-	e.mu.Unlock()
-}
-
-// Chan returns a channel that emits a signal when the execution is done.
-func (e *DoneSignal) Chan() (c <-chan struct{}) {
-	// if the channel is not allocated, it means it's not started yet, we need to return a closed channel
-	e.mu.Lock()
-	if e.doneC == nil {
-		e.doneC = make(chan struct{})
-		c = e.doneC
-	} else {
-		c = e.doneC
-	}
-	e.mu.Unlock()
-
-	return c
-}
-
 // FixedQuantityExecutor is a TWAP executor that places orders on the exchange using the exchange's stream API.
 // It uses a fixed target quantity to place orders.
 type FixedQuantityExecutor struct {