Refactor log

log/observable.go

@@ -0,0 +1,136 @@
+package log
+
+import (
+	"context"
+	"io"
+	"os"
+	"time"
+
+	F "github.com/sagernet/sing/common/format"
+	"github.com/sagernet/sing/common/observable"
+)
+
+var _ Factory = (*observableFactory)(nil)
+
+type observableFactory struct {
+	formatter  Formatter
+	writer     io.Writer
+	level      Level
+	subscriber *observable.Subscriber[Entry]
+	observer   *observable.Observer[Entry]
+}
+
+func NewObservableFactory(formatter Formatter, writer io.Writer) ObservableFactory {
+	factory := &observableFactory{
+		formatter:  formatter,
+		writer:     writer,
+		level:      LevelTrace,
+		subscriber: observable.NewSubscriber[Entry](128),
+	}
+	factory.observer = observable.NewObserver[Entry](factory.subscriber, 64)
+	return factory
+}
+
+func (f *observableFactory) Level() Level {
+	return f.level
+}
+
+func (f *observableFactory) SetLevel(level Level) {
+	f.level = level
+}
+
+func (f *observableFactory) Logger() ContextLogger {
+	return f.NewLogger("")
+}
+
+func (f *observableFactory) NewLogger(tag string) ContextLogger {
+	return &observableLogger{f, tag}
+}
+
+func (f *observableFactory) Subscribe() (subscription observable.Subscription[Entry], done <-chan struct{}, err error) {
+	return f.observer.Subscribe()
+}
+
+func (f *observableFactory) UnSubscribe(sub observable.Subscription[Entry]) {
+	f.observer.UnSubscribe(sub)
+}
+
+var _ ContextLogger = (*observableLogger)(nil)
+
+type observableLogger struct {
+	*observableFactory
+	tag string
+}
+
+func (l *observableLogger) Log(ctx context.Context, level Level, args []any) {
+	if level > l.level {
+		return
+	}
+	message := l.formatter.Format(ctx, level, l.tag, F.ToString(args...), time.Now()) + "\n"
+	if level == LevelPanic {
+		panic(message)
+	}
+	l.writer.Write([]byte(message))
+	if level == LevelFatal {
+		os.Exit(1)
+	}
+	if l.subscriber != nil {
+		l.subscriber.Emit(Entry{level, message})
+	}
+}
+
+func (l *observableLogger) Trace(args ...any) {
+	l.Log(nil, LevelTrace, args)
+}
+
+func (l *observableLogger) Debug(args ...any) {
+	l.Log(nil, LevelDebug, args)
+}
+
+func (l *observableLogger) Info(args ...any) {
+	l.Log(nil, LevelInfo, args)
+}
+
+func (l *observableLogger) Warn(args ...any) {
+	l.Log(nil, LevelWarn, args)
+}
+
+func (l *observableLogger) Error(args ...any) {
+	l.Log(nil, LevelError, args)
+}
+
+func (l *observableLogger) Fatal(args ...any) {
+	l.Log(nil, LevelFatal, args)
+}
+
+func (l *observableLogger) Panic(args ...any) {
+	l.Log(nil, LevelPanic, args)
+}
+
+func (l *observableLogger) TraceContext(ctx context.Context, args ...any) {
+	l.Log(ctx, LevelTrace, args)
+}
+
+func (l *observableLogger) DebugContext(ctx context.Context, args ...any) {
+	l.Log(ctx, LevelDebug, args)
+}
+
+func (l *observableLogger) InfoContext(ctx context.Context, args ...any) {
+	l.Log(ctx, LevelInfo, args)
+}
+
+func (l *observableLogger) WarnContext(ctx context.Context, args ...any) {
+	l.Log(ctx, LevelWarn, args)
+}
+
+func (l *observableLogger) ErrorContext(ctx context.Context, args ...any) {
+	l.Log(ctx, LevelError, args)
+}
+
+func (l *observableLogger) FatalContext(ctx context.Context, args ...any) {
+	l.Log(ctx, LevelFatal, args)
+}
+
+func (l *observableLogger) PanicContext(ctx context.Context, args ...any) {
+	l.Log(ctx, LevelPanic, args)
+}