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 info = loadEntityInfo(clazz); - final ArrayList ids = new ArrayList<>(); - keyStream.forEach(k -> ids.add(k)); final Attribute primary = info.getPrimary(); - List rs = queryList(clazz, FilterNodes.in(primary.field(), ids)); + List rs = queryList(clazz, FilterNodes.in(primary.field(), keyStream)); Map 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 info = loadEntityInfo(clazz); - final ArrayList pks = new ArrayList<>(); - keyStream.forEach(k -> pks.add(k)); final Attribute primary = info.getPrimary(); - return queryListAsync(clazz, FilterNodes.in(primary.field(), pks)).thenApply((List rs) -> { + return queryListAsync(clazz, FilterNodes.in(primary.field(), keyStream)).thenApply((List rs) -> { Map map = new LinkedHashMap<>(); if (rs.isEmpty()) { return new LinkedHashMap<>();