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