|
@@ -1,6 +1,7 @@
|
|
|
package loggerutils
|
|
|
|
|
|
import (
|
|
|
+ "errors"
|
|
|
"os"
|
|
|
"strconv"
|
|
|
"sync"
|
|
@@ -11,6 +12,7 @@ import (
|
|
|
// RotateFileWriter is Logger implementation for default Docker logging.
|
|
|
type RotateFileWriter struct {
|
|
|
f *os.File // store for closing
|
|
|
+ closed bool
|
|
|
mu sync.Mutex
|
|
|
capacity int64 //maximum size of each file
|
|
|
currentSize int64 // current size of the latest file
|
|
@@ -42,6 +44,10 @@ func NewRotateFileWriter(logPath string, capacity int64, maxFiles int) (*RotateF
|
|
|
//WriteLog write log message to File
|
|
|
func (w *RotateFileWriter) Write(message []byte) (int, error) {
|
|
|
w.mu.Lock()
|
|
|
+ if w.closed {
|
|
|
+ w.mu.Unlock()
|
|
|
+ return -1, errors.New("cannot write because the output file was closed")
|
|
|
+ }
|
|
|
if err := w.checkCapacityAndRotate(); err != nil {
|
|
|
w.mu.Unlock()
|
|
|
return -1, err
|
|
@@ -100,6 +106,8 @@ func rotate(name string, maxFiles int) error {
|
|
|
|
|
|
// LogPath returns the location the given writer logs to.
|
|
|
func (w *RotateFileWriter) LogPath() string {
|
|
|
+ w.mu.Lock()
|
|
|
+ defer w.mu.Unlock()
|
|
|
return w.f.Name()
|
|
|
}
|
|
|
|
|
@@ -120,5 +128,14 @@ func (w *RotateFileWriter) NotifyRotateEvict(sub chan interface{}) {
|
|
|
|
|
|
// Close closes underlying file and signals all readers to stop.
|
|
|
func (w *RotateFileWriter) Close() error {
|
|
|
- return w.f.Close()
|
|
|
+ w.mu.Lock()
|
|
|
+ defer w.mu.Unlock()
|
|
|
+ if w.closed {
|
|
|
+ return nil
|
|
|
+ }
|
|
|
+ if err := w.f.Close(); err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ w.closed = true
|
|
|
+ return nil
|
|
|
}
|