Golang 语言中 kafka 客户端库 Sarama

01、介绍

Apache Kafka 是一款开源的消息引擎系统。它在项目中的作用主要是削峰填谷和解耦。本文我们只介绍 Apache Kafka 的 Golang 客户端库 Sarama。Sarama 是 MIT 许可的 Apache Kafka 0.8 及更高版本的 Golang 客户端库。

如果读者朋友对 Apache Kafka 服务端还不了解，建议先阅读官方文档中的入门部分，本文使用的版本是 Apache Kafka 2.8。

02、生产者

我们可以使用 Sarama 库的 AsyncProducer 或 SyncProducer 生产消息。在大多数情况下首选使用 AsyncProducer 生产消息。它通过一个 channel 接收消息，并在后台尽可能高效的异步生产消息。

SyncProducer 发送 Kafka 消息后阻塞，直到接收到 ACK 确认。SyncProducer 有两个警告：它通常效率较低，并且实际的耐用性保证取决于 Producer.RequiredAcks 的配置值。在某些配置中，有时仍会丢失由 SyncProducer 确认的消息，但是使用比较简单。

为了读者朋友们容易理解，本文我们介绍 SyncProducer 作为生产者的使用方式。如果读者朋友想了解 AsyncProducer 作为生产者的使用方式，请参考官方文档。

使用 SyncProducer 作为生产者的示例代码：

1. func sendMessage (brokerAddr []string, config *sarama.Config, topic string, value sarama.Encoder) { 
2.  producer, err := sarama.NewSyncProducer(brokerAddr, config) 
3.  if err != nil { 
4.   fmt.Println(err) 
5.   return 
6.  } 
7.  defer func() { 
8.   if err = producer.Close(); err != nil { 
9.    fmt.Println(err) 
10.    return 
11.   } 
12.  }() 
13.  msg := &sarama.ProducerMessage{ 
14.   Topic: topic, 
15.   Value: value, 
16.  } 
17.  partition, offset, err := producer.SendMessage(msg) 
18.  if err != nil { 
19.   fmt.Println(err) 
20.   return 
21.  } 
22.  fmt.Printf("partition:%d offset:%d\n", partition, offset) 
23. } 

阅读上面这段代码，我们调用 NewSyncProducer() 创建一个新的 SyncProducer，给定 broker 地址和配置信息。调用 SendMessage() 生产给定的消息，并且仅在生产成功或失败时返回。它将返回分区(Partition)和生产的消息的偏移量(Offset)，如果消息生产失败，则返回错误。

需要注意的是，为了避免泄露，必须在生产者上调用 Close()，因为当它超出范围时，可能不会自动垃圾回收。

03、消费者

我们可以使用 Sarama 库的消费者 Consumer 或消费者组 ConsumerGroup API 消费消息。为了读者朋友们容易理解，本文我们介绍使用 Consumer 消费消息。

Consumer 管理 PartitionConsumers，该 PartitionConsumers 处理来自 brokers 的 Kafka 消息。

Consumer 消费消息的示例代码：

1. func consumer (brokenAddr []string, topic string, partition int32, offset int64) { 
2.  consumer, err := sarama.NewConsumer(brokenAddr, nil) 
3.  if err != nil { 
4.   fmt.Println(err) 
5.   return 
6.  } 
7.  defer func() { 
8.   if err = consumer.Close(); err != nil { 
9.    fmt.Println(err) 
10.    return 
11.   } 
12.  }() 
13.  partitionConsumer, err := consumer.ConsumePartition(topic, partition, offset) 
14.  if err != nil { 
15.   fmt.Println(err) 
16.   return 
17.  } 
18.  defer func() { 
19.   if err = partitionConsumer.Close(); err != nil { 
20.    fmt.Println(err) 
21.    return 
22.   } 
23.  }() 
24.  for msg := range partitionConsumer.Messages() { 
25.   fmt.Printf("partition:%d offset:%d key:%s val:%s\n", msg.Partition, msg.Offset, msg.Key, msg.Value) 
26.  } 
27. } 

阅读上面这段代码，我们调用 NewConsumer() 创建一个新的 consumer，给定 broker 地址和配置信息。调用 ConsumePartition() 创建 PartitionConsumer，给定 topic、partition 和 offset。PartitionConsumer 处理来自给定 topic 和 partition 的 Kafka 消息。

需要注意的是，为了防止泄露，必须调用 consumer 和 partitionConsumer 的 Close()，因为当它超出范围时，可能不会自动垃圾回收。

04、总结

本文主要介绍如何使用 Apache Kafka 的 Golang 语言客户端库 Sarama 生产和消费 Kafka 消息。关于生产者和消费者，分别列举了一个简单示例。除此之外，Sarama 库还提供了很多其它 Api，感兴趣的读者朋友可以阅读官方文档了解更多。

原文链接：https://mp.weixin.qq.com/s?__biz=MzA4Mjc1NTMyOQ==&mid=2247484783&idx=1&sn=fe8416af18eca365f6ea7875943ba365&utm_source=tuicool&utm_medium=referral