1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-11-02 09:17:55 -04:00
v2fly/app/log/internal/log_writer.go

93 lines
1.6 KiB
Go
Raw Normal View History

2016-05-12 02:24:41 -04:00
package internal
2015-12-05 15:10:14 -05:00
import (
"log"
"os"
2016-07-10 17:11:42 -04:00
"time"
2015-12-05 15:10:14 -05:00
2016-08-20 14:55:45 -04:00
"v2ray.com/core/common/platform"
"v2ray.com/core/common/signal"
2015-12-05 15:10:14 -05:00
)
2016-05-12 02:24:41 -04:00
type LogWriter interface {
2016-04-29 07:38:40 -04:00
Log(LogEntry)
2016-07-10 17:11:42 -04:00
Close()
2015-12-05 15:10:14 -05:00
}
2016-05-12 02:24:41 -04:00
type NoOpLogWriter struct {
2015-12-05 15:10:14 -05:00
}
2017-01-04 06:52:24 -05:00
func (v *NoOpLogWriter) Log(entry LogEntry) {}
2015-12-05 15:10:14 -05:00
2016-11-27 15:39:09 -05:00
func (v *NoOpLogWriter) Close() {
2016-07-10 17:11:42 -04:00
}
2016-05-12 02:24:41 -04:00
type StdOutLogWriter struct {
2015-12-05 15:10:14 -05:00
logger *log.Logger
2016-07-10 17:11:42 -04:00
cancel *signal.CancelSignal
2015-12-05 15:10:14 -05:00
}
2016-05-12 02:24:41 -04:00
func NewStdOutLogWriter() LogWriter {
return &StdOutLogWriter{
logger: log.New(os.Stdout, "", log.Ldate|log.Ltime),
2016-07-10 17:11:42 -04:00
cancel: signal.NewCloseSignal(),
2015-12-05 15:10:14 -05:00
}
}
2016-11-27 15:39:09 -05:00
func (v *StdOutLogWriter) Log(log LogEntry) {
v.logger.Print(log.String() + platform.LineSeparator())
2015-12-05 15:10:14 -05:00
}
2016-11-27 15:39:09 -05:00
func (v *StdOutLogWriter) Close() {
2016-07-10 17:11:42 -04:00
time.Sleep(500 * time.Millisecond)
}
2016-05-12 02:24:41 -04:00
type FileLogWriter struct {
2016-05-12 20:18:07 -04:00
queue chan string
2015-12-05 15:10:14 -05:00
logger *log.Logger
file *os.File
2016-07-10 17:11:42 -04:00
cancel *signal.CancelSignal
2015-12-05 15:10:14 -05:00
}
2016-11-27 15:39:09 -05:00
func (v *FileLogWriter) Log(log LogEntry) {
2015-12-05 15:10:14 -05:00
select {
2016-11-27 15:39:09 -05:00
case v.queue <- log.String():
2015-12-05 15:10:14 -05:00
default:
// We don't expect this to happen, but don't want to block main thread as well.
}
}
2016-11-27 15:39:09 -05:00
func (v *FileLogWriter) run() {
v.cancel.WaitThread()
defer v.cancel.FinishThread()
2016-11-18 15:30:03 -05:00
2016-05-01 11:19:09 -04:00
for {
2016-11-27 15:39:09 -05:00
entry, open := <-v.queue
2016-05-01 11:19:41 -04:00
if !open {
break
}
2016-11-27 15:39:09 -05:00
v.logger.Print(entry + platform.LineSeparator())
2015-12-05 15:10:14 -05:00
}
}
2016-11-27 15:39:09 -05:00
func (v *FileLogWriter) Close() {
close(v.queue)
v.cancel.WaitForDone()
v.file.Close()
2015-12-05 15:10:14 -05:00
}
2016-05-12 02:24:41 -04:00
func NewFileLogWriter(path string) (*FileLogWriter, error) {
2015-12-05 15:10:14 -05:00
file, err := os.OpenFile(path, os.O_APPEND|os.O_WRONLY|os.O_CREATE, 0600)
if err != nil {
return nil, err
}
2016-05-12 02:24:41 -04:00
logger := &FileLogWriter{
2016-05-12 20:18:07 -04:00
queue: make(chan string, 16),
2015-12-05 15:10:14 -05:00
logger: log.New(file, "", log.Ldate|log.Ltime),
file: file,
2016-07-10 17:11:42 -04:00
cancel: signal.NewCloseSignal(),
2015-12-05 15:10:14 -05:00
}
go logger.run()
return logger, nil
}