轉載自 技術世界,原文連接 Kafka設計解析(八)- Exactly Once語義與事務機制原理html
本文介紹了Kafka實現事務性的幾個階段——正好一次語義與原子操做。以後詳細分析了Kafka事務機制的實現原理,並介紹了Kafka如何處理事務相關的異常狀況,如Transaction Coordinator宕機。最後介紹了Kafka的事務機制與PostgreSQL的MVCC以及Zookeeper的原子廣播實現事務的異同。sql
本文全部Kafka原理性的描述除特殊說明外均基於Kafka 1.0.0版本。緩存
Kafka事務機制的實現主要是爲了支持服務器
《Kafka背景及架構介紹》一文中有說明Kafka在0.11.0.0以前的版本中只支持At Least Once和At Most Once語義,尚不支持Exactly Once語義。架構
可是在不少要求嚴格的場景下,如使用Kafka處理交易數據,Exactly Once語義是必須的。咱們能夠經過讓下游系統具備冪等性來配合Kafka的At Least Once語義來間接實現Exactly Once。可是:併發
所以,Kafka自己對Exactly Once語義的支持就很是必要。mvc
操做的原子性是指,多個操做要麼所有成功要麼所有失敗,不存在部分紅功部分失敗的可能。app
實現原子性操做的意義在於:分佈式
上文提到,實現Exactly Once的一種方法是讓下游系統具備冪等處理特性,而在Kafka Stream中,Kafka Producer自己就是「下游」系統,所以若是能讓Producer具備冪等處理特性,那就可讓Kafka Stream在必定程度上支持Exactly Once語義。ide
爲了實現Producer的冪等語義,Kafka引入了Producer ID(即PID)和Sequence Number。每一個新的Producer在初始化的時候會被分配一個惟一的PID,該PID對用戶徹底透明而不會暴露給用戶。
對於每一個PID,該Producer發送數據的每一個<Topic, Partition>都對應一個從0開始單調遞增的Sequence Number。
相似地,Broker端也會爲每一個<PID, Topic, Partition>維護一個序號,而且每次Commit一條消息時將其對應序號遞增。對於接收的每條消息,若是其序號比Broker維護的序號(即最後一次Commit的消息的序號)大一,則Broker會接受它,不然將其丟棄:
上述設計解決了0.11.0.0以前版本中的兩個問題:
上述冪等設計只能保證單個Producer對於同一個<Topic, Partition>的Exactly Once語義。
另外,它並不能保證寫操做的原子性——即多個寫操做,要麼所有被Commit要麼所有不被Commit。
更不能保證多個讀寫操做的的原子性。尤爲對於Kafka Stream應用而言,典型的操做便是從某個Topic消費數據,通過一系列轉換後寫回另外一個Topic,保證從源Topic的讀取與向目標Topic的寫入的原子性有助於從故障中恢復。
事務保證可以使得應用程序將生產數據和消費數據看成一個原子單元來處理,要麼所有成功,要麼所有失敗,即便該生產或消費跨多個<Topic, Partition>。
另外,有狀態的應用也能夠保證重啓後從斷點處繼續處理,也即事務恢復。
爲了實現這種效果,應用程序必須提供一個穩定的(重啓後不變)惟一的ID,也即Transaction ID。Transaction ID與PID可能一一對應。區別在於Transaction ID由用戶提供,而PID是內部的實現對用戶透明。
另外,爲了保證新的Producer啓動後,舊的具備相同Transaction ID的Producer當即失效,每次Producer經過Transaction ID拿到PID的同時,還會獲取一個單調遞增的epoch。因爲舊的Producer的epoch比新Producer的epoch小,Kafka能夠很容易識別出該Producer是老的Producer並拒絕其請求。
有了Transaction ID後,Kafka可保證:
須要注意的是,上述的事務保證是從Producer的角度去考慮的。從Consumer的角度來看,該保證會相對弱一些。尤爲是不能保證全部被某事務Commit過的全部消息都被一塊兒消費,由於:
這一節所說的事務主要指原子性,也即Producer將多條消息做爲一個事務批量發送,要麼所有成功要麼所有失敗。
爲了實現這一點,Kafka 0.11.0.0引入了一個服務器端的模塊,名爲Transaction Coordinator,用於管理Producer發送的消息的事務性。
該Transaction Coordinator維護Transaction Log,該log存於一個內部的Topic內。因爲Topic數據具備持久性,所以事務的狀態也具備持久性。
Producer並不直接讀寫Transaction Log,它與Transaction Coordinator通訊,而後由Transaction Coordinator將該事務的狀態插入相應的Transaction Log。
Transaction Log的設計與Offset Log用於保存Consumer的Offset相似。
許多基於Kafka的應用,尤爲是Kafka Stream應用中同時包含Consumer和Producer,前者負責從Kafka中獲取消息,後者負責將處理完的數據寫回Kafka的其它Topic中。
爲了實現該場景下的事務的原子性,Kafka須要保證對Consumer Offset的Commit與Producer對發送消息的Commit包含在同一個事務中。不然,若是在兩者Commit中間發生異常,根據兩者Commit的順序可能會形成數據丟失和數據重複:
爲了區分寫入Partition的消息被Commit仍是Abort,Kafka引入了一種特殊類型的消息,即Control Message。該類消息的Value內不包含任何應用相關的數據,而且不會暴露給應用程序。它只用於Broker與Client間的內部通訊。
對於Producer端事務,Kafka以Control Message的形式引入一系列的Transaction Marker。Consumer便可經過該標記斷定對應的消息被Commit了仍是Abort了,而後結合該Consumer配置的隔離級別決定是否應該將該消息返回給應用程序。
Producer<String, String> producer = new KafkaProducer<String, String>(props); // 初始化事務,包括結束該Transaction ID對應的未完成的事務(若是有) // 保證新的事務在一個正確的狀態下啓動 producer.initTransactions(); // 開始事務 producer.beginTransaction(); // 消費數據 ConsumerRecords<String, String> records = consumer.poll(100); try{ // 發送數據 producer.send(new ProducerRecord<String, String>("Topic", "Key", "Value")); // 發送消費數據的Offset,將上述數據消費與數據發送歸入同一個Transaction內 producer.sendOffsetsToTransaction(offsets, "group1"); // 數據發送及Offset發送均成功的狀況下,提交事務 producer.commitTransaction(); } catch (ProducerFencedException | OutOfOrderSequenceException | AuthorizationException e) { // 數據發送或者Offset發送出現異常時,終止事務 producer.abortTransaction(); } finally { // 關閉Producer和Consumer producer.close(); consumer.close(); }
因爲Transaction Coordinator是分配PID和管理事務的核心,所以Producer要作的第一件事情就是經過向任意一個Broker發送FindCoordinator請求找到Transaction Coordinator的位置。
注意:只有應用程序爲Producer配置了Transaction ID時纔可以使用事務特性,也才須要這一步。另外,因爲事務性要求Producer開啓冪等特性,所以經過將transactional.id設置爲非空從而開啓事務特性的同時也須要經過將enable.idempotence設置爲true來開啓冪等特性。
找到Transaction Coordinator後,具備冪等特性的Producer必須發起InitPidRequest請求以獲取PID。
注意:只要開啓了冪等特性即必須執行該操做,而無須考慮該Producer是否開啓了事務特性。
若是事務特性被開啓
InitPidRequest會發送給Transaction Coordinator。若是Transaction Coordinator是第一次收到包含有該Transaction ID的InitPidRequest請求,它將會把該<TransactionID, PID>存入Transaction Log,如上圖中步驟2.1所示。這樣可保證該對應關係被持久化,從而保證即便Transaction Coordinator宕機該對應關係也不會丟失。
除了返回PID外,InitPidRequest還會執行以下任務:
注意:InitPidRequest的處理過程是同步阻塞的。一旦該調用正確返回,Producer便可開始新的事務。
另外,若是事務特性未開啓,InitPidRequest可發送至任意Broker,而且會獲得一個全新的惟一的PID。該Producer將只能使用冪等特性以及單一Session內的事務特性,而不能使用跨Session的事務特性。
Kafka從0.11.0.0版本開始,提供beginTransaction()方法用於開啓一個事務。調用該方法後,Producer本地會記錄已經開啓了事務,但Transaction Coordinator只有在Producer發送第一條消息後才認爲事務已經開啓。
這一階段,包含了整個事務的數據處理過程,而且包含了多種請求。
AddPartitionsToTxnRequest
一個Producer可能會給多個<Topic, Partition>發送數據,給一個新的<Topic, Partition>發送數據前,它須要先向Transaction Coordinator發送AddPartitionsToTxnRequest。
Transaction Coordinator會將該<Transaction, Topic, Partition>存於Transaction Log內,並將其狀態置爲BEGIN,如上圖中步驟4.1所示。有了該信息後,咱們才能夠在後續步驟中爲每一個<Topic, Partition>設置COMMIT或者ABORT標記(如上圖中步驟5.2所示)。
另外,若是該<Topic, Partition>爲該事務中第一個<Topic, Partition>,Transaction Coordinator還會啓動對該事務的計時(每一個事務都有本身的超時時間)。
ProduceRequest
Producer經過一個或多個ProduceRequest發送一系列消息。除了應用數據外,該請求還包含了PID,epoch,和Sequence Number。該過程如上圖中步驟4.2所示。
AddOffsetsToTxnRequest
爲了提供事務性,Producer新增了sendOffsetsToTransaction方法,該方法將多組消息的發送和消費放入同一批處理內。
該方法先判斷在當前事務中該方法是否已經被調用並傳入了相同的Group ID。如果,直接跳到下一步;若不是,則向Transaction Coordinator發送AddOffsetsToTxnRequests請求,Transaction Coordinator將對應的全部<Topic, Partition>存於Transaction Log中,並將其狀態記爲BEGIN,如上圖中步驟4.3所示。該方法會阻塞直到收到響應。
TxnOffsetCommitRequest
做爲sendOffsetsToTransaction方法的一部分,在處理完AddOffsetsToTxnRequest後,Producer也會發送TxnOffsetCommit請求給Consumer Coordinator從而將本事務包含的與讀操做相關的各<Topic, Partition>的Offset持久化到內部的__consumer_offsets中,如上圖步驟4.4所示。
在此過程當中,Consumer Coordinator會經過PID和對應的epoch來驗證是否應該容許該Producer的該請求。
這裏須要注意:
一旦上述數據寫入操做完成,應用程序必須調用KafkaProducer的commitTransaction方法或者abortTransaction方法以結束當前事務。
EndTxnRequest
commitTransaction方法使得Producer寫入的數據對下游Consumer可見。abortTransaction方法經過Transaction Marker將Producer寫入的數據標記爲Aborted狀態。下游的Consumer若是將isolation.level設置爲READ_COMMITTED,則它讀到被Abort的消息後直接將其丟棄而不會返回給客戶程序,也即被Abort的消息對應用程序不可見。
不管是Commit仍是Abort,Producer都會發送EndTxnRequest請求給Transaction Coordinator,並經過標誌位標識是應該Commit仍是Abort。
收到該請求後,Transaction Coordinator會進行以下操做
補充說明:對於commitTransaction方法,它會在發送EndTxnRequest以前先調用flush方法以確保全部發送出去的數據都獲得相應的ACK。對於abortTransaction方法,在發送EndTxnRequest以前直接將當前Buffer中的事務性消息(若是有)所有丟棄,但必須等待全部被髮送但還沒有收到ACK的消息發送完成。
上述第二步是實現將一組讀操做與寫操做做爲一個事務處理的關鍵。由於Producer寫入的數據Topic以及記錄Comsumer Offset的Topic會被寫入相同的Transactin Marker,因此這一組讀操做與寫操做要麼所有COMMIT要麼所有ABORT。
WriteTxnMarkerRequest
上面提到的WriteTxnMarkerRequest由Transaction Coordinator發送給當前事務涉及到的每一個<Topic, Partition>的Leader。收到該請求後,對應的Leader會將對應的COMMIT(PID)或者ABORT(PID)控制信息寫入日誌,如上圖中步驟5.2所示。
該控制消息向Broker以及Consumer代表對應PID的消息被Commit了仍是被Abort了。
這裏要注意,若是事務也涉及到__consumer_offsets,即該事務中有消費數據的操做且將該消費的Offset存於__consumer_offsets中,Transaction Coordinator也須要向該內部Topic的各Partition的Leader發送WriteTxnMarkerRequest從而寫入COMMIT(PID)或COMMIT(PID)控制信息。
寫入最終的COMPLETE_COMMIT
或COMPLETE_ABORT
消息
寫完全部的Transaction Marker後,Transaction Coordinator會將最終的COMPLETE_COMMIT或COMPLETE_ABORT消息寫入Transaction Log中以標明該事務結束,如上圖中步驟5.3所示。
此時,Transaction Log中全部關於該事務的消息所有能夠移除。固然,因爲Kafka內數據是Append Only的,不可直接更新和刪除,這裏說的移除只是將其標記爲null從而在Log Compact時再也不保留。
另外,COMPLETE_COMMIT或COMPLETE_ABORT的寫入並不須要獲得全部Rreplica的ACK,由於若是該消息丟失,能夠根據事務協議重發。
補充說明,若是參與該事務的某些<Topic, Partition>在被寫入Transaction Marker前不可用,它對READ_COMMITTED的Consumer不可見,但不影響其它可用<Topic, Partition>的COMMIT或ABORT。在該<Topic, Partition>恢復可用後,Transaction Coordinator會從新根據PREPARE_COMMIT或PREPARE_ABORT向該<Topic, Partition>發送Transaction Marker。
InvalidProducerEpoch
這是一種Fatal Error,它說明當前Producer是一個過時的實例,有Transaction ID相同但epoch更新的Producer實例被建立並使用。此時Producer會中止並拋出Exception。
InvalidPidMapping
Transaction Coordinator沒有與該Transaction ID對應的PID。此時Producer會經過包含有Transaction ID的InitPidRequest請求建立一個新的PID。Transaction Coordinator沒有與該Transaction ID對應的PID。此時Producer會經過包含有Transaction ID的InitPidRequest請求建立一個新的PID。
NotCorrdinatorForGTransactionalId
該Transaction Coordinator不負責該當前事務。Producer會經過FindCoordinatorRequest請求從新尋找對應的Transaction Coordinator。該Transaction Coordinator不負責該當前事務。Producer會經過FindCoordinatorRequest請求從新尋找對應的Transaction Coordinator。
InvalidTxnRequest
違反了事務協議。正確的Client實現不該該出現這種Exception。若是該異常發生了,用戶須要檢查本身的客戶端實現是否有問題。
CoordinatorNotAvailable
Transaction Coordinator仍在初始化中。Producer只須要重試便可。
DuplicateSequenceNumber
發送的消息的序號低於Broker預期。該異常說明該消息已經被成功處理過,Producer能夠直接忽略該異常並處理下一條消息
InvalidSequenceNumber
這是一個Fatal Error,它說明發送的消息中的序號大於Broker預期。此時有兩種可能
InvalidTransactionTimeout
InitPidRequest調用出現的Fatal Error。它代表Producer傳入的timeout時間不在可接受範圍內,應該中止Producer並報告給用戶。
Transaction Coordinator
失敗PREPARE_COMMIT/PREPARE_ABORT
前失敗Producer經過FindCoordinatorRequest找到新的Transaction Coordinator,並經過EndTxnRequest請求發起COMMIT或ABORT流程,新的Transaction Coordinator繼續處理EndTxnRequest請求——寫PREPARE_COMMIT或PREPARE_ABORT,寫Transaction Marker,寫COMPLETE_COMMIT或COMPLETE_ABORT。
PREPARE_COMMIT/PREPARE_ABORT
後失敗此時舊的Transaction Coordinator可能已經成功寫入部分Transaction Marker。新的Transaction Coordinator會重複這些操做,因此部分Partition中可能會存在重複的COMMIT或ABORT,但只要該Producer在此期間沒有發起新的事務,這些重複的Transaction Marker就不是問題。
COMPLETE_COMMIT/ABORT
後失敗舊的Transaction Coordinator可能已經寫完了COMPLETE_COMMIT或COMPLETE_ABORT但在返回EndTxnRequest以前失敗。該場景下,新的Transaction Coordinator會直接給Producer返回成功。
transaction.timeout.ms
當Producer失敗時,Transaction Coordinator必須可以主動的讓某些進行中的事務過時。不然沒有Producer的參與,Transaction Coordinator沒法判斷這些事務應該如何處理,這會形成:
爲了不上述問題,Transaction Coordinator會週期性遍歷內存中的事務狀態Map,並執行以下操做
BEGIN
而且其最後更新時間與當前時間差大於transaction.remove.expired.transaction.cleanup.interval.ms(默認值爲1小時),則主動將其終止:1)未避免原Producer臨時恢復與當前終止流程衝突,增長該Producer對應的PID的epoch,並確保將該更新的信息寫入Transaction Log;2)以更新後的epoch回滾事務,從而使得該事務相關的全部Broker都更新其緩存的該PID的epoch從而拒絕舊Producer的寫操做Transaction ID
某Transaction ID的Producer可能很長時間再也不發送數據,Transaction Coordinator不必再保存該Transaction ID與PID等的映射,不然可能會形成大量的資源浪費。所以須要有一個機制探測再也不活躍的Transaction ID並將其信息刪除。
Transaction Coordinator會週期性遍歷內存中的Transaction ID與PID映射,若是某Transaction ID沒有對應的正在進行中的事務而且它對應的最後一個事務的結束時間與當前時間差大於transactional.id.expiration.ms(默認值是7天),則將其從內存中刪除並在Transaction Log中將其對應的日誌的值設置爲null從而使得Log Compact可將其記錄刪除。
Kafka的事務機制與《MVCC PostgreSQL實現事務和多版本併發控制的精華》一文中介紹的PostgreSQL經過MVCC實現事務的機制很是相似,對於事務的回滾,並不須要刪除已寫入的數據,都是將寫入數據的事務標記爲Rollback/Abort從而在讀數據時過濾該數據。
Kafka的事務機制與《分佈式事務(一)兩階段提交及JTA》一文中所介紹的兩階段提交機制看似類似,都分PREPARE階段和最終COMMIT階段,但又有很大不一樣。
Zookeeper的原子廣播協議與兩階段提交以及Kafka事務機制有類似之處,但又有各自的特色