Initial commit

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>

common/go.go

@@ -0,0 +1,108 @@
+package common
+
+import (
+	"fmt"
+	"runtime/debug"
+	"sync"
+	"sync/atomic"
+	"time"
+
+	"91porn-server/common/log"
+	"91porn-server/common/stderr"
+)
+
+var (
+	routinePanicHandler func(string)
+	routineCount        int64
+)
+
+func SetPanicHandler(handler func(string)) {
+	routinePanicHandler = handler
+}
+
+func Go(f func()) {
+	atomic.AddInt64(&routineCount, 1)
+	go func() {
+		defer func() {
+			atomic.AddInt64(&routineCount, -1)
+			if r := recover(); r != nil {
+				log.Error("【Warning】serevr occour error,please attention", log.Any("Detail", r))
+				if routinePanicHandler != nil {
+					var stack string
+					if !stderr.IsContain(r, stderr.WriteBrokenError) {
+						stack = string(debug.Stack())
+					}
+					routinePanicHandler(fmt.Sprintf("[Panic] routineCount:%d \n err: %+v\n", routineCount, r) + stack)
+				}
+			}
+		}()
+		f()
+	}()
+}
+
+func GoParam(i int, f func(i int)) {
+	atomic.AddInt64(&routineCount, 1)
+	go func(i int) {
+		defer func() {
+			atomic.AddInt64(&routineCount, -1)
+			if r := recover(); r != nil {
+				log.Error("【Warning】serevr occour error,please attention", log.Any("Detail", r))
+				if routinePanicHandler != nil {
+					var stack string
+					if !stderr.IsContain(r, stderr.WriteBrokenError) {
+						stack = string(debug.Stack())
+					}
+					routinePanicHandler(fmt.Sprintf("[Panic] routineCount:%d \n err: %+v\n", routineCount, r) + stack)
+				}
+			}
+		}()
+		f(i)
+	}(i)
+}
+
+// WaitGoQuit 等待go退出，timeoutSec 超时秒数，<=0表示不超时
+func WaitGoQuit(timeoutSec int32) {
+	for {
+		count := atomic.LoadInt64(&routineCount)
+		if count == 0 {
+			return
+		}
+		time.Sleep(time.Second)
+		if timeoutSec > 0 {
+			timeoutSec--
+			if timeoutSec == 0 {
+				return
+			}
+		}
+	}
+}
+
+func GoAndWait(funcs ...func() error) (err error) {
+	var wg sync.WaitGroup
+	wg.Add(len(funcs))
+	for i := range funcs {
+		go func(f func() error) {
+			atomic.AddInt64(&routineCount, 1)
+			defer wg.Done()
+			defer func() {
+				atomic.AddInt64(&routineCount, -1)
+				if r := recover(); r != nil {
+					log.Error("【Warning】serevr occour error,please attention", log.Any("Detail", r))
+					if routinePanicHandler != nil {
+						var stack string
+						if !stderr.IsContain(r, stderr.WriteBrokenError) {
+							stack = string(debug.Stack())
+						}
+						routinePanicHandler(fmt.Sprintf("[Panic] routineCount:%d \n err: %+v\n", routineCount, r) + stack)
+					}
+				}
+			}()
+			if routineErr := f(); routineErr != nil {
+				err = routineErr
+
+			}
+		}(funcs[i])
+	}
+	wg.Wait()
+	return
+}