From a7f1ef2c3b1ba6cdb4544447478f33ea79446bd4 Mon Sep 17 00:00:00 2001
From: redkale <redkale@qq.com>
Date: Sat, 6 Jan 2024 19:58:16 +0800
Subject: [PATCH] AbstractDataSqlSource

---
 .../java/org/redkale/source/AbstractDataSqlSource.java    | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/src/main/java/org/redkale/source/AbstractDataSqlSource.java b/src/main/java/org/redkale/source/AbstractDataSqlSource.java
index c911db9d3..f385e3765 100644
--- a/src/main/java/org/redkale/source/AbstractDataSqlSource.java
+++ b/src/main/java/org/redkale/source/AbstractDataSqlSource.java
@@ -3075,10 +3075,8 @@ public abstract class AbstractDataSqlSource extends AbstractDataSource implement
             return new LinkedHashMap<>();
         }
         final EntityInfo<T> info = loadEntityInfo(clazz);
-        final ArrayList<K> ids = new ArrayList<>();
-        keyStream.forEach(k -> ids.add(k));
         final Attribute<T, Serializable> primary = info.getPrimary();
-        List<T> rs = queryList(clazz, FilterNodes.in(primary.field(), ids));
+        List<T> rs = queryList(clazz, FilterNodes.in(primary.field(), keyStream));
         Map<K, T> map = new LinkedHashMap<>();
         if (rs.isEmpty()) {
             return new LinkedHashMap<>();
@@ -3095,10 +3093,8 @@ public abstract class AbstractDataSqlSource extends AbstractDataSource implement
             return CompletableFuture.completedFuture(new LinkedHashMap<>());
         }
         final EntityInfo<T> info = loadEntityInfo(clazz);
-        final ArrayList<K> pks = new ArrayList<>();
-        keyStream.forEach(k -> pks.add(k));
         final Attribute<T, Serializable> primary = info.getPrimary();
-        return queryListAsync(clazz, FilterNodes.in(primary.field(), pks)).thenApply((List<T> rs) -> {
+        return queryListAsync(clazz, FilterNodes.in(primary.field(), keyStream)).thenApply((List<T> rs) -> {
             Map<K, T> map = new LinkedHashMap<>();
             if (rs.isEmpty()) {
                 return new LinkedHashMap<>();