gol4ng/logger

package writer

import (
    "io"
    "sync"
)

// RotateIoWriter will rotate the io.writer when rotate is called
// it delegates the io.writer creation to the Provider
type RotateIoWriter struct {
    mutex          sync.Mutex
    writerProvider Provider
    writer         io.Writer
}

// Write will pass-through data to the underlying io.writer
func (w *RotateIoWriter) Write(output []byte) (int, error) {
    w.mutex.Lock()
    defer w.mutex.Unlock()
    return w.writer.Write(output)
}

// Rotate will ask the provider to change the underlying io.writer with a new one
func (w *RotateIoWriter) Rotate() (err error) {
    w.mutex.Lock()
    defer w.mutex.Unlock()
    w.writer, err = w.writerProvider(w.writer)
    return err
}

// NewRotateIoWriter will create a RotateIoWriter
// create a new io.Writer that can be changed other the time when Rotate() is called
func NewRotateIoWriter(provider Provider) (*RotateIoWriter, error) {
    // create new io.Writer
    writer, err := provider(nil)
    return &RotateIoWriter{writer: writer, writerProvider: provider}, err
}