tiennm99/zfoo
1
0
Fork 0
mirror of https://github.com/tiennm99/zfoo.git synced 2026-08-09 00:27:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #134 from awakeyoyoyo/main

Browse Source 
perf[orm]: code style
This commit is contained in:
jaysunxiao
2024-10-21 15:35:05 +08:00
committed by GitHub
parent 3aaab2dc21 ca0fc63380
commit 12116091c4
3 changed files with 23 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files
orm/src/main/java/com/zfoo/orm/accessor/IAccessor.java
-2
View File
@@ -34,8 +34,6 @@ public interface IAccessor {
<PK extends Comparable<PK>, E extends IEntity<PK>> void batchUpdate(List<E> entities);
<PK extends Comparable<PK>, E extends IEntity<PK>> void batchUpdateNode(List<PNode<PK,E>> entities);
<PK extends Comparable<PK>, E extends IEntity<PK>> boolean delete(E entity);
<PK extends Comparable<PK>, E extends IEntity<PK>> boolean delete(PK pk, Class<E> entityClazz);
orm/src/main/java/com/zfoo/orm/accessor/MongodbAccessor.java
-31
View File
@@ -104,37 +104,6 @@ public class MongodbAccessor implements IAccessor {
}
}
@Override
public <PK extends Comparable<PK>, E extends IEntity<PK>> void batchUpdateNode(List<PNode<PK,E>> nodes) {
if (CollectionUtils.isEmpty(nodes)) {
return;
}
try {
@SuppressWarnings("unchecked")
var entityClazz = (Class<E>) nodes.get(0).getClass();
var collection = OrmContext.getOrmManager().getCollection(entityClazz);
List<E> entities = nodes.stream().map(PNode::getEntity).toList();
var batchList = entities.stream()
.map(it -> new ReplaceOneModel<E>(Filters.eq("_id", it.id()), it))
.toList();
var result = collection.bulkWrite(batchList, new BulkWriteOptions().ordered(false));
//设置修改时间
long currentTime = TimeUtils.currentTimeMillis();
nodes.forEach(k->k.resetTime(currentTime));
if (result.getMatchedCount() != entities.size()) {
// 在数据库的批量更新操作中需要更新的数量和最终更新的数量不相同
logger.warn("database:[{}] update size:[{}] not equal with matched size:[{}](some entity of id not exist in database)"
, entityClazz.getSimpleName(), entities.size(), result.getMatchedCount());
}
} catch (Throwable t) {
logger.error("batchUpdate unknown exception", t);
}
}
@Override
public <PK extends Comparable<PK>, E extends IEntity<PK>> boolean delete(E entity) {
@SuppressWarnings("unchecked")
orm/src/main/java/com/zfoo/orm/cache/EntityCache.java
Vendored
+23 -1
View File
@@ -309,7 +309,29 @@ public class EntityCache<PK extends Comparable<PK>, E extends IEntity<PK>> imple
for (var currentPage = 1; currentPage <= maxPageSize; currentPage++) {
page.setPage(currentPage);
var currentUpdateList = page.currentPageList(updateList);
OrmContext.getAccessor().batchUpdateNode(currentUpdateList);
try {
@SuppressWarnings("unchecked")
var entityClazz = (Class<E>) currentUpdateList.get(0).getClass();
var collection = OrmContext.getOrmManager().getCollection(entityClazz);
List<E> entities = currentUpdateList.stream().map(PNode::getEntity).toList();
var batchList = entities.stream()
.map(it -> new ReplaceOneModel<E>(Filters.eq("_id", it.id()), it))
.toList();
var result = collection.bulkWrite(batchList, new BulkWriteOptions().ordered(false));
//设置修改时间
long currentTime = TimeUtils.currentTimeMillis();
currentUpdateList.forEach(k->k.resetTime(currentTime));
if (result.getMatchedCount() != entities.size()) {
// 在数据库的批量更新操作中需要更新的数量和最终更新的数量不相同
logger.warn("database:[{}] update size:[{}] not equal with matched size:[{}](some entity of id not exist in database)"
, entityClazz.getSimpleName(), entities.size(), result.getMatchedCount());
}
} catch (Throwable t) {
logger.error("batchUpdate unknown exception", t);
}
}
}