FileWriter.go

package logger

import (
	"io"
	"sync"
)

type FileWriter struct {
	writer    io.Writer
	encoder   Encoder
	separator []byte
	lock      sync.Mutex
}

type separators struct {
	Empty   []byte
	NewLine []byte
}

var (
	Separators = separators{
		Empty:   make([]byte, 0),
		NewLine: []byte("\n"),
	}
)

func NewFileWriter(out io.Writer, e Encoder, separator []byte) *FileWriter {
	if nil == e {
		e = NewStringEncoder()
	}
	return &FileWriter{
		writer:    out,
		encoder:   e,
		separator: separator,
		lock:      sync.Mutex{},
	}
}

func (f *FileWriter) Write(m Message) {
	f.lock.Lock()
	defer f.lock.Unlock()
	f.writer.Write(f.encoder.Encode(m))
	f.writer.Write(f.separator)
}