From fe7873dbc68869b9b19b2aac7dd0ee91d54c2bf2 Mon Sep 17 00:00:00 2001
From: awake <lqh19980716@163.com>
Date: Mon, 28 Oct 2024 14:37:41 +0800
Subject: [PATCH] fix[orm]: fix error entity clazz

---
 orm/src/main/java/com/zfoo/orm/cache/EntityCache.java | 5 ++---
 1 file changed, 2 insertions(+), 3 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
index e04327ae..10e434cb 100644
--- a/orm/src/main/java/com/zfoo/orm/cache/EntityCache.java
+++ b/orm/src/main/java/com/zfoo/orm/cache/EntityCache.java
@@ -307,8 +307,7 @@ public class EntityCache<PK extends Comparable<PK>, E extends IEntity<PK>> imple
             var currentUpdateList = page.currentPageList(updateList);
             try {
                 @SuppressWarnings("unchecked")
-                var entityClazz = (Class<E>) currentUpdateList.get(0).getClass();
-                var collection = OrmContext.getOrmManager().getCollection(entityClazz);
+                var collection = OrmContext.getOrmManager().getCollection(clazz);
                 List<E> entities = currentUpdateList.stream().map(PNode::getEntity).toList();
                 var batchList = entities.stream()
                         .map(it -> new ReplaceOneModel<E>(Filters.eq("_id", it.id()), it))
@@ -323,7 +322,7 @@ public class EntityCache<PK extends Comparable<PK>, E extends IEntity<PK>> imple
                 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());
+                            , clazz.getSimpleName(), entities.size(), result.getMatchedCount());
                 }
             } catch (Throwable t) {
                 logger.error("batchUpdate unknown exception", t);