75 lines
1.5 KiB
Go
75 lines
1.5 KiB
Go
|
// vi:ts=4:sts=4:sw=4:noet:tw=72
|
||
|
|
//
|
||
|
|
// logforward
|
||
|
|
//
|
||
|
|
// Copyright (c) 2018 Andreas Neue <an@dnix.de>
|
||
|
|
|
||
|
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bufio"
|
||
|
|
"bytes"
|
||
|
|
"compress/gzip"
|
||
|
|
"flag"
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"git.dnix.de/an/xlog"
|
||
|
|
goscp "github.com/bramvdbogaerde/go-scp"
|
||
|
|
"github.com/bramvdbogaerde/go-scp/auth"
|
||
|
|
"golang.org/x/crypto/ssh"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
bufferLimit = flag.Int("b", 1024*1024*1024, "Buffer size limit")
|
||
|
|
scpUser = flag.String("u", "", "User")
|
||
|
|
scpHost = flag.String("h", "localhost:22", "Host:Port")
|
||
|
|
scpKey = flag.String("k", "", "Key")
|
||
|
|
filePrefix = flag.String("p", "log-", "Filename prefix")
|
||
|
|
)
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
flag.Parse()
|
||
|
|
}
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
log := ""
|
||
|
|
reader := bufio.NewReader(os.Stdin)
|
||
|
|
for {
|
||
|
|
time.Sleep(1 * time.Microsecond)
|
||
|
|
line, err := reader.ReadString('\n')
|
||
|
|
if err != nil {
|
||
|
|
xlog.Error(err.Error())
|
||
|
|
} else {
|
||
|
|
log += line
|
||
|
|
fmt.Printf("%d\r", len(log))
|
||
|
|
if len(log) >= *bufferLimit {
|
||
|
|
l := []byte(log)
|
||
|
|
go gzipAndSend(l)
|
||
|
|
log = ""
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func gzipAndSend(l []byte) {
|
||
|
|
var b bytes.Buffer
|
||
|
|
fmt.Printf("\nSending buffer ...\n")
|
||
|
|
w := gzip.NewWriter(&b)
|
||
|
|
w.Write(l)
|
||
|
|
w.Close()
|
||
|
|
r := bufio.NewReader(&b)
|
||
|
|
clientConfig, _ := auth.PrivateKey(*scpUser, *scpKey, ssh.InsecureIgnoreHostKey())
|
||
|
|
client := goscp.NewClient(*scpHost, &clientConfig)
|
||
|
|
err := client.Connect()
|
||
|
|
if err != nil {
|
||
|
|
fmt.Printf(err.Error() + "\n")
|
||
|
|
return
|
||
|
|
}
|
||
|
|
t := time.Now()
|
||
|
|
date := t.Format("20060102150405")
|
||
|
|
client.CopyFile(r, *filePrefix+date+".gz", "0644")
|
||
|
|
fmt.Printf("Buffer sent\n")
|
||
|
|
}
|