/
default_router_func.go
66 lines (54 loc) · 1.55 KB
/
default_router_func.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
/**
* @File: default_router_func.go
* @Author: Jason Woo
* @Date: 2023/6/30 16:34
**/
package fastnet
import (
"bytes"
"fmt"
"github.com/dyowoo/fastnet/xlog"
"path"
"runtime"
"strings"
"time"
)
const (
StackBegin = 3 // 开始追踪堆栈信息的层数
StackEnd = 5 // 追踪到最后的层数
)
// RouterRecovery
// 用来存放一些RouterSlicesMode下的路由可用的默认中间件
// 如果使用NewDefaultRouterSlicesServer方法初始化的获得的server将自带这个函数
// 作用是接收业务执行上产生的panic并且尝试记录现场信息
func RouterRecovery(request IRequest) {
defer func() {
if err := recover(); err != nil {
panicInfo := getInfo(StackBegin)
xlog.ErrorF("msgId:%d handler panic: info:%s err:%v", request.GetMsgID(), panicInfo, err)
}
}()
request.RouterSlicesNext()
}
// RouterTime 简单累计所有路由组的耗时,不启用
func RouterTime(request IRequest) {
now := time.Now()
request.RouterSlicesNext()
duration := time.Since(now)
fmt.Println(duration.String())
}
func getInfo(ship int) string {
info := new(bytes.Buffer)
// 也可以不指定终点层数即i := ship;; i++ 通过if!ok 结束循环,但是会一直追到最底层报错信息
for i := ship; i <= StackEnd; i++ {
pc, file, lineNo, ok := runtime.Caller(i)
if !ok {
break
}
funcName := runtime.FuncForPC(pc).Name()
fileName := path.Base(file)
funcName = strings.Split(funcName, ".")[1]
_, _ = fmt.Fprintf(info, "funcName:%s fileName:%s LineNo:%d\n", funcName, fileName, lineNo)
}
return info.String()
}