fleet/server/logwriter/logwriter.go
2017-12-21 18:37:32 -08:00

76 lines
1.6 KiB
Go

// Package logwriter provides logging utilities for writing osquery results and status logs
package logwriter
import (
"bufio"
"io"
"os"
"sync"
"github.com/pkg/errors"
)
type logWriter struct {
file *os.File
buff *bufio.Writer
mtx sync.Mutex
}
// New creates a logwriter, path refers to file that will receive log content
func New(path string) (io.WriteCloser, error) {
file, err := os.OpenFile(path, os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0644)
if err != nil {
return nil, err
}
buff := bufio.NewWriter(file)
return &logWriter{file: file, buff: buff}, nil
}
// Write bytes to file
func (l *logWriter) Write(b []byte) (int, error) {
l.mtx.Lock()
defer l.mtx.Unlock()
if l.buff == nil || l.file == nil {
return 0, errors.New("logwriter: can't write to closed file")
}
if _, statErr := os.Stat(l.file.Name()); os.IsNotExist(statErr) {
f, err := os.OpenFile(l.file.Name(), os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0644)
if err != nil {
return 0, errors.Wrapf(err, "create file for logWriter %s", l.file.Name())
}
l.file = f
l.buff = bufio.NewWriter(f)
}
return l.buff.Write(b)
}
// Flush write all buffered bytes to log file
func (l *logWriter) Flush() error {
l.mtx.Lock()
defer l.mtx.Unlock()
if l.buff == nil {
return errors.New("can't write to a closed file")
}
return l.buff.Flush()
}
// Close log file
func (l *logWriter) Close() error {
l.mtx.Lock()
defer l.mtx.Unlock()
if l.buff != nil {
if err := l.buff.Flush(); err != nil {
return err
}
l.buff = nil
}
if l.file != nil {
if err := l.file.Close(); err != nil {
return err
}
l.file = nil
}
return nil
}