Create main.go
This commit is contained in:
commit
e3875b3f08
93
main.go
Normal file
93
main.go
Normal file
|
@ -0,0 +1,93 @@
|
||||||
|
//
|
||||||
|
// tail2redis
|
||||||
|
// Go Namhyeon <abuse@catswords.net>
|
||||||
|
//
|
||||||
|
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
"context"
|
||||||
|
|
||||||
|
"github.com/urfave/cli/v2"
|
||||||
|
"github.com/nxadm/tail"
|
||||||
|
"github.com/redis/go-redis/v9"
|
||||||
|
)
|
||||||
|
|
||||||
|
var ctx = context.Background()
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
app := &cli.App{
|
||||||
|
Name: "tail2redis",
|
||||||
|
Usage: "Tailing a file to redis PubSub",
|
||||||
|
Flags: []cli.Flag{
|
||||||
|
&cli.StringFlag{
|
||||||
|
Name: "file",
|
||||||
|
Aliases: []string{"f"},
|
||||||
|
Value: "/var/log/messages",
|
||||||
|
Usage: "Choose a file",
|
||||||
|
},
|
||||||
|
&cli.StringFlag{
|
||||||
|
Name: "host",
|
||||||
|
Aliases: []string{"H"},
|
||||||
|
Value: "localhost",
|
||||||
|
Usage: "Redis host",
|
||||||
|
},
|
||||||
|
&cli.IntFlag{
|
||||||
|
Name: "port",
|
||||||
|
Aliases: []string{"P"},
|
||||||
|
Value: 6379,
|
||||||
|
Usage: "Redis host",
|
||||||
|
},
|
||||||
|
&cli.StringFlag{
|
||||||
|
Name: "password",
|
||||||
|
Value: "",
|
||||||
|
Aliases: []string{"p"},
|
||||||
|
Usage: "Redis password",
|
||||||
|
},
|
||||||
|
&cli.StringFlag{
|
||||||
|
Name: "channel",
|
||||||
|
Aliases: []string{"c"},
|
||||||
|
Value: "tail2redis",
|
||||||
|
Usage: "Redis PubSub channel",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
Action: func(c *cli.Context) error {
|
||||||
|
file := c.String("file")
|
||||||
|
host := c.String("host")
|
||||||
|
port := c.Int("port")
|
||||||
|
password := c.String("password")
|
||||||
|
channel := c.String("channel")
|
||||||
|
|
||||||
|
// Connect to Redis server
|
||||||
|
rdb := redis.NewClient(&redis.Options{
|
||||||
|
Addr: fmt.Sprintf("%s:%d", host, port),
|
||||||
|
Password: password, // no password set
|
||||||
|
DB: 0, // use default DB
|
||||||
|
})
|
||||||
|
|
||||||
|
|
||||||
|
// Tailing a file
|
||||||
|
t, err := tail.TailFile(file, tail.Config{Follow: true, ReOpen: true})
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
for line := range t.Lines {
|
||||||
|
err2 := rdb.Publish(ctx, channel, line.Text).Err()
|
||||||
|
if err2 != nil {
|
||||||
|
fmt.Println(err2)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := app.Run(os.Args); err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user