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

perf[orm]: 分布式id生成器优雅写法

Browse Source
This commit is contained in:
box
2024-07-26 14:53:49 +08:00
parent 9a336d8ab5
commit 25f6c06faf
1 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
orm/src/main/java/com/zfoo/orm/util/MongoIdUtils.java
+9 -4
View File
@@ -57,15 +57,20 @@ public abstract class MongoIdUtils {
public static long getIncrementIdFromMongo(String collectionName, String documentName) {
var collection = OrmContext.getOrmManager().getCollection(collectionName);
var document = collection.findOneAndUpdate(Filters.eq("_id", documentName)
, new Document("$inc", new Document(COUNT, 1L)));
if (document != null) {
return document.getLong("count") + 1;
}
Bson query = Filters.eq("_id", documentName);
Document inc = new Document("$inc", new Document(COUNT, 1L));
Document setOnInsert = new Document("$setOnInsert", new Document("_id", documentName));
// 报错后重试获取id，在大并发create这个key的时候mongodb总是会报错一次“duplicate key error!” 所以重试一次
// 报错后重试创建并获取id，在大并发create的时候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));
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);