41 lines
1023 B
Go
41 lines
1023 B
Go
|
package mq
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/apache/rocketmq-client-go/v2/primitive"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type Logger interface {
|
|||
|
|
// Debugf logs a formatted debugging message.
|
|||
|
|
Debugf(format string, args ...interface{})
|
|||
|
|
|
|||
|
|
// Infof logs a formatted informational message.
|
|||
|
|
Infof(format string, args ...interface{})
|
|||
|
|
|
|||
|
|
// Warnf logs a formatted warning message.
|
|||
|
|
Warnf(format string, args ...interface{})
|
|||
|
|
|
|||
|
|
// Errorf logs a formatted error message.
|
|||
|
|
Errorf(format string, args ...interface{})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func init() {
|
|||
|
|
SetRecoverLogger(nil)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SetRecoverLogger 设置RocketMQ的recover的处理函数日志，不设置会导致协程panic后程序退出
|
|||
|
|
func SetRecoverLogger(logger Logger) {
|
|||
|
|
SetRecoverHandler(func(err any) {
|
|||
|
|
if logger == nil {
|
|||
|
|
fmt.Printf("rocketmq 发生 panic：%+v\n", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
logger.Errorf("rocketmq 发生 panic：%+v", err)
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SetRecoverHandler 设置RocketMQ的recover的处理函数，不设置会导致协程panic后程序退出
|
|||
|
|
func SetRecoverHandler(fn func(any)) {
|
|||
|
|
primitive.PanicHandler = fn
|
|||
|
|
}
|