From a4a27ea9da7d1d61ad1a5972077f888f309e8f4d Mon Sep 17 00:00:00 2001
From: kercylan98 <kercylan@gmail.com>
Date: Wed, 12 Jul 2023 17:45:30 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=8F=AF=E4=BD=BF=E7=94=A8=20super.New?=
 =?UTF-8?q?StackGo=20=E5=88=9B=E5=BB=BA=E7=94=A8=E4=BA=8E=E5=AF=B9?=
 =?UTF-8?q?=E4=B8=8A=E4=B8=80=E4=B8=AA=E5=8D=8F=E7=A8=8B=E5=A0=86=E6=A0=88?=
 =?UTF-8?q?=E8=BF=9B=E8=A1=8C=E6=94=B6=E9=9B=86=E7=9A=84=E6=94=B6=E9=9B=86?=
 =?UTF-8?q?=E5=99=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 utils/super/stack.go | 50 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 50 insertions(+)
 create mode 100644 utils/super/stack.go

diff --git a/utils/super/stack.go b/utils/super/stack.go
new file mode 100644
index 0000000..2458695
--- /dev/null
+++ b/utils/super/stack.go
@@ -0,0 +1,50 @@
+package super
+
+import (
+	"runtime/debug"
+)
+
+// NewStackGo 返回一个用于获取上一个协程调用的堆栈信息的收集器
+func NewStackGo() *StackGo {
+	return new(StackGo)
+}
+
+// StackGo 用于获取上一个协程调用的堆栈信息
+//   - 应当最先运行 Wait 函数，然后在其他协程中调用 Stack 函数或者 GiveUp 函数
+type StackGo struct {
+	stack   chan *struct{} // 消息堆栈
+	collect chan []byte    // 消息堆栈收集
+}
+
+// Wait 等待收集消息堆栈
+//   - 在调用 Wait 函数后，当前协程将会被挂起，直到调用 Stack 或 GiveUp 函数
+func (slf *StackGo) Wait() {
+	slf.stack = make(chan *struct{}, 0)
+	if s := <-slf.stack; s != nil {
+		slf.collect <- debug.Stack()
+		close(slf.collect)
+		slf.collect = nil
+	}
+	close(slf.stack)
+	slf.stack = nil
+}
+
+// Stack 获取消息堆栈
+//   - 在调用 Wait 函数后调用该函数，将会返回上一个协程的堆栈信息
+//   - 在调用 GiveUp 函数后调用该函数，将会 panic
+func (slf *StackGo) Stack() []byte {
+	slf.stack <- &struct{}{}
+	slf.collect = make(chan []byte, 1)
+	stack := <-slf.collect
+	slf.stack = nil
+	return stack
+}
+
+// GiveUp 放弃收集消息堆栈
+//   - 在调用 Wait 函数后调用该函数，将会放弃收集消息堆栈并且释放资源
+//   - 在调用 GiveUp 函数后调用 Stack 函数，将会 panic
+func (slf *StackGo) GiveUp() {
+	if slf.stack != nil {
+		slf.stack <- nil
+	}
+}