108 lines
2.3 KiB
Go
108 lines
2.3 KiB
Go
package logging
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
)
|
|
|
|
// File writes its logs to file.
|
|
type File struct {
|
|
fo, fe *os.File
|
|
*LogWriter
|
|
}
|
|
|
|
// NewFile creates a new Logger that writes all logs to the file
|
|
// specified by path. If overwrite is specified, the log file will be
|
|
// truncated before writing. Otherwise, the log file will be appended
|
|
// to.
|
|
func NewFile(path string, overwrite bool) (*File, error) {
|
|
fl := new(File)
|
|
|
|
var err error
|
|
|
|
if overwrite {
|
|
fl.fo, err = os.Create(path)
|
|
} else {
|
|
fl.fo, err = os.OpenFile(path, os.O_WRONLY|os.O_APPEND, 0600) // #nosec G302
|
|
}
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
fl.LogWriter = NewLogWriter(fl.fo, fl.fo)
|
|
return fl, nil
|
|
}
|
|
|
|
// NewSplitFile creates a new Logger that writes debug and information
|
|
// messages to the output file, and warning and higher messages to the
|
|
// error file. If overwrite is specified, the log files will be
|
|
// truncated before writing.
|
|
func NewSplitFile(outpath, errpath string, overwrite bool) (*File, error) {
|
|
fl := new(File)
|
|
|
|
var err error
|
|
|
|
if overwrite {
|
|
fl.fo, err = os.Create(outpath)
|
|
} else {
|
|
fl.fo, err = os.OpenFile(outpath, os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0600)
|
|
}
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if overwrite {
|
|
fl.fe, err = os.Create(errpath)
|
|
} else {
|
|
fl.fe, err = os.OpenFile(errpath, os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0600)
|
|
}
|
|
|
|
if err != nil {
|
|
if closeErr := fl.Close(); closeErr != nil {
|
|
return nil, fmt.Errorf("failed to open error log: %w", errors.Join(closeErr, err))
|
|
}
|
|
return nil, fmt.Errorf("failed to open error log: %w", err)
|
|
}
|
|
|
|
fl.LogWriter = NewLogWriter(fl.fo, fl.fe)
|
|
return fl, nil
|
|
}
|
|
|
|
// Close calls close on the underlying log files.
|
|
func (fl *File) Close() error {
|
|
if fl.fo != nil {
|
|
if err := fl.fo.Close(); err != nil {
|
|
return err
|
|
}
|
|
fl.fo = nil
|
|
}
|
|
|
|
if fl.fe != nil {
|
|
return fl.fe.Close()
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (fl *File) Flush() error {
|
|
if err := fl.fo.Sync(); err != nil {
|
|
return err
|
|
}
|
|
return fl.fe.Sync()
|
|
}
|
|
|
|
func (fl *File) Chmod(mode os.FileMode) error {
|
|
if err := fl.fo.Chmod(mode); err != nil {
|
|
return fmt.Errorf("failed to chmod output log: %w", err)
|
|
}
|
|
|
|
if err := fl.fe.Chmod(mode); err != nil {
|
|
return fmt.Errorf("failed to chmod error log: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|