-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
99 lines (78 loc) · 2.11 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
package main
import (
"fmt"
"log/slog"
"os"
"github.com/IBM/sarama"
"github.com/spf13/cobra"
"github.com/aaronjheng/kafka-cli/internal/config"
"github.com/aaronjheng/kafka-cli/internal/kafka"
)
var (
cfg *config.Config
cluster string
)
func rootCmd() *cobra.Command {
cmd := &cobra.Command{
Use: "kafka",
Short: "Command line tool for Apache Kafka",
SilenceUsage: true,
PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
cfgFilepath, err := cmd.Flags().GetString("config")
if err != nil {
return fmt.Errorf("config flag error: %w", err)
}
cfg, err = config.LoadConfig(cfgFilepath)
if err != nil {
return fmt.Errorf("config flag error: %w", err)
}
return nil
},
}
cmd.PersistentFlags().StringVarP(&cluster, "cluster", "c", "", "Cluster name to operate.")
cmd.PersistentFlags().StringP("config", "f", "", "Config file path.")
cmd.AddCommand(configCmd())
cmd.AddCommand(topicCmd())
cmd.AddCommand(groupCmd())
cmd.AddCommand(producerCmd())
cmd.AddCommand(consumerCmd())
cmd.AddCommand(versionCmd)
cmd.AddCommand(completionCmd)
return cmd
}
func main() {
// Bootstrap logging
slog.SetDefault(slog.New(slog.NewTextHandler(os.Stderr, nil)))
if err := rootCmd().Execute(); err != nil {
os.Exit(1)
}
}
func newCluster() (*kafka.Kafka, error) {
cfg, err := cfg.Cluster(cluster)
if err != nil {
return nil, fmt.Errorf("cfg.Cluster error: %w", err)
}
return kafka.New(cfg)
}
func newClusterAdmin() (sarama.ClusterAdmin, error) {
cluster, err := newCluster()
if err != nil {
return nil, fmt.Errorf("newCluster error: %w", err)
}
return sarama.NewClusterAdminFromClient(cluster)
}
func newSyncProducer() (sarama.SyncProducer, error) {
cluster, err := newCluster()
if err != nil {
return nil, fmt.Errorf("newCluster error: %w", err)
}
cluster.Config().Producer.Return.Successes = true
return sarama.NewSyncProducerFromClient(cluster)
}
func newConsumer() (sarama.Consumer, error) {
cluster, err := newCluster()
if err != nil {
return nil, fmt.Errorf("newCluster error: %w", err)
}
return sarama.NewConsumerFromClient(cluster)
}