queryRaw

Author: ZorTik

asql-core/src/main/java/me/zort/sqllib/SQLDatabaseConnection.java

@@ -1,5 +1,6 @@
 package me.zort.sqllib;
 
+import com.google.common.annotations.Beta;
 import lombok.Getter;
 import me.zort.sqllib.api.DefsVals;
 import me.zort.sqllib.api.ObjectMapper;
@@ -25,6 +26,7 @@
 
 import java.io.Closeable;
 import java.sql.Connection;
+import java.sql.ResultSet;
 import java.sql.SQLException;
 import java.util.ArrayList;
 import java.util.List;
@@ -141,6 +143,10 @@ public SQLDatabaseConnection(final @NotNull SQLConnectionFactory connectionFacto
 
   public abstract QueryRowsResult<Row> query(String query);
 
+  @Beta
+  @Nullable
+  public abstract ResultSet queryRaw(Query query) throws SQLException;
+
   /**
    * Executes given query and returns execution result.
    * This result does not contain any rows. If you want to

asql-core/src/main/java/me/zort/sqllib/SQLDatabaseConnectionImpl.java

@@ -1,5 +1,6 @@
 package me.zort.sqllib;
 
+import com.google.common.annotations.Beta;
 import com.google.gson.Gson;
 import lombok.Getter;
 import lombok.RequiredArgsConstructor;
@@ -32,6 +33,8 @@
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 
+import javax.sql.rowset.CachedRowSet;
+import javax.sql.rowset.RowSetProvider;
 import java.sql.*;
 import java.util.Objects;
 import java.util.Optional;
@@ -274,11 +277,14 @@ public QueryRowsResult<Row> query(final @NotNull String query) {
   @NotNull
   QueryRowsResult<Row> query(final @NotNull Query query, boolean isRetry) {
     Objects.requireNonNull(query);
-    if (!handleAutoReconnect())
+    if (!handleAutoReconnect()) {
       return new QueryRowsResult<>(false, "Cannot connect to database!");
+    }
 
     QueryResult cachedResult = cacheManager.get(query, false);
-    if (cachedResult instanceof QueryRowsResult) return (QueryRowsResult<Row>) cachedResult;
+    if (cachedResult instanceof QueryRowsResult) {
+      return (QueryRowsResult<Row>) cachedResult;
+    }
 
     try (PreparedStatement stmt = buildStatement(query);
          ResultSet resultSet = stmt.executeQuery()) {
@@ -310,6 +316,31 @@ QueryRowsResult<Row> query(final @NotNull Query query, boolean isRetry) {
     }
   }
 
+  /**
+   * Executes given query and returns raw ResultSet.
+   * Please note that this function is not recommended to be frequently used and is provided
+   * as is as feature.
+   *
+   * @param query Query to use for building query string.
+   * @return ResultSet object or null if there was an error in connection.
+   * @throws SQLException If there was an error while executing query.
+   */
+  @Beta
+  @Nullable
+  public ResultSet queryRaw(Query query) throws SQLException {
+    Objects.requireNonNull(query);
+    if (!handleAutoReconnect()) {
+      return null;
+    }
+    try (PreparedStatement stmt = buildStatement(query);
+         ResultSet resultSet = stmt.executeQuery()) {
+      // Create in-memory cached result set
+      CachedRowSet cachedResultSet = RowSetProvider.newFactory().createCachedRowSet();
+      cachedResultSet.populate(resultSet);
+      return cachedResultSet;
+    }
+  }
+
   /**
    * Executes given query and returns execution result.
    * This result does not contain any rows. If you want to

asql-core/src/main/java/me/zort/sqllib/internal/query/QueryNode.java

@@ -1,5 +1,6 @@
 package me.zort.sqllib.internal.query;
 
+import com.google.common.annotations.Beta;
 import lombok.Getter;
 import me.zort.sqllib.SQLDatabaseConnection;
 import me.zort.sqllib.SQLDatabaseConnectionImpl;
@@ -62,6 +63,7 @@ public QueryNode(@Nullable P parent, List<QueryNode<?>> initial, int priority) {
    * @param params
    * @return
    */
+  @Beta
   public static QueryNode<?> fromRawQuery(String query, Object... params) {
     return new QueryNode<>(null, Collections.emptyList(), QueryPriority.GENERAL) {
       @Override

src/test/java/me/zort/sqllib/test/TestCase1.java

@@ -3,6 +3,7 @@
 import lombok.AllArgsConstructor;
 import lombok.extern.log4j.Log4j2;
 import me.zort.sqllib.SQLConnectionBuilder;
+import me.zort.sqllib.api.Query;
 import me.zort.sqllib.internal.annotation.NullableField;
 import me.zort.sqllib.internal.annotation.PrimaryKey;
 import me.zort.sqllib.internal.query.QueryDetails;
@@ -23,6 +24,7 @@
 import org.junit.jupiter.api.condition.EnabledOnOs;
 import org.junit.jupiter.api.condition.OS;
 
+import java.sql.ResultSet;
 import java.sql.SQLException;
 import java.util.Optional;
 
@@ -200,15 +202,28 @@ public void test6_Transactions() {
 
   @Test
   public void test7_RawNode() {
-    String raw = "SELECT * FROM users WHERE nickname = ?";
+    String raw = "SELECT * FROM " + table +  " WHERE nickname = ?";
     QueryNode<?> query = QueryNode.fromRawQuery(raw, "User1");
     Pair<String, Object[]> preparedQuery = query.toPreparedQuery();
     assertEquals(raw, preparedQuery.getFirst());
     assertArrayEquals(new Object[]{"User1"}, preparedQuery.getSecond());
   }
 
   @Test
-  public void test8_Close() {
+  public void test8_RawQuery() {
+    Query query = QueryNode.fromRawQuery("SELECT * FROM " + table + " WHERE nickname = ?", "User1");
+      try {
+          ResultSet result = connection.queryRaw(query);
+          assertNotNull(result);
+          assertTrue(result.next());
+            assertEquals("User1", result.getString("nickname"));
+      } catch (SQLException e) {
+          fail(e);
+      }
+  }
+
+  @Test
+  public void test9_Close() {
     System.out.println("Closing connection...");
     connection.disconnect();
     System.out.println("Connection closed");