Added RETURN fields in search queries

Author: Julien Ruaux

subprojects/trino-redisearch/src/main/java/com/redis/trino/RediSearchPageSource.java

@@ -37,7 +37,7 @@ public RediSearchPageSource(RediSearchSession rediSearchSession, RediSearchTable
 			List<RediSearchColumnHandle> columns) {
 		this.columnNames = columns.stream().map(RediSearchColumnHandle::getName).collect(toList());
 		this.columnTypes = columns.stream().map(RediSearchColumnHandle::getType).collect(toList());
-		this.cursor = rediSearchSession.search(tableHandle).iterator();
+		this.cursor = rediSearchSession.search(tableHandle, columns).iterator();
 		this.currentDoc = null;
 		this.pageBuilder = new PageBuilder(columnTypes);
 	}

subprojects/trino-redisearch/src/main/java/com/redis/trino/RediSearchSession.java

@@ -198,11 +198,13 @@ private Type columnType(TypeSignature typeSignature) {
 		return typeManager.fromSqlType(typeSignature.toString());
 	}
 
-	public SearchResults<String, String> search(RediSearchTableHandle tableHandle) {
+	public SearchResults<String, String> search(RediSearchTableHandle tableHandle,
+			List<RediSearchColumnHandle> columns) {
 		String index = index(tableHandle);
 		String query = RediSearchQueryBuilder.buildQuery(tableHandle.getConstraint());
 		Builder<String, String> options = SearchOptions.builder();
 		options.limit(Limit.offset(0).num(limit(tableHandle)));
+		options.returnFields(columns.stream().map(RediSearchColumnHandle::getName).toArray(String[]::new));
 		log.info("Running search on index %s with query '%s'", index, query);
 		return connection.sync().search(index, query, options.build());
 	}