Fujitsu OSS團隊和PostgreSQL開源社區合作在PG14中添加了在邏輯復制中對兩階段提交進行解密的功能。下面看看這項功能是什么？

背景

兩階段提交是事務以兩階段進行提交的一種機制。通常在分布式數據庫中用于保證一致性。事務的兩階段是PREPARE階段和COMMIT／ROLLBACK階段。PG中兩階段提交的命令是：

PREPARE TRANSACTION

COMMIT PREPARED

ROLLBACK PREPARED

PG在8．0版本已經支持了兩階段提交，10．0版本支持邏輯復制。但是邏輯復制中一直都不支持兩階段提交。單實例中已經支持了PREPARE TRANSACTION、COMMIT PREPARED和ROLLBACK PREPARED命令，但是當這些命令需要邏輯復制到備機時，他們不再保持原始含義。PREPARE TRANSACTION命令被視為NOP，而根本沒有解碼。COMMIT PREPARED命令被視為COMMIT，ROLLBACK PREPARED命令被視為ABORT。

什么是兩階段提交

兩階段提交是一種原子提交協議，有助于維護分布式數據庫之間的一致性。提供數據庫內原子性的普通提交不足以為跨數據庫的事務提供一致性。為說明這個問題，我們舉一個例子：

1） John在A銀行有300＄

2） Mark在B銀行有100＄

3） John想給Mark轉100＄

事務進行過程中，需要從A銀行提取100＄到銀行B。事務結束的時候，應該都有200＄．如果在轉賬的過程中，任何時候任何一筆交易失敗，那么賬戶狀態應該恢復到轉賬開始前的狀態。事務可能因各種原因而失敗。如果在事務提交之前發生任何中斷，則該事務會回滾。在我們的示例中，如果John的賬戶中扣除金額時發生中斷，那么中斷口John的賬戶不應該減少。這就是簡單的提交如何保持數據庫內的一致性。

但是我們考慮這樣一種情況，即從John賬戶中扣除100＄的事務在一次提交時成功，但向Mark在B銀行的賬戶中添加100＄的事務失敗而被回滾。然后此操作結束后，雖然John賬戶已扣款，但Mark將不會收到該金額。100＄消失了。在處理分布式事務時，簡單的提交有可能失敗。

分布式事務的分步執行

對于兩階段提交，其中一個數據庫充當分布式事務的協調器。

階段1

一個數據庫開始應用事務，然后做Prepare。它以prepare消息形式發送prepared事務到其它數據庫。第2個數據庫獲取到Prepare消息，然后prepare該事務。Prepare涉及事務中的修改，但不提交。這些臟數據寫到磁盤以持久化。一旦所有數據庫都prepare了事務，并且有關該事務的所有信息都存儲到磁盤上，prepare階段就完成了。

階段2

接下來，仲裁器啟動提交階段。如果第2個數據庫由于某種原因未能準備事務，則仲裁器啟動回滾階段。因此根據prepare是否成功，事務要么提交，要么回滾。在最后提交階段發生中斷是可以恢復的，因為所需的prepare事務已經寫入磁盤并可以重新應用。

兩階段提交與單實例數據庫并不相關，但若數據復制跨多個數據庫實例時，就相關了。

邏輯復制中支持兩階段提交非常重要。

功能概述

在PG14版本前，邏輯復制事務僅在事務提交后才被解碼和復制。這是為了避免復制事務可能最終被中止。

提交時解碼事務

PG14的邏輯復制支持PREPARE TRANSACTION、COMMIT PREPARED和ROOLBACK PREPARED命令。當PREPARE TRANSACTION命令解碼時，事務被解碼并復制。PREPARE TRANSACTION就像WAL SENDER中COMMIT一樣啟動事務重放和解碼。

prepare時解碼事務

我們還定義了新的插件回調，允許邏輯解碼插件支持兩階段提交。

回調函數

描述

filter＿prepare＿cb

允許插件根據PREPARE TRANSACTION命令中使用的GID過濾Prepare時不需要解碼的事務

begin＿prepare＿cb

Prepare事務的開始

prepare＿cb

當PREPARE TRANSACTION命令被解碼時調用

commit＿prepared＿cb

當COMMIT PREPARED命令解碼時調用

rollback＿prepared＿cb

當ROLLBACK PREPARED命令解碼時調用

插件修改

test＿decoding

該插件是一個邏輯解碼輸出插件，作為一個示例幫助用戶開發自己的邏輯解碼插件。test＿decoding通過邏輯解碼機制接收WAL，并將其解碼為所執行操作的文本表示。

它被修改為能夠在prepare時使用新的兩階段回調函數和解碼事務

APIs的修改

pg＿create＿logical＿replication＿slot（）

該API添加了新的選項指定slot是否支持兩階段提交。輸出插件可以使用帶有兩階段選項的復制槽以支持兩階段提交。

pg＿create＿logical＿replication＿slot（slot＿name name， plugin name ［， temporary boolean， two＿phase boolean ］ ）

案例

看下怎么檢測兩階段提交的事務解碼輸出：

1） 創建一個復制槽

使用test＿decoding作為輸出插件，傳入true，這樣slot支持兩階段提交解碼。

postgres＝＃ SELECT ＊ FROM pg＿create＿logical＿replication＿slot（＇regression＿slot＇， ＇test＿decoding＇， false， true）；

slot＿name ｜ lsn

－－－－－－－－－－－－－－－－－＋－－－－－－－－－－－

regression＿slot ｜ 0／16B1970

（1 row）

2） 創建一個表

postgres＝＃ CREATE TABLE data（id serial primary key， data text）；

CREATE TABLE

3） 檢測prepare事務和commit事務的解碼輸出內容

postgres＝＃ BEGIN；

postgres＝＊＃ INSERT INTO data（data） VALUES（＇5＇）；

postgres＝＊＃ PREPARE TRANSACTION ＇test＿prepared1＇；

postgres＝＃ SELECT ＊ FROM pg＿logical＿slot＿get＿changes（＇regression＿slot＇， NULL， NULL）；

lsn ｜ xid ｜ data

－－－－－－－－－－－＋－－－－－＋－－－－－－－－－－－－－－－－－

0／1689DC0 ｜ 529 ｜ BEGIN 529

0／1689DC0 ｜ 529 ｜ table public．data： INSERT： id［integer］：3 data［text］：＇5＇

0／1689FC0 ｜ 529 ｜ PREPARE TRANSACTION ＇test＿prepared1＇， txid 529

（3 rows）

postgres＝＃ COMMIT PREPARED ＇test＿prepared1＇；

postgres＝＃ select ＊ from pg＿logical＿slot＿get＿changes（＇regression＿slot＇， NULL， NULL）；

lsn ｜ xid ｜ data

－－－－－－－－－－－＋－－－－－＋－－－－－－－－－－－－－－－－－－

0／168A060 ｜ 529 ｜ COMMIT PREPARED ＇test＿prepared1＇， txid 529

（4 rows）

postgres＝＃ select ＊ from data；

id ｜ data

－－－－＋－－－－－－

1 ｜ 5

（1 row）

未來

PG14對此功能的更改，有了解碼器端的基礎架構，允許在prepare時解碼兩階段提交。我們還修改了test＿decoding插件以利用此基礎架構。

下一步就是把對兩階段的支持實現到PG內部最大的邏輯解碼插件－－pgoutput插件中。這個插件支持邏輯復制的PUBLISHER／SUBSCRIBER 模式。他是邏輯復制中使用最廣泛的插件。富士通OSS團隊正在和開源社區合作，以在PG15中添加此功能。

對于分布式數據庫中的兩階段事務，PG也需要支持：備機通知主機PREPARE失敗了，發起回滾。這種反饋機制在PG中不支持，是未來改進的方向之一。