2021-09-06 02:03:27 -07:00
|
|
|
package proxy
|
2021-09-05 08:14:08 -07:00
|
|
|
|
|
|
|
import (
|
|
|
|
"log"
|
|
|
|
"os"
|
|
|
|
)
|
|
|
|
|
2021-09-13 03:14:11 -07:00
|
|
|
var logWriter *LogWriter
|
|
|
|
|
2021-09-05 08:14:08 -07:00
|
|
|
type LogWriter struct {
|
|
|
|
f *os.File
|
|
|
|
}
|
|
|
|
|
2021-09-10 03:47:19 -07:00
|
|
|
// Write writes the input data to os.Stderr and the log file.
|
|
|
|
// It returns the number of bytes written and an error.
|
2021-09-05 08:14:08 -07:00
|
|
|
func (lw *LogWriter) Write(p []byte) (n int, err error) {
|
|
|
|
n, err = os.Stderr.Write(p)
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
return lw.f.Write(p)
|
|
|
|
}
|
|
|
|
|
|
|
|
func init() {
|
2021-09-13 03:14:11 -07:00
|
|
|
log.SetPrefix("[proxy] ")
|
|
|
|
log.SetFlags(log.Flags() | log.Lmsgprefix)
|
|
|
|
|
2021-09-13 06:54:57 -07:00
|
|
|
f, err := os.OpenFile(Path("latest.log"), os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0666)
|
2021-09-05 08:14:08 -07:00
|
|
|
if err != nil {
|
2021-09-13 03:14:11 -07:00
|
|
|
log.Fatal(err)
|
2021-09-05 08:14:08 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
go func() {
|
|
|
|
defer f.Close()
|
|
|
|
select {}
|
|
|
|
}()
|
|
|
|
|
2021-09-13 03:14:11 -07:00
|
|
|
logWriter = &LogWriter{f}
|
|
|
|
log.SetOutput(logWriter)
|
2021-09-05 08:14:08 -07:00
|
|
|
}
|