Add Chronos vectorbench case and helper method in JdbcExecutor

Author: yabsp

src/main/java/org/polypheny/simpleclient/executor/JdbcExecutor.java

@@ -108,6 +108,9 @@ public long executeQuery( Query query ) throws ExecutorException {
                         case ARRAY_REAL:
                             preparedStatement.setArray( entry.getKey(), connection.createArrayOf( "REAL", (Object[]) entry.getValue().right ) );
                             break;
+                        case ARRAY_BOOLEAN:
+                            preparedStatement.setArray( entry.getKey(), connection.createArrayOf( "BOOLEAN", (Object[]) entry.getValue().right ) );
+                            break;
                         case BYTE_ARRAY:
                             preparedStatement.setBytes( entry.getKey(), (byte[]) entry.getValue().right );
                             break;
@@ -156,6 +159,23 @@ public long executeQuery( Query query ) throws ExecutorException {
     }
 
 
+    /**
+     * Executes the given query and returns the values of the first column of the result set (e.g. the ids of a
+     * top-k nearest-neighbor query). Used for recall measurement where the actual returned rows are needed.
+     */
+    public List<Long> executeQueryAndGetIds( Query query ) throws ExecutorException {
+        List<Long> ids = new ArrayList<>();
+        try ( ResultSet resultSet = executeStatement.executeQuery( query.getSql() ) ) {
+            while ( resultSet.next() ) {
+                ids.add( resultSet.getLong( 1 ) );
+            }
+        } catch ( SQLException e ) {
+            throw new ExecutorException( e );
+        }
+        return ids;
+    }
+
+
     @Override
     public long executeQueryAndGetNumber( Query query ) throws ExecutorException {
         try {

src/main/java/org/polypheny/simpleclient/main/ChronosAgent.java

@@ -83,6 +83,9 @@
 import org.polypheny.simpleclient.scenario.graph.GraphBenchConfig;
 import org.polypheny.simpleclient.scenario.knnbench.KnnBench;
 import org.polypheny.simpleclient.scenario.knnbench.KnnBenchConfig;
+ import org.polypheny.simpleclient.scenario.vectorbench.PgVectorBench;
+import org.polypheny.simpleclient.scenario.vectorbench.VectorBench;
+import org.polypheny.simpleclient.scenario.vectorbench.VectorBenchConfig;
 import org.polypheny.simpleclient.scenario.multibench.MultiBench;
 import org.polypheny.simpleclient.scenario.multibench.MultiBenchConfig;
 import org.polypheny.simpleclient.scenario.multimedia.MultimediaBench;
@@ -253,6 +256,14 @@ protected Object prepare( ChronosJob chronosJob, final File inputDirectory, fina
                 config = new KnnBenchConfig( parsedConfig );
                 scenario = new KnnBench( executorFactory, (KnnBenchConfig) config, true, dumpQueryList );
                 break;
+            case "vectorBench":
+                config = new VectorBenchConfig( parsedConfig );
+                if ( config.system.equals( "postgres" ) ) {
+                    scenario = new PgVectorBench( executorFactory, (VectorBenchConfig) config, true, dumpQueryList );
+                } else {
+                    scenario = new VectorBench( executorFactory, (VectorBenchConfig) config, true, dumpQueryList );
+                }
+                break;
             case "multimedia":
                 config = new MultimediaConfig( parsedConfig );
                 scenario = new MultimediaBench( executorFactory, (MultimediaConfig) config, true, dumpQueryList );