support notes for Table Group

nilswende · Oct 24, 2024 · af3702c · af3702c
1 parent acf265e
commit af3702c
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 4 deletions.
diff --git a/src/main/java/com/wn/dbml/compiler/parser/ParserImpl.java b/src/main/java/com/wn/dbml/compiler/parser/ParserImpl.java
@@ -401,10 +401,15 @@ private void parseTableGroup() {
 		} else {
 			next(LBRACE);
 			while (true) {
-				var tableName = parseTableName();
-				var table = findTable(tableName);
-				if (!tableGroup.addTable(table)) {
-					error("Table '%s' is already defined", table);
+				if (lookaheadTypeIs(NOTE)) {
+					next(NOTE);
+					tableGroup.setNote(parseNote());
+				} else {
+					var tableName = parseTableName();
+					var table = findTable(tableName);
+					if (!tableGroup.addTable(table)) {
+						error("Table '%s' is already defined", table);
+					}
 				}
 				if (lookaheadTypeIs(RBRACE)) {
 					next(RBRACE);

diff --git a/src/main/java/com/wn/dbml/model/TableGroup.java b/src/main/java/com/wn/dbml/model/TableGroup.java
@@ -9,6 +9,7 @@ public class TableGroup {
 	private final Schema schema;
 	private final String name;
 	private final Set<Table> tables = new LinkedHashSet<>();
+	private Note note;
 
 	TableGroup(final Schema schema, final String name) {
 		this.schema = Objects.requireNonNull(schema);
@@ -31,6 +32,14 @@ public Set<Table> getTables() {
 		return Collections.unmodifiableSet(tables);
 	}
 
+	public Note getNote() {
+		return note;
+	}
+
+	public void setNote(Note note) {
+		this.note = note;
+	}
+
 	@Override
 	public boolean equals(final Object o) {
 		if (this == o) return true;

diff --git a/src/test/java/com/wn/dbml/compiler/ParserTest.java b/src/test/java/com/wn/dbml/compiler/ParserTest.java
@@ -373,6 +373,7 @@ void testParseTableGroup() {
 				  table1
 				  table2
 				  C
+				  note: 'group note'
 				}""";
 		var database = parse(dbml);
 
@@ -384,6 +385,9 @@ void testParseTableGroup() {
 		assertTrue(schema.containsTableGroup(tableGroupName));
 		var tables = tableGroup.getTables();
 		assertEquals(3, tables.size());
+		var note = tableGroup.getNote();
+		assertNotNull(note);
+		assertEquals("group note", note.getValue());
 	}
 
 	@Test