我們?cè)谑褂?Kafka 時(shí)，最簡(jiǎn)單、最常用的方式是只設(shè)置 topic（主題）和 value（消息體），如下所示：這樣的話(huà)獲取消息的代碼也很簡(jiǎn)單，如下所示：

@KafkaListener(topics = "mytopic", groupId = "my-group")
public void listen(String data) {
    System.out.println("監(jiān)聽(tīng)到消息：" + data);
}

但是，除了我們可以設(shè)置和傳遞 topic 和 value 之外，我們還可以傳遞 key，如下圖所示：那問(wèn)題來(lái)了，發(fā)送消息時(shí)設(shè)置這個(gè) key 有什么用呢？

key的作用

發(fā)送消息時(shí)，設(shè)置 key 的作用如下：

1.決定分區(qū)

當(dāng)生產(chǎn)者發(fā)送消息時(shí)，如果指定了 key，Kafka 會(huì)根據(jù) key 的 hash 值來(lái)決定這條消息應(yīng)該發(fā)送到哪個(gè)分區(qū)。

“

如果沒(méi)有指定 key，Kafka 會(huì)采用輪詢(xún)（早期版本）或隨機(jī)（最新版本）的方式將消息分配到其他分區(qū)中。

分區(qū)的具體實(shí)現(xiàn)源碼在 DefaultPartitioner 中 partition 方法中體現(xiàn)，核心源碼如下：

public int partition(String topic, Object key, byte[] keyBytes, Object value, byte[] valueBytes, Cluster cluster, int numPartitions) {
    return keyBytes == null ? this.stickyPartitionCache.partition(topic, cluster) : BuiltInPartitioner.partitionForKey(keyBytes, numPartitions);
}

指定 key 之后的分區(qū)實(shí)現(xiàn)代碼如下：

public static int partitionForKey(byte[] serializedKey, int numPartitions) {
    return Utils.toPositive(Utils.murmur2(serializedKey)) % numPartitions;
}

以上源碼的大概含義是：使用 MurmurHash2 算法對(duì)字節(jié)數(shù)組 serializedKey 進(jìn)行哈希運(yùn)算，并將其結(jié)果轉(zhuǎn)換為正數(shù)，然后對(duì) numPartitions 取模，以確定鍵在分區(qū)中的位置，返回值表示鍵所在的分區(qū)編號(hào)。

“

所以，從上述源碼可以看出，發(fā)送消息如果設(shè)置了 key 之后，會(huì)將相同 key 放到同一個(gè)分區(qū)中。

2.保證消息順序

在 Kafka 中，同一個(gè)分區(qū)中的消息是有序的。而相同的 key，根據(jù)上面的分區(qū)算法可知，它們會(huì)存放到同一個(gè)分區(qū)，這樣就能保證消息的有序性了。

3.消息過(guò)濾

對(duì)于某些應(yīng)用場(chǎng)景，消費(fèi)者可以根據(jù)消息的鍵來(lái)進(jìn)行過(guò)濾或聚合操作。例如，在實(shí)時(shí)數(shù)據(jù)分析場(chǎng)景中，可能需要對(duì)具有相同鍵的消息進(jìn)行分組處理。

Kafka 設(shè)置了 key 之后，可以通過(guò)以下方式實(shí)現(xiàn)消息過(guò)濾，如下代碼所示：

@KafkaListener(topics = "topicName", groupId = "groupId")
public void listen(String message, ConsumerRecord<?,?> record) {
    Object key = record.key();
    if (key instanceof String && ((String) key).matches("regexPattern")) {
        // 處理滿(mǎn)足正則表達(dá)式條件的消息
    }
}

也就是，我們?cè)诮邮盏较⒅螅ㄟ^(guò)對(duì) key 的正則匹配實(shí)現(xiàn)消息的過(guò)濾和聚合等操作。