From bad8e04bb4410b94b8e875ccde287f74ab94121a Mon Sep 17 00:00:00 2001
From: Paul Buetow <paul@dionysus>
Date: Sat, 19 Sep 2020 18:38:36 +0100
Subject: server handler context refactoring

---
 internal/done.go | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)
 create mode 100644 internal/done.go

(limited to 'internal/done.go')

diff --git a/internal/done.go b/internal/done.go
new file mode 100644
index 0000000..2326eee
--- /dev/null
+++ b/internal/done.go
@@ -0,0 +1,32 @@
+package internal
+
+import (
+	"sync"
+)
+
+type Done struct {
+	ch    chan struct{}
+	mutex sync.Mutex
+}
+
+func NewDone() *Done {
+	return &Done{
+		ch: make(chan struct{}),
+	}
+}
+
+func (d *Done) Done() <-chan struct{} {
+	return d.ch
+}
+
+func (d *Done) Shutdown() {
+	d.mutex.Lock()
+	defer d.mutex.Unlock()
+
+	select {
+	case <-d.ch:
+		return
+	default:
+		close(d.ch)
+	}
+}
-- 
cgit v1.2.3