feat: 可使用 super.NewStackGo 创建用于对上一个协程堆栈进行收集的收集器

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
+	}
+}