package common

import (
	"fmt"
	"runtime/debug"
	"stat/common/log"
	"sync/atomic"
	"time"
)

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】server occur error,please attention", log.Any("Detail", r))
				if routinePanicHandler != nil {
					routinePanicHandler(fmt.Sprintf("[Panic] routineCount:%d \n err: %+v\n", routineCount, r) + string(debug.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 {
					routinePanicHandler(fmt.Sprintf("[Panic] routineCount:%d \n err: %+v\n", routineCount, r) + string(debug.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
			}
		}
	}
}