Christian Schwarz
64 lines
1.2 KiB
Go
64 lines
1.2 KiB
Go
package cmd
|
|
|
|
import (
|
|
"context"
|
|
"github.com/pkg/errors"
|
|
"github.com/zrepl/zrepl/logger"
|
|
"io"
|
|
"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
|
|
}
|