From bec3d3ae50d051f8132281685b27f8133ea04fd8 Mon Sep 17 00:00:00 2001
From: xxj <346944475@qq.com>
Date: Thu, 17 Dec 2020 18:02:14 +0800
Subject: [PATCH] add notify

---
 mynotify/mynotify.go | 53 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)
 create mode 100644 mynotify/mynotify.go

diff --git a/mynotify/mynotify.go b/mynotify/mynotify.go
new file mode 100644
index 0000000..64940c3
--- /dev/null
+++ b/mynotify/mynotify.go
@@ -0,0 +1,53 @@
+package mynotify
+
+import (
+	"time"
+)
+
+type notify struct {
+	tag int
+	cc  chan int // -2:超时退出 -1:已关闭 0:默认状态 1:正在运行
+	tm  time.Duration
+}
+
+// New new signal
+func New(tm time.Duration) *notify {
+	if tm == 0 { // 默认一个超时时间 一小时超时
+		tm = 1 * time.Hour
+	}
+	return &notify{
+		cc: make(chan int, 1),
+		tm: tm,
+	}
+}
+
+// Signal 发送一个信号
+func (s *notify) Signal() {
+	s.cc <- 1
+}
+
+// 等等一个信息
+func (s *notify) Wait() bool {
+	timeout := time.After(s.tm)
+	select {
+	case s.tag = <-s.cc:
+	case <-timeout:
+		s.tag = -2
+	}
+
+	if s.tag == 1 {
+		return true
+	}
+
+	return false
+}
+
+// Stop 发送停止信号
+func (s *notify) Stop() {
+	s.cc <- -1
+}
+
+// Close 关闭
+func (s *notify) Close() {
+	close(s.cc)
+}