Merge pull request #15 from Matts966/feature/better-comment-placement

Better placement of comments
Matts966 · Apr 18, 2022 · 5988095 · 5988095
2 parents e42a474 + c25e070
commit 5988095
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/zetasql/parser/unparser.cc b/zetasql/parser/unparser.cc
@@ -1373,6 +1373,8 @@ void Unparser::visitASTQuery(const ASTQuery* node, void* data) {
     visitASTChildren(node, data);
   }
   PrintCloseParenIfNeeded(node);
+  // ASTQuery always ends parent.
+  formatter_.FlushCommentsPassedBy(node->parent()->GetParseLocationRange().end(), data);
 }
 
 void Unparser::visitASTSetOperation(const ASTSetOperation* node, void* data) {

diff --git a/zetasql/parser/unparser.h b/zetasql/parser/unparser.h
@@ -130,7 +130,9 @@ class Unparser : public ParseTreeVisitor {
   }
 
   void visitASTChildren(const ASTNode* node, void* data) {
+    formatter_.FlushCommentsPassedBy(node->GetParseLocationRange().start(), data);
     node->ChildrenAccept(this, data);
+    formatter_.FlushCommentsPassedBy(node->GetParseLocationRange().end(), data);
   }
 
   void visit(const ASTNode* node, void* data) override {