Fix bug of LazyResultSet

yuu-nkjm · Mar 3, 2021 · e2ee44a · e2ee44a
1 parent 00081ee
commit e2ee44a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main/java/org/nkjmlab/sorm4j/LazyResultSet.java b/src/main/java/org/nkjmlab/sorm4j/LazyResultSet.java
@@ -8,6 +8,7 @@
 import java.sql.ResultSet;
 import java.sql.SQLException;
 import java.util.Iterator;
+import java.util.LinkedHashMap;
 import java.util.List;
 import java.util.Map;
 import java.util.function.Supplier;
@@ -142,7 +143,7 @@ private final class LazyResultSetIterator<S> implements Iterator<S> {
     @SuppressWarnings("unchecked")
     public LazyResultSetIterator(AbstractOrmMapper orMapper, Class<S> objectClass,
         PreparedStatement stmt, ResultSet resultSet) {
-      this.getFunction = objectClass.equals(Map.class)
+      this.getFunction = objectClass.equals(LinkedHashMap.class)
           ? Try.createSupplierWithThrow(() -> (S) orMapper.toSingleMap(resultSet),
               OrmException::new)
           : Try.createSupplierWithThrow(() -> orMapper.toSingleObject(objectClass, resultSet),