From 8d5ba14de55cab527542cf430396d8d69bfe76f5 Mon Sep 17 00:00:00 2001 From: Craig Swank Date: Tue, 8 Oct 2019 15:36:59 -0600 Subject: [PATCH] Added kafka.Opt type (func(*kafka.Client)) --- internal/kafka/kafka.go | 4 +++- main.go | 4 ++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/internal/kafka/kafka.go b/internal/kafka/kafka.go index f2c13abb..f8a95ecd 100644 --- 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 index 9a28435f..02ee810f 100644 --- 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...)