77 lines
1.9 KiB
Go
77 lines
1.9 KiB
Go
package ksync
|
||
|
||
import (
|
||
"fmt"
|
||
"github.com/fox/fox/log"
|
||
"github.com/wanghuiyt/ding"
|
||
"runtime/debug"
|
||
"strings"
|
||
)
|
||
|
||
// 辅助函数列表
|
||
var (
|
||
DingAccessToken = "" // 钉钉token
|
||
DingSecret = "" // 钉钉加签
|
||
lastAttributeCode = "" // 最近一条error信息
|
||
)
|
||
|
||
func getAttributeCode(stackErr string) string {
|
||
lines := strings.Split(stackErr, "\n")
|
||
// 检查是否有足够的行数，并提取第9行（索引为8，因为索引从0开始）
|
||
if len(lines) >= 9 {
|
||
goIndex := strings.LastIndex(lines[8], ".go")
|
||
if goIndex != -1 {
|
||
filteredStr := lines[8][:goIndex]
|
||
filteredStr = strings.TrimLeft(filteredStr, "\t ")
|
||
return filteredStr
|
||
}
|
||
}
|
||
return ""
|
||
}
|
||
|
||
// 发送消息给钉钉
|
||
func sendMessage(msg interface{}, stackErr string, dingToken, dingSecret string) {
|
||
if dingToken != "" {
|
||
d := ding.Webhook{
|
||
AccessToken: dingToken, // 上面获取的 access_token
|
||
Secret: dingSecret, // 上面获取的加签的值
|
||
}
|
||
_ = d.SendMessageText(fmt.Sprintf("Recover panic:%v\n%v", msg, stackErr), "13145922265", "17353003985")
|
||
}
|
||
}
|
||
|
||
func Recover(recoverFunc func()) {
|
||
if msg := recover(); msg != nil {
|
||
stackErr := string(debug.Stack())
|
||
attributeCode := getAttributeCode(stackErr)
|
||
if lastAttributeCode != attributeCode {
|
||
lastAttributeCode = attributeCode
|
||
log.ErrorF("Recover panic:%v", msg)
|
||
log.Error(stackErr)
|
||
sendMessage(msg, stackErr, DingAccessToken, DingSecret)
|
||
}
|
||
if recoverFunc != nil {
|
||
recoverFunc()
|
||
}
|
||
}
|
||
|
||
}
|
||
|
||
// RunSafe runs the given fn, recovers if fn panics.
|
||
func RunSafe(fn func(), recoverFunc func()) {
|
||
defer Recover(recoverFunc)
|
||
fn()
|
||
}
|
||
|
||
// GoSafe runs the given fn using another goroutine, recovers if fn panics.
|
||
func GoSafe(fn func(), recoverFunc func()) {
|
||
go RunSafe(fn, recoverFunc)
|
||
}
|
||
|
||
// WrapSafe return a RunSafe wrap func
|
||
func WrapSafe(fn func(), recoverFunc func()) func() {
|
||
return func() {
|
||
RunSafe(fn, recoverFunc)
|
||
}
|
||
}
|