|
@@ -3,6 +3,8 @@
|
|
|
package log
|
|
package log
|
|
|
|
|
|
|
|
import (
|
|
import (
|
|
|
|
|
+ "sync"
|
|
|
|
|
+
|
|
|
"github.com/v2fly/v2ray-core/v4/common"
|
|
"github.com/v2fly/v2ray-core/v4/common"
|
|
|
"github.com/v2fly/v2ray-core/v4/common/log"
|
|
"github.com/v2fly/v2ray-core/v4/common/log"
|
|
|
)
|
|
)
|
|
@@ -15,16 +17,24 @@ type HandlerCreator func(LogType, HandlerCreatorOptions) (log.Handler, error)
|
|
|
|
|
|
|
|
var handlerCreatorMap = make(map[LogType]HandlerCreator)
|
|
var handlerCreatorMap = make(map[LogType]HandlerCreator)
|
|
|
|
|
|
|
|
|
|
+var handlerCreatorMapLock = &sync.RWMutex{}
|
|
|
|
|
+
|
|
|
func RegisterHandlerCreator(logType LogType, f HandlerCreator) error {
|
|
func RegisterHandlerCreator(logType LogType, f HandlerCreator) error {
|
|
|
if f == nil {
|
|
if f == nil {
|
|
|
return newError("nil HandlerCreator")
|
|
return newError("nil HandlerCreator")
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|
|
+ handlerCreatorMapLock.Lock()
|
|
|
|
|
+ defer handlerCreatorMapLock.Unlock()
|
|
|
|
|
+
|
|
|
handlerCreatorMap[logType] = f
|
|
handlerCreatorMap[logType] = f
|
|
|
return nil
|
|
return nil
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
func createHandler(logType LogType, options HandlerCreatorOptions) (log.Handler, error) {
|
|
func createHandler(logType LogType, options HandlerCreatorOptions) (log.Handler, error) {
|
|
|
|
|
+ handlerCreatorMapLock.RLock()
|
|
|
|
|
+ defer handlerCreatorMapLock.RUnlock()
|
|
|
|
|
+
|
|
|
creator, found := handlerCreatorMap[logType]
|
|
creator, found := handlerCreatorMap[logType]
|
|
|
if !found {
|
|
if !found {
|
|
|
return nil, newError("unable to create log handler for ", logType)
|
|
return nil, newError("unable to create log handler for ", logType)
|