Check newline

Matts966 · Apr 24, 2022 · b7ba232 · b7ba232
1 parent 85187ee
commit b7ba232
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/zetasql/parser/unparser.cc b/zetasql/parser/unparser.cc
@@ -55,10 +55,16 @@ std::string UnparseWithComments(const ASTNode* node, std::deque<std::pair<std::s
   // Print comments by visitors and pop.
   node->Accept(&unparser, &parse_tokens);
   // Emit left comments in parse_tokens.
+  bool newline_emitted = false;
   for (const auto& parse_token : parse_tokens) {
-    unparser.println(parse_token.first);
+    if (parse_token.first.find("\n") != std::string::npos) {
+      newline_emitted = true;
+    }
+    unparser.print(parse_token.first);
+  }
+  if (!newline_emitted) {
+    unparser.FlushLine();
   }
-  unparser.FlushLine();
   return unparsed_;
 }