#919 - io.ebean package initial

src/main/java/io/ebeaninternal/server/query/DefaultRelationalQueryEngine.java

@@ -0,0 +1,109 @@
+package io.ebeaninternal.server.query;
+
+import io.ebean.SqlRow;
+import io.ebeaninternal.server.core.Message;
+import io.ebeaninternal.server.core.RelationalQueryEngine;
+import io.ebeaninternal.server.core.RelationalQueryRequest;
+import io.ebeaninternal.server.persist.Binder;
+
+import javax.persistence.PersistenceException;
+import java.sql.SQLException;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.function.Consumer;
+import java.util.function.Predicate;
+
+/**
+ * Perform native sql fetches.
+ */
+public class DefaultRelationalQueryEngine implements RelationalQueryEngine {
+
+  private final Binder binder;
+
+  private final String dbTrueValue;
+
+  public DefaultRelationalQueryEngine(Binder binder, String dbTrueValue) {
+    this.binder = binder;
+    this.dbTrueValue = dbTrueValue == null ? "true" : dbTrueValue;
+  }
+
+  @Override
+  public void findEach(RelationalQueryRequest request, Predicate<SqlRow> consumer) {
+
+    long startTime = System.currentTimeMillis();
+    try {
+      request.executeSql(binder);
+      while (request.next()) {
+        if (!consumer.test(readRow(request))) {
+          break;
+        }
+      }
+      logSummary(request, startTime);
+
+    } catch (Exception e) {
+      throw new PersistenceException(Message.msg("fetch.error", e.getMessage(), request.getSql()), e);
+
+    } finally {
+      request.close();
+    }
+  }
+
+  @Override
+  public void findEach(RelationalQueryRequest request, Consumer<SqlRow> consumer) {
+
+    long startTime = System.currentTimeMillis();
+
+    try {
+      request.executeSql(binder);
+      while (request.next()) {
+        consumer.accept(readRow(request));
+      }
+      logSummary(request, startTime);
+
+    } catch (Exception e) {
+      throw new PersistenceException(Message.msg("fetch.error", e.getMessage(), request.getSql()), e);
+
+    } finally {
+      request.close();
+    }
+  }
+
+  public List<SqlRow> findList(RelationalQueryRequest request) {
+
+    long startTime = System.currentTimeMillis();
+    try {
+      request.executeSql(binder);
+
+      List<SqlRow> rows = new ArrayList<>();
+      while (request.next()) {
+        rows.add(readRow(request));
+      }
+
+      logSummary(request, startTime);
+
+      return rows;
+
+    } catch (Exception e) {
+      throw new PersistenceException(Message.msg("fetch.error", e.getMessage(), request.getSql()), e);
+
+    } finally {
+      request.close();
+    }
+  }
+
+  private void logSummary(RelationalQueryRequest request, long startTime) {
+
+    if (request.isLogSummary()) {
+      long exeTime = System.currentTimeMillis() - startTime;
+      request.getTransaction().logSummary("SqlQuery  rows[" + request.getRowCount() + "] time[" + exeTime + "] bind[" + request.getBindLog() + "]");
+    }
+  }
+
+  /**
+   * Read the row from the ResultSet and return as a MapBean.
+   */
+  private SqlRow readRow(RelationalQueryRequest request) throws SQLException {
+    return request.createNewRow(dbTrueValue);
+  }
+
+}