summaryrefslogtreecommitdiffhomepage
path: root/signal.go
diff options
context:
space:
mode:
Diffstat (limited to 'signal.go')
-rw-r--r--signal.go53
1 files changed, 53 insertions, 0 deletions
diff --git a/signal.go b/signal.go
new file mode 100644
index 0000000..2cefad4
--- /dev/null
+++ b/signal.go
@@ -0,0 +1,53 @@
+package main
+
+type Signal struct {
+ enabled AtomicBool
+ C chan struct{}
+}
+
+func NewSignal() (s Signal) {
+ s.C = make(chan struct{}, 1)
+ s.Enable()
+ return
+}
+
+func (s *Signal) Disable() {
+ s.enabled.Set(false)
+ s.Clear()
+}
+
+func (s *Signal) Enable() {
+ s.enabled.Set(true)
+}
+
+/* Unblock exactly one listener
+ */
+func (s *Signal) Send() {
+ if s.enabled.Get() {
+ select {
+ case s.C <- struct{}{}:
+ default:
+ }
+ }
+}
+
+/* Clear the signal if already fired
+ */
+func (s Signal) Clear() {
+ select {
+ case <-s.C:
+ default:
+ }
+}
+
+/* Unblocks all listeners (forever)
+ */
+func (s Signal) Broadcast() {
+ close(s.C)
+}
+
+/* Wait for the signal
+ */
+func (s Signal) Wait() chan struct{} {
+ return s.C
+}