mirror of
https://github.com/Luzifer/elastic_cron.git
synced 2024-12-23 02:41:20 +00:00
82 lines
1.6 KiB
Go
82 lines
1.6 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"net"
|
|
"time"
|
|
|
|
"github.com/cenkalti/backoff"
|
|
)
|
|
|
|
const (
|
|
readWriteTimeout = 1 * time.Second
|
|
tcpDialTimeout = 5 * time.Second
|
|
)
|
|
|
|
func NewSyslogAdapter(address string) (*SyslogAdapter, error) {
|
|
return &SyslogAdapter{
|
|
address: address,
|
|
dialer: &net.Dialer{
|
|
Timeout: tcpDialTimeout,
|
|
},
|
|
}, nil
|
|
}
|
|
|
|
type message struct {
|
|
Date time.Time
|
|
JobName string
|
|
Message string
|
|
Severity int
|
|
}
|
|
|
|
type SyslogAdapter struct {
|
|
address string
|
|
dialer *net.Dialer
|
|
}
|
|
|
|
func (a *SyslogAdapter) Stream(logstream chan *message) {
|
|
backoff.Retry(func() error {
|
|
conn, err := a.dialer.Dial("tcp", a.address)
|
|
if err != nil {
|
|
fmt.Printf("syslog: Unable to dial to remote address\n")
|
|
return fmt.Errorf("Catch me if you can.")
|
|
}
|
|
defer conn.Close()
|
|
|
|
b := new(bytes.Buffer)
|
|
for msg := range logstream {
|
|
b.Reset()
|
|
|
|
fmt.Fprintf(b, "<%d>%s %s %s: %s\n",
|
|
16*8+msg.Severity,
|
|
msg.Date.Format("Jan 02 15:04:05"),
|
|
cfg.Hostname,
|
|
msg.JobName,
|
|
msg.Message,
|
|
)
|
|
|
|
if err := conn.SetDeadline(time.Now().Add(readWriteTimeout)); err != nil {
|
|
fmt.Printf("syslog: Unable to set deadline: %s\n", err)
|
|
return fmt.Errorf("Catch me if you can.")
|
|
}
|
|
|
|
logLine := b.Bytes()
|
|
written, err := io.Copy(conn, b)
|
|
|
|
if err != nil {
|
|
if written > 0 {
|
|
fmt.Printf("syslog: (%d/%d) %s\n", written, len(logLine), err)
|
|
} else {
|
|
fmt.Printf("syslog: %s\n", err)
|
|
}
|
|
return fmt.Errorf("syslog: %s", err)
|
|
}
|
|
}
|
|
|
|
fmt.Printf("syslog: I got out of the channel watch. This should never happen.\n")
|
|
return fmt.Errorf("Wat? Why am I here?")
|
|
|
|
}, &backoff.ZeroBackOff{})
|
|
}
|