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
|
|
|
|
}
|