mirror of https://github.com/v2fly/v2ray-core.git
65 lines
1.2 KiB
Go
65 lines
1.2 KiB
Go
package log
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/v2fly/v2ray-core/v4/common/serial"
|
|
)
|
|
|
|
// Message is the interface for all log messages.
|
|
type Message interface {
|
|
String() string
|
|
}
|
|
|
|
// Handler is the interface for log handler.
|
|
type Handler interface {
|
|
Handle(msg Message)
|
|
}
|
|
|
|
// GeneralMessage is a general log message that can contain all kind of content.
|
|
type GeneralMessage struct {
|
|
Severity Severity
|
|
Content interface{}
|
|
}
|
|
|
|
// String implements Message.
|
|
func (m *GeneralMessage) String() string {
|
|
return serial.Concat("[", m.Severity, "] ", m.Content)
|
|
}
|
|
|
|
// Record writes a message into log stream.
|
|
func Record(msg Message) {
|
|
logHandler.Handle(msg)
|
|
}
|
|
|
|
var logHandler syncHandler
|
|
|
|
// RegisterHandler register a new handler as current log handler. Previous registered handler will be discarded.
|
|
func RegisterHandler(handler Handler) {
|
|
if handler == nil {
|
|
panic("Log handler is nil")
|
|
}
|
|
logHandler.Set(handler)
|
|
}
|
|
|
|
type syncHandler struct {
|
|
sync.RWMutex
|
|
Handler
|
|
}
|
|
|
|
func (h *syncHandler) Handle(msg Message) {
|
|
h.RLock()
|
|
defer h.RUnlock()
|
|
|
|
if h.Handler != nil {
|
|
h.Handler.Handle(msg)
|
|
}
|
|
}
|
|
|
|
func (h *syncHandler) Set(handler Handler) {
|
|
h.Lock()
|
|
defer h.Unlock()
|
|
|
|
h.Handler = handler
|
|
}
|