65 lines
1.2 KiB
Go
65 lines
1.2 KiB
Go
//go:build windows
|
|
|
|
package logger
|
|
|
|
import (
|
|
"io"
|
|
"log"
|
|
"strings"
|
|
|
|
"github.com/influxdata/wlog"
|
|
"golang.org/x/sys/windows/svc/eventlog"
|
|
)
|
|
|
|
const (
|
|
LogTargetEventlog = "eventlog"
|
|
eidInfo = 1
|
|
eidWarning = 2
|
|
eidError = 3
|
|
)
|
|
|
|
type eventLogger struct {
|
|
logger *eventlog.Log
|
|
}
|
|
|
|
func (t *eventLogger) Write(b []byte) (int, error) {
|
|
var err error
|
|
|
|
loc := prefixRegex.FindIndex(b)
|
|
n := len(b)
|
|
if loc == nil {
|
|
err = t.logger.Info(1, string(b))
|
|
} else if n > 2 { //skip empty log messages
|
|
line := strings.Trim(string(b[loc[1]:]), " \t\r\n")
|
|
switch rune(b[loc[0]]) {
|
|
case 'I':
|
|
err = t.logger.Info(eidInfo, line)
|
|
case 'W':
|
|
err = t.logger.Warning(eidWarning, line)
|
|
case 'E':
|
|
err = t.logger.Error(eidError, line)
|
|
}
|
|
}
|
|
|
|
return n, err
|
|
}
|
|
|
|
type eventLoggerCreator struct {
|
|
logger *eventlog.Log
|
|
}
|
|
|
|
func (e *eventLoggerCreator) CreateLogger(_ LogConfig) (io.Writer, error) {
|
|
return wlog.NewWriter(&eventLogger{logger: e.logger}), nil
|
|
}
|
|
|
|
func RegisterEventLogger(name string) error {
|
|
eventLog, err := eventlog.Open(name)
|
|
if err != nil {
|
|
log.Printf("E! An error occurred while initializing an event logger. %s", err)
|
|
return err
|
|
}
|
|
|
|
registerLogger(LogTargetEventlog, &eventLoggerCreator{logger: eventLog})
|
|
return nil
|
|
}
|