47 lines
1.1 KiB
Go
47 lines
1.1 KiB
Go
package log
|
|
|
|
import (
|
|
"reflect"
|
|
"sync"
|
|
)
|
|
|
|
// LogEntry 定义了高性能日志必须实现的接口
|
|
type LogEntry interface {
|
|
Reset()
|
|
Base() *BaseLog
|
|
}
|
|
|
|
// PoolManager 管理不同日志类型的对象池
|
|
type PoolManager struct {
|
|
pools sync.Map // map[reflect.Type]*sync.Pool
|
|
}
|
|
|
|
var globalPools = &PoolManager{}
|
|
|
|
// GetEntry 从池中获取一个指定类型的日志对象，并确保其处于 Reset 后的干净状态
|
|
func GetEntry(t reflect.Type) LogEntry {
|
|
pool, _ := globalPools.pools.LoadOrStore(t, &sync.Pool{
|
|
New: func() any {
|
|
return reflect.New(t.Elem()).Interface()
|
|
},
|
|
})
|
|
entry := pool.(*sync.Pool).Get().(LogEntry)
|
|
entry.Reset() // 确保获取到的对象永远是干净且预分配好的
|
|
return entry
|
|
}
|
|
|
|
// PutEntry 将日志对象归还到池中，不再进行 Reset
|
|
func PutEntry(entry LogEntry) {
|
|
t := reflect.TypeOf(entry)
|
|
if pool, ok := globalPools.pools.Load(t); ok {
|
|
pool.(*sync.Pool).Put(entry)
|
|
}
|
|
}
|
|
|
|
// WithEntry 执行闭包并在结束后自动回收对象
|
|
func WithEntry(t reflect.Type, fn func(LogEntry)) {
|
|
entry := GetEntry(t)
|
|
defer PutEntry(entry)
|
|
fn(entry)
|
|
}
|