在高并發(fā)下怎么保證接口的冪等性？

前言

接口冪等性問題，對于開發(fā)人員來說，是一個跟語言無關(guān)的公共問題。本文分享了一些解決這類問題非常實用的辦法，絕大部分內(nèi)容我在項目中實踐過的，給有需要的小伙伴一個參考。

不知道你有沒有遇到過這些場景：

有時我們在填寫某些form表單時，保存按鈕不小心快速點了兩次，表中竟然產(chǎn)生了兩條重復(fù)的數(shù)據(jù)，只是id不一樣。

我們在項目中為了解決接口超時問題，通常會引入了重試機制。第一次請求接口超時了，請求方?jīng)]能及時獲取返回結(jié)果（此時有可能已經(jīng)成功了），為了避免返回錯誤的結(jié)果（這種情況不可能直接返回失敗吧？），于是會對該請求重試幾次，這樣也會產(chǎn)生重復(fù)的數(shù)據(jù)。

mq消費者在讀取消息時，有時候會讀取到重復(fù)消息（至于什么原因這里先不說，有興趣的小伙伴，可以找我私聊），如果處理不好，也會產(chǎn)生重復(fù)的數(shù)據(jù)。

沒錯，這些都是冪等性問題。

接口冪等性是指用戶對于同一操作發(fā)起的一次請求或者多次請求的結(jié)果是一致的，不會因為多次點擊而產(chǎn)生了副作用。

這類問題多發(fā)于接口的：

insert操作，這種情況下多次請求，可能會產(chǎn)生重復(fù)數(shù)據(jù)。

update操作，如果只是單純的更新數(shù)據(jù)，比如：update user set status=1 where id=1，是沒有問題的。如果還有計算，比如：update user set status=status+1 where id=1，這種情況下多次請求，可能會導(dǎo)致數(shù)據(jù)錯誤。

那么我們要如何保證接口冪等性？本文將會告訴你答案。

1. insert前先select

通常情況下，在保存數(shù)據(jù)的接口中，我們?yōu)榱朔乐巩a(chǎn)生重復(fù)數(shù)據(jù)，一般會在insert前，先根據(jù)name或code字段select一下數(shù)據(jù)。如果該數(shù)據(jù)已存在，則執(zhí)行update操作，如果不存在，才執(zhí)行 insert操作。

該方案可能是我們平時在防止產(chǎn)生重復(fù)數(shù)據(jù)時，使用最多的方案。但是該方案不適用于并發(fā)場景，在并發(fā)場景中，要配合其他方案一起使用，否則同樣會產(chǎn)生重復(fù)數(shù)據(jù)。我在這里提一下，是為了避免大家踩坑。

2. 加悲觀鎖

在支付場景中，用戶A的賬號余額有150元，想轉(zhuǎn)出100元，正常情況下用戶A的余額只剩50元。一般情況下，sql是這樣的：

update user amount = amount-100 where id=123;

如果出現(xiàn)多次相同的請求，可能會導(dǎo)致用戶A的余額變成負數(shù)。這種情況，用戶A來可能要哭了。于此同時，系統(tǒng)開發(fā)人員可能也要哭了，因為這是很嚴重的系統(tǒng)bug。

為了解決這個問題，可以加悲觀鎖，將用戶A的那行數(shù)據(jù)鎖住，在同一時刻只允許一個請求獲得鎖，更新數(shù)據(jù)，其他的請求則等待。

通常情況下通過如下sql鎖住單行數(shù)據(jù)：

select * from user id=123 for update;

具體流程如下：

具體步驟：

多個請求同時根據(jù)id查詢用戶信息。

判斷余額是否不足100，如果余額不足，則直接返回余額不足。

如果余額充足，則通過for update再次查詢用戶信息，并且嘗試獲取鎖。

只有第一個請求能獲取到行鎖，其余沒有獲取鎖的請求，則等待下一次獲取鎖的機會。

第一個請求獲取到鎖之后，判斷余額是否不足100，如果余額足夠，則進行update操作。

如果余額不足，說明是重復(fù)請求，則直接返回成功。

需要特別注意的是：如果使用的是mysql數(shù)據(jù)庫，存儲引擎必須用innodb，因為它才支持事務(wù)。此外，這里id字段一定要是主鍵或者唯一索引，不然會鎖住整張表。

悲觀鎖需要在同一個事務(wù)操作過程中鎖住一行數(shù)據(jù)，如果事務(wù)耗時比較長，會造成大量的請求等待，影響接口性能。 此外，每次請求接口很難保證都有相同的返回值，所以不適合冪等性設(shè)計場景，但是在防重場景中是可以的使用的。 在這里順便說一下，防重設(shè)計和冪等設(shè)計，其實是有區(qū)別的。防重設(shè)計主要為了避免產(chǎn)生重復(fù)數(shù)據(jù)，對接口返回沒有太多要求。而冪等設(shè)計除了避免產(chǎn)生重復(fù)數(shù)據(jù)之外，還要求每次請求都返回一樣的結(jié)果。

3. 加樂觀鎖

既然悲觀鎖有性能問題，為了提升接口性能，我們可以使用樂觀鎖。需要在表中增加一個timestamp或者version字段，這里以version字段為例。

在更新數(shù)據(jù)之前先查詢一下數(shù)據(jù)：

select id,amount,version from user id=123;

如果數(shù)據(jù)存在，假設(shè)查到的version等于1，再使用id和version字段作為查詢條件更新數(shù)據(jù)：

update user set amount=amount+100,version=version+1where id=123 and version=1;

更新數(shù)據(jù)的同時version+1，然后判斷本次update操作的影響行數(shù)，如果大于0，則說明本次更新成功，如果等于0，則說明本次更新沒有讓數(shù)據(jù)變更。

由于第一次請求version等于1是可以成功的，操作成功后version變成2了。這時如果并發(fā)的請求過來，再執(zhí)行相同的sql：

update user setamount=amount+100,version=version+1where id=123 and version=1;

該update操作不會真正更新數(shù)據(jù)，最終sql的執(zhí)行結(jié)果影響行數(shù)是0，因為version已經(jīng)變成2了，where中的version=1肯定無法滿足條件。但為了保證接口冪等性，接口可以直接返回成功，因為version值已經(jīng)修改了，那么前面必定已經(jīng)成功過一次，后面都是重復(fù)的請求。

具體流程如下：

具體步驟：

先根據(jù)id查詢用戶信息，包含version字段

根據(jù)id和version字段值作為where條件的參數(shù)，更新用戶信息，同時version+1

判斷操作影響行數(shù)，如果影響1行，則說明是一次請求，可以做其他數(shù)據(jù)操作。

如果影響0行，說明是重復(fù)請求，則直接返回成功。

4. 加唯一索引

絕大數(shù)情況下，為了防止重復(fù)數(shù)據(jù)的產(chǎn)生，我們都會在表中加唯一索引，這是一個非常簡單，并且有效的方案。

alter table `order` add UNIQUE KEY `un_code` (`code`);

加了唯一索引之后，第一次請求數(shù)據(jù)可以插入成功。但后面的相同請求，插入數(shù)據(jù)時會報Duplicate entry '002' for key 'order.un_code異常，表示唯一索引有沖突。

雖說拋異常對數(shù)據(jù)來說沒有影響，不會造成錯誤數(shù)據(jù)。但是為了保證接口冪等性，我們需要對該異常進行捕獲，然后返回成功。

如果是java程序需要捕獲：DuplicateKeyException異常，如果使用了spring框架還需要捕獲：MySQLIntegrityConstraintViolationException異常。

具體流程圖如下：

具體步驟：

用戶通過瀏覽器發(fā)起請求，服務(wù)端收集數(shù)據(jù)。

將該數(shù)據(jù)插入mysql

判斷是否執(zhí)行成功，如果成功，則操作其他數(shù)據(jù)（可能還有其他的業(yè)務(wù)邏輯）。

如果執(zhí)行失敗，捕獲唯一索引沖突異常，直接返回成功。

5. 建防重表

有時候表中并非所有的場景都不允許產(chǎn)生重復(fù)的數(shù)據(jù)，只有某些特定場景才不允許。這時候，直接在表中加唯一索引，顯然是不太合適的。

針對這種情況，我們可以通過建防重表來解決問題。

該表可以只包含兩個字段：id 和 唯一索引，唯一索引可以是多個字段比如：name、code等組合起來的唯一標識，例如：susan_0001。

具體流程圖如下：

具體步驟：

用戶通過瀏覽器發(fā)起請求，服務(wù)端收集數(shù)據(jù)。

將該數(shù)據(jù)插入mysql防重表

判斷是否執(zhí)行成功，如果成功，則做mysql其他的數(shù)據(jù)操作（可能還有其他的業(yè)務(wù)邏輯）。

如果執(zhí)行失敗，捕獲唯一索引沖突異常，直接返回成功。

需要特別注意的是：防重表和業(yè)務(wù)表必須在同一個數(shù)據(jù)庫中，并且操作要在同一個事務(wù)中。

6. 根據(jù)狀態(tài)機

很多時候業(yè)務(wù)表是有狀態(tài)的，比如訂單表中有：1-下單、2-已支付、3-完成、4-撤銷等狀態(tài)。如果這些狀態(tài)的值是有規(guī)律的，按照業(yè)務(wù)節(jié)點正好是從小到大，我們就能通過它來保證接口的冪等性。

假如id=123的訂單狀態(tài)是已支付，現(xiàn)在要變成完成狀態(tài)。

update `order` set status=3 where id=123 and status=2;

第一次請求時，該訂單的狀態(tài)是已支付，值是2，所以該update語句可以正常更新數(shù)據(jù)，sql執(zhí)行結(jié)果的影響行數(shù)是1，訂單狀態(tài)變成了3。

后面有相同的請求過來，再執(zhí)行相同的sql時，由于訂單狀態(tài)變成了3，再用status=2作為條件，無法查詢出需要更新的數(shù)據(jù)，所以最終sql執(zhí)行結(jié)果的影響行數(shù)是0，即不會真正的更新數(shù)據(jù)。但為了保證接口冪等性，影響行數(shù)是0時，接口也可以直接返回成功。

具體流程圖如下：

具體步驟：

用戶通過瀏覽器發(fā)起請求，服務(wù)端收集數(shù)據(jù)。

根據(jù)id和當前狀態(tài)作為條件，更新成下一個狀態(tài)

判斷操作影響行數(shù)，如果影響了1行，說明當前操作成功，可以進行其他數(shù)據(jù)操作。

如果影響了0行，說明是重復(fù)請求，直接返回成功。

主要特別注意的是，該方案僅限于要更新的表有狀態(tài)字段，并且剛好要更新狀態(tài)字段的這種特殊情況，并非所有場景都適用。

7. 加分布式鎖

其實前面介紹過的加唯一索引或者加防重表，本質(zhì)是使用了數(shù)據(jù)庫的分布式鎖，也屬于分布式鎖的一種。但由于數(shù)據(jù)庫分布式鎖的性能不太好，我們可以改用：redis或zookeeper。

鑒于現(xiàn)在很多公司分布式配置中心改用apollo或nacos，已經(jīng)很少用zookeeper了，我們以redis為例介紹分布式鎖。

目前主要有三種方式實現(xiàn)redis的分布式鎖：

setNx命令

set命令

Redission框架

每種方案各有利弊，具體實現(xiàn)細節(jié)我就不說了，有興趣的朋友可以加我微信找我私聊。

具體流程圖如下：

具體步驟：

用戶通過瀏覽器發(fā)起請求，服務(wù)端會收集數(shù)據(jù)，并且生成訂單號code作為唯一業(yè)務(wù)字段。

使用redis的set命令，將該訂單code設(shè)置到redis中，同時設(shè)置超時時間。

判斷是否設(shè)置成功，如果設(shè)置成功，說明是第一次請求，則進行數(shù)據(jù)操作。

如果設(shè)置失敗，說明是重復(fù)請求，則直接返回成功。

需要特別注意的是：分布式鎖一定要設(shè)置一個合理的過期時間，如果設(shè)置過短，無法有效的防止重復(fù)請求。如果設(shè)置過長，可能會浪費redis的存儲空間，需要根據(jù)實際業(yè)務(wù)情況而定。

8. 獲取token

除了上述方案之外，還有最后一種使用token的方案。該方案跟之前的所有方案都有點不一樣，需要兩次請求才能完成一次業(yè)務(wù)操作。

第一次請求獲取token

第二次請求帶著這個token，完成業(yè)務(wù)操作。

具體流程圖如下：

第一步，先獲取token。

第二步，做具體業(yè)務(wù)操作。

具體步驟：

用戶訪問頁面時，瀏覽器自動發(fā)起獲取token請求。

服務(wù)端生成token，保存到redis中，然后返回給瀏覽器。

用戶通過瀏覽器發(fā)起請求時，攜帶該token。

在redis中查詢該token是否存在，如果不存在，說明是第一次請求，做則后續(xù)的數(shù)據(jù)操作。

如果存在，說明是重復(fù)請求，則直接返回成功。

在redis中token會在過期時間之后，被自動刪除。

以上方案是針對冪等設(shè)計的。

如果是防重設(shè)計，流程圖要改改：

需要特別注意的是：token必須是全局唯一的。

編輯：jq