diff --git a/lib/build.gradle b/lib/build.gradle index fb72220..48372ea 100644 --- a/lib/build.gradle +++ b/lib/build.gradle @@ -38,7 +38,7 @@ dependencies { implementation "io.grpc:grpc-stub:1.57.1" implementation "io.grpc:grpc-services:1.57.1" implementation "io.grpc:grpc-testing:1.57.1" - implementation "io.cloudquery:plugin-pb-java:0.0.5" + implementation "io.cloudquery:plugin-pb-java:0.0.6" implementation "org.apache.arrow:arrow-memory-core:12.0.1" implementation "org.apache.arrow:arrow-vector:12.0.1" diff --git a/lib/src/main/java/io/cloudquery/internal/servers/plugin/v3/PluginServer.java b/lib/src/main/java/io/cloudquery/internal/servers/plugin/v3/PluginServer.java index 126c3c0..8b665a1 100644 --- a/lib/src/main/java/io/cloudquery/internal/servers/plugin/v3/PluginServer.java +++ b/lib/src/main/java/io/cloudquery/internal/servers/plugin/v3/PluginServer.java @@ -57,7 +57,11 @@ public void getTables( io.cloudquery.plugin.v3.GetTables.Request request, StreamObserver responseObserver) { try { - List tables = plugin.tables(); + List
tables = + plugin.tables( + request.getTablesList(), + request.getSkipTablesList(), + request.getSkipDependentTables()); List byteStrings = new ArrayList<>(); for (Table table : tables) { try (BufferAllocator bufferAllocator = new RootAllocator()) { diff --git a/lib/src/main/java/io/cloudquery/memdb/MemDB.java b/lib/src/main/java/io/cloudquery/memdb/MemDB.java index f524fbd..be3d2fd 100644 --- a/lib/src/main/java/io/cloudquery/memdb/MemDB.java +++ b/lib/src/main/java/io/cloudquery/memdb/MemDB.java @@ -29,8 +29,10 @@ public void init() { } @Override - public List
tables() throws SchemaException { - return Table.filterDFS(allTables, List.of("*"), List.of(), false); + public List
tables( + List includeList, List skipList, boolean skipDependentTables) + throws SchemaException { + return Table.filterDFS(allTables, includeList, skipList, skipDependentTables); } @Override diff --git a/lib/src/main/java/io/cloudquery/plugin/Plugin.java b/lib/src/main/java/io/cloudquery/plugin/Plugin.java index 0e22eee..9f5cc1f 100644 --- a/lib/src/main/java/io/cloudquery/plugin/Plugin.java +++ b/lib/src/main/java/io/cloudquery/plugin/Plugin.java @@ -18,7 +18,9 @@ public abstract class Plugin { public abstract void init(); - public abstract List
tables() throws SchemaException; + public abstract List
tables( + List includeList, List skipList, boolean skipDependentTables) + throws SchemaException; public abstract void sync();