Honor Wrap Flag When Formatting List of SELECT Fields

octoberswimmer · Dec 9, 2023 · 3a986a4 · 3a986a4
1 parent ee63be0
commit 3a986a4
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/formatter/visitors.go b/formatter/visitors.go
@@ -772,11 +772,15 @@ func (v *FormatVisitor) VisitFieldNameAlias(ctx *parser.FieldNameAliasContext) i
 }
 
 func (v *FormatVisitor) VisitSelectList(ctx *parser.SelectListContext) interface{} {
+	sep := ", "
+	if v.wrap {
+		sep = ",\n"
+	}
 	selectEntries := []string{}
 	for _, p := range ctx.AllSelectEntry() {
 		selectEntries = append(selectEntries, v.visitRule(p).(string))
 	}
-	return strings.Join(selectEntries, ",\n")
+	return strings.Join(selectEntries, sep)
 }
 
 func (v *FormatVisitor) VisitSubFieldList(ctx *parser.SubFieldListContext) interface{} {