From 547be3c9516b498a383b0cd5332d2460962146d0 Mon Sep 17 00:00:00 2001
From: Richard Ramos <info@richardramos.me>
Date: Wed, 14 Apr 2021 22:17:12 -0400
Subject: [PATCH] extract subscription to separate file

---
 waku/v2/node/subscription.go | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)
 create mode 100644 waku/v2/node/subscription.go

diff --git a/waku/v2/node/subscription.go b/waku/v2/node/subscription.go
new file mode 100644
index 00000000..1afe77b7
--- /dev/null
+++ b/waku/v2/node/subscription.go
@@ -0,0 +1,26 @@
+package node
+
+import (
+	"sync"
+
+	"github.com/status-im/go-waku/waku/common"
+)
+
+type Subscription struct {
+	C      chan *common.Envelope
+	closed bool
+	mutex  sync.Mutex
+	quit   chan struct{}
+}
+
+func (subs *Subscription) Unsubscribe() {
+	if !subs.closed {
+		close(subs.quit)
+	}
+}
+
+func (subs *Subscription) IsClosed() bool {
+	subs.mutex.Lock()
+	defer subs.mutex.Unlock()
+	return subs.closed
+}