转载自 技术世界,原文连接 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事务机制有类似之处,但又有各自的特色