tiennm99/zfoo
1
0
Fork 0
mirror of https://github.com/tiennm99/zfoo.git synced 2026-08-05 20:23:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #124 from 1281956265/main

Browse Source 
perf[orm]: for循环解决分布式id生成器创建时duplicate key问题
This commit is contained in:
godotg
2024-07-26 14:35:01 +08:00
committed by GitHub
parent fd92263427 267cbbbff2
commit 9a336d8ab5
1 changed files with 15 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
orm/src/main/java/com/zfoo/orm/util/MongoIdUtils.java
+15 -2
View File
@@ -22,6 +22,8 @@ import com.zfoo.protocol.util.AssertionUtils;
import com.zfoo.protocol.util.StringUtils;
import org.bson.Document;
import org.bson.conversions.Bson;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.util.ArrayList;
import java.util.function.Consumer;
@@ -32,6 +34,8 @@ import java.util.function.Consumer;
*/
public abstract class MongoIdUtils {
private static final Logger logger = LoggerFactory.getLogger(MongoIdUtils.class);
private static final long INIT_ID = 1L;
private static final String COLLECTION_NAME = "uuid";
@@ -57,9 +61,18 @@ public abstract class MongoIdUtils {
Document inc = new Document("$inc", new Document(COUNT, 1L));
Document setOnInsert = new Document("$setOnInsert", new Document("_id", documentName));
var document = collection.findOneAndUpdate(query, Updates.combine(inc, setOnInsert), new FindOneAndUpdateOptions().upsert(true));
return null == document ? INIT_ID : document.getLong(COUNT) + 1;
// 报错后重试获取id，在大并发create这个key的时候mongodb总是会报错一次“duplicate key error!” 所以重试一次
for (int i = 0; i < 2; i++) {
try {
var document = collection.findOneAndUpdate(query, Updates.combine(inc, setOnInsert), new FindOneAndUpdateOptions().upsert(true));
return null == document ? INIT_ID : document.getLong(COUNT) + 1;
} catch (Throwable throwable) {
logger.info("getIncrementIdFromMongo error! retry! ", throwable);
}
}
throw new RuntimeException("getIncrementIdFromMongo error!");
}
public static long getIncrementIdFromMongoDefault(String documentName) {