@@ -0,0 +1,59 @@
|
||||
package mediaService
|
||||
|
||||
import (
|
||||
"91porn-server/common/log"
|
||||
"fmt"
|
||||
"runtime"
|
||||
"strings"
|
||||
"sync"
|
||||
)
|
||||
|
||||
type worker struct {
|
||||
num int // 最大
|
||||
ch chan struct{}
|
||||
wg sync.WaitGroup
|
||||
}
|
||||
|
||||
var w *worker
|
||||
|
||||
func init() {
|
||||
num := 30 //最大同时执行的执程,防止同时开启的协程过多,把其他服务弄挂
|
||||
w = &worker{
|
||||
num: num,
|
||||
ch: make(chan struct{}, num),
|
||||
}
|
||||
}
|
||||
|
||||
func (w *worker) Wait() {
|
||||
w.wg.Wait()
|
||||
}
|
||||
|
||||
func (w *worker) Exec(f func()) {
|
||||
// 超出则阻塞
|
||||
w.ch <- struct{}{}
|
||||
w.wg.Add(1)
|
||||
go func() {
|
||||
defer func() {
|
||||
<-w.ch
|
||||
w.wg.Done()
|
||||
if r := recover(); r != nil {
|
||||
dep := 0
|
||||
t := make([]string, 0, 10)
|
||||
for i := 1; i < 10; i++ {
|
||||
_, file, line, ok := runtime.Caller(i)
|
||||
if !ok {
|
||||
break
|
||||
}
|
||||
if strings.Contains(file, "/runtime/") || strings.Contains(file, "/reflect/") {
|
||||
continue
|
||||
}
|
||||
t = append(t, fmt.Sprintf("%s∟%s:%d", strings.Repeat(" ", dep), file, line))
|
||||
dep++
|
||||
}
|
||||
exception := strings.Join(t, "\n")
|
||||
log.Error("worker exec Panic: ", log.Any("Exception", exception))
|
||||
}
|
||||
}()
|
||||
f()
|
||||
}()
|
||||
}
|
||||
Reference in New Issue
Block a user