perf[orm]: avoid resize

zfoo-project · Aug 3, 2024 · a844013 · a844013
1 parent a455317
commit a844013
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/orm/src/main/java/com/zfoo/orm/cache/EntityCache.java b/orm/src/main/java/com/zfoo/orm/cache/EntityCache.java
@@ -222,9 +222,7 @@ public void persist(PK pk) {
             return;
         }
         pnode.resetTime(TimeUtils.currentTimeMillis());
-        var updateList = new ArrayList<E>();
-        updateList.add(pnode.getEntity());
-        doPersist(updateList);
+        doPersist(List.of(pnode.getEntity()));
     }
 
 
@@ -239,13 +237,14 @@ public void persistAll() {
             var currentTime = TimeUtils.currentTimeMillis();
             // key为threadId
             var updateMap = new HashMap<Long, List<E>>();
+            var initSize = cache.size() >> 2;
             cache.forEach(new BiConsumer<PK, PNode<PK, E>>() {
                 @Override
                 public void accept(PK pk, PNode<PK, E> pnode) {
                     var entity = pnode.getEntity();
                     if (pnode.getModifiedTime() != pnode.getWriteToDbTime()) {
                         pnode.resetTime(currentTime);
-                        var updateList = updateMap.computeIfAbsent(pnode.getThreadId(), it -> new ArrayList<>());
+                        var updateList = updateMap.computeIfAbsent(pnode.getThreadId(), it -> new ArrayList<>(initSize));
                         updateList.add(entity);
                     }
                 }
@@ -268,7 +267,7 @@ public void accept(PK pk, PNode<PK, E> pnode) {
     @Override
     public void persistAllBlock() {
         var currentTime = TimeUtils.currentTimeMillis();
-        var updateList = new ArrayList<E>();
+        var updateList = new ArrayList<E>(cache.size());
         cache.forEach(new BiConsumer<PK, PNode<PK, E>>() {
             @Override
             public void accept(PK pk, PNode<PK, E> pnode) {