Added kafka.Opt type (func(*kafka.Client))

cswank · Oct 8, 2019 · 8d5ba14 · 8d5ba14
1 parent 755cdac
commit 8d5ba14
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/internal/kafka/kafka.go b/internal/kafka/kafka.go
@@ -52,8 +52,10 @@ type Message struct {
 	Offset    int64     `json:"offset"`
 }
 
+type Opt func(*Client)
+
 //New returns a kafka Client.
-func New(addrs []string, opts ...func(*Client)) (*Client, error) {
+func New(addrs []string, opts ...Opt) (*Client, error) {
 	cfg, err := getConfig()
 	if err != nil {
 		return nil, err

diff --git a/main.go b/main.go
@@ -43,10 +43,10 @@ func main() {
 }
 
 func connect() *kafka.Client {
-	var opts []func(*kafka.Client)
+	var opts []kafka.Opt
 	if *decoder != "" {
 		dec := getDecoder(*decoder)
-		opts = []func(*kafka.Client){kafka.WithDecoder(dec)}
+		opts = []kafka.Opt{kafka.WithDecoder(dec)}
 	}
 
 	cli, err := kafka.New(getAddresses(*addrs), opts...)