mirror of
https://github.com/zrepl/zrepl.git
synced 2024-11-26 18:34:43 +01:00
109 lines
2.1 KiB
Go
109 lines
2.1 KiB
Go
package cmd
|
|
|
|
import (
|
|
"context"
|
|
"github.com/pkg/errors"
|
|
"github.com/zrepl/zrepl/logger"
|
|
"io"
|
|
"log/syslog"
|
|
"net"
|
|
"time"
|
|
)
|
|
|
|
type WriterOutlet struct {
|
|
Formatter EntryFormatter
|
|
Writer io.Writer
|
|
}
|
|
|
|
func (h WriterOutlet) WriteEntry(ctx context.Context, entry logger.Entry) error {
|
|
bytes, err := h.Formatter.Format(&entry)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = h.Writer.Write(bytes)
|
|
h.Writer.Write([]byte("\n"))
|
|
return err
|
|
}
|
|
|
|
type TCPOutlet struct {
|
|
Formatter EntryFormatter
|
|
Net, Address string
|
|
Dialer net.Dialer
|
|
RetryInterval time.Duration
|
|
conn net.Conn
|
|
retry time.Time
|
|
}
|
|
|
|
func (h *TCPOutlet) WriteEntry(ctx context.Context, e logger.Entry) error {
|
|
|
|
b, err := h.Formatter.Format(&e)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if h.conn == nil {
|
|
if time.Now().Sub(h.retry) < h.RetryInterval {
|
|
return nil // this is not an error toward the logger
|
|
//return errors.New("TCP hook reconnect prohibited by retry interval")
|
|
}
|
|
h.conn, err = h.Dialer.DialContext(ctx, h.Net, h.Address)
|
|
if err != nil {
|
|
h.retry = time.Now()
|
|
return errors.Wrap(err, "cannot dial")
|
|
}
|
|
}
|
|
|
|
_, err = h.conn.Write(b)
|
|
if err != nil {
|
|
h.conn.Close()
|
|
h.conn = nil
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
type SyslogOutlet struct {
|
|
Formatter EntryFormatter
|
|
RetryInterval time.Duration
|
|
writer *syslog.Writer
|
|
lastConnectAttempt time.Time
|
|
}
|
|
|
|
func (o *SyslogOutlet) WriteEntry(ctx context.Context, entry logger.Entry) error {
|
|
|
|
bytes, err := o.Formatter.Format(&entry)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
s := string(bytes)
|
|
|
|
if o.writer == nil {
|
|
now := time.Now()
|
|
if now.Sub(o.lastConnectAttempt) < o.RetryInterval {
|
|
return nil // not an error toward logger
|
|
}
|
|
o.writer, err = syslog.New(syslog.LOG_LOCAL0, "zrepl")
|
|
o.lastConnectAttempt = time.Now()
|
|
if err != nil {
|
|
o.writer = nil
|
|
return err
|
|
}
|
|
}
|
|
|
|
switch entry.Level {
|
|
case logger.Debug:
|
|
return o.writer.Debug(s)
|
|
case logger.Info:
|
|
return o.writer.Info(s)
|
|
case logger.Warn:
|
|
return o.writer.Warning(s)
|
|
case logger.Error:
|
|
return o.writer.Err(s)
|
|
default:
|
|
return o.writer.Err(s) // write as error as reaching this case is in fact an error
|
|
}
|
|
|
|
}
|