/
main.go
108 lines (89 loc) · 1.89 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
package main
import (
"bufio"
"fmt"
kafka "github.com/Shopify/sarama"
"github.com/jeffchao/gomkafka/gomkafka"
"log"
"os"
"os/signal"
"strings"
"syscall"
"time"
)
func run() error {
go handleSignals()
err := work()
if err != nil {
return err
}
return nil
}
func initConfig() (*gomkafka.KafkaConfig, error) {
config := &gomkafka.KafkaConfig{}
if len(os.Args) != 4 {
printUsage()
os.Exit(2)
}
config.ClientID = os.Args[1]
for _, h := range strings.Split(os.Args[2], ",") {
config.Hosts = append(config.Hosts, h)
}
config.Topic = os.Args[3]
if config.ClientID == "" || len(config.Hosts) == 0 || config.Topic == "" {
printUsage()
os.Exit(2)
}
return config, nil
}
func printUsage() {
fmt.Printf("Usage: gomkafka id hosts topic\n")
fmt.Printf("\tid\tKafka client id (REQUIRED)\n")
fmt.Printf("\thosts\tComma-separated list of host:port pairs (REQUIRED)\n")
fmt.Printf("\ttopic\tKafka topic (REQUIRED)\n")
}
func work() error {
config, err := initConfig()
if err != nil {
return err
}
client, producer, err := gomkafka.Gomkafka(config)
if err != nil {
panic(err)
}
defer client.Close()
defer producer.Close()
in := bufio.NewReader(os.Stdin)
for {
msg, err := in.ReadString('\n')
if err != nil {
return err
}
err = producer.SendMessage("monitoring", nil, kafka.StringEncoder(msg))
if err != nil {
return err
}
time.Sleep(1 * time.Millisecond)
}
}
func handleSignals() {
signals := make(chan os.Signal, 1)
signal.Notify(signals, syscall.SIGUSR1, syscall.SIGTERM, syscall.SIGKILL, os.Interrupt)
for s := range signals {
switch s {
case syscall.SIGUSR1, syscall.SIGTERM, syscall.SIGKILL, os.Interrupt:
// Catch signals that might terminate the process on behalf all goroutines.
quit()
}
}
}
func quit() {
// Perform any necessary cleanup here.
os.Exit(1)
}
func main() {
err := run()
if err != nil {
log.Fatalf("%+v", err)
}
}