1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-12-23 02:26:49 -05:00
v2fly/app/log/internal/log_writer.go
Darien Raymond 59a1e2d736
refactor
2017-04-23 19:16:56 +02:00

89 lines
1.6 KiB
Go

package internal
import (
"context"
"log"
"os"
"v2ray.com/core/common/platform"
)
type LogWriter interface {
Log(LogEntry)
Close()
}
type NoOpLogWriter struct {
}
func (*NoOpLogWriter) Log(entry LogEntry) {}
func (*NoOpLogWriter) Close() {}
type StdOutLogWriter struct {
logger *log.Logger
}
func NewStdOutLogWriter() LogWriter {
return &StdOutLogWriter{
logger: log.New(os.Stdout, "", log.Ldate|log.Ltime),
}
}
func (w *StdOutLogWriter) Log(log LogEntry) {
w.logger.Print(log.String() + platform.LineSeparator())
}
func (*StdOutLogWriter) Close() {}
type FileLogWriter struct {
queue chan string
logger *log.Logger
file *os.File
ctx context.Context
cancel context.CancelFunc
}
func (w *FileLogWriter) Log(log LogEntry) {
select {
case <-w.ctx.Done():
return
case w.queue <- log.String():
default:
// We don't expect this to happen, but don't want to block main thread as well.
}
}
func (w *FileLogWriter) run(ctx context.Context) {
for {
select {
case <-ctx.Done():
w.file.Close()
return
case entry := <-w.queue:
w.logger.Print(entry + platform.LineSeparator())
}
}
}
func (w *FileLogWriter) Close() {
w.cancel()
}
func NewFileLogWriter(path string) (*FileLogWriter, error) {
file, err := os.OpenFile(path, os.O_APPEND|os.O_WRONLY|os.O_CREATE, 0600)
if err != nil {
return nil, err
}
ctx, cancel := context.WithCancel(context.Background())
logger := &FileLogWriter{
queue: make(chan string, 16),
logger: log.New(file, "", log.Ldate|log.Ltime),
file: file,
ctx: ctx,
cancel: cancel,
}
go logger.run(ctx)
return logger, nil
}