From 267afd6bc91f58d10bc4284644e4a4d517cf45c5 Mon Sep 17 00:00:00 2001 From: erezrokah Date: Thu, 11 Apr 2024 13:17:47 +0100 Subject: [PATCH] chore: Call setParents on MemDB tables --- .../main/java/io/cloudquery/memdb/MemDB.java | 121 +++++++++--------- 1 file changed, 64 insertions(+), 57 deletions(-) diff --git a/lib/src/main/java/io/cloudquery/memdb/MemDB.java b/lib/src/main/java/io/cloudquery/memdb/MemDB.java index 7bd87df..799b8f8 100644 --- a/lib/src/main/java/io/cloudquery/memdb/MemDB.java +++ b/lib/src/main/java/io/cloudquery/memdb/MemDB.java @@ -23,64 +23,71 @@ public class MemDB extends Plugin { private static List getTables() { - return List.of( - Table.builder() - .name("table1") - .resolver( - new TableResolver() { - @Override - public void resolve( - ClientMeta clientMeta, Resource parent, TableOutputStream stream) { - stream.write( - Table1Data.builder() - .id(UUID.fromString("46b2b6e6-8f3e-4340-a721-4aa0786b1cc0")) - .name("name1") - .timestamp(LocalDateTime.now()) - .json(Map.of("key1", "value1", "key2", "value2")) - .build()); - stream.write( - Table1Data.builder() - .id(UUID.fromString("e89f95df-a389-4f1b-9ba6-1fab565523d6")) - .name("name2") - .build()); - } - }) - .transform(TransformWithClass.builder(Table1Data.class).pkField("id").build()) - .build(), - Table.builder() - .name("table2") - .resolver( - new TableResolver() { - @Override - public void resolve( - ClientMeta clientMeta, Resource parent, TableOutputStream stream) { - stream.write(Table2Data.builder().id(1).name("name1").build()); - stream.write(Table2Data.builder().id(2).name("name2").build()); - } - }) - .transform(TransformWithClass.builder(Table2Data.class).pkField("id").build()) - .relations( - List.of( - Table.builder() - .name("table2_child") - .resolver( - new TableResolver() { + List
tables = + List.of( + Table.builder() + .name("table1") + .resolver( + new TableResolver() { + @Override + public void resolve( + ClientMeta clientMeta, Resource parent, TableOutputStream stream) { + stream.write( + Table1Data.builder() + .id(UUID.fromString("46b2b6e6-8f3e-4340-a721-4aa0786b1cc0")) + .name("name1") + .timestamp(LocalDateTime.now()) + .json(Map.of("key1", "value1", "key2", "value2")) + .build()); + stream.write( + Table1Data.builder() + .id(UUID.fromString("e89f95df-a389-4f1b-9ba6-1fab565523d6")) + .name("name2") + .build()); + } + }) + .transform(TransformWithClass.builder(Table1Data.class).pkField("id").build()) + .build(), + Table.builder() + .name("table2") + .resolver( + new TableResolver() { + @Override + public void resolve( + ClientMeta clientMeta, Resource parent, TableOutputStream stream) { + stream.write(Table2Data.builder().id(1).name("name1").build()); + stream.write(Table2Data.builder().id(2).name("name2").build()); + } + }) + .transform(TransformWithClass.builder(Table2Data.class).pkField("id").build()) + .relations( + List.of( + Table.builder() + .name("table2_child") + .resolver( + new TableResolver() { - @Override - public void resolve( - ClientMeta clientMeta, - Resource parent, - TableOutputStream stream) { - String parentName = parent.get("name").toString(); - stream.write( - Table2ChildData.builder().name(parentName + "_name1").build()); - stream.write( - Table2ChildData.builder().name(parentName + "_name2").build()); - } - }) - .transform(TransformWithClass.builder(Table2ChildData.class).build()) - .build())) - .build()); + @Override + public void resolve( + ClientMeta clientMeta, + Resource parent, + TableOutputStream stream) { + String parentName = parent.get("name").toString(); + stream.write( + Table2ChildData.builder() + .name(parentName + "_name1") + .build()); + stream.write( + Table2ChildData.builder() + .name(parentName + "_name2") + .build()); + } + }) + .transform(TransformWithClass.builder(Table2ChildData.class).build()) + .build())) + .build()); + Tables.setParents(tables, null); + return tables; } private List
allTables;