Method formatWithClause

cmd/gosqlx/cmd/sql_formatter.go:548–594

formatWithClause formats WITH (CTE) clauses

(with *ast.WithClause)

// formatWithClause formats WITH (CTE) clauses
func (f *SQLFormatter) formatWithClause(with *ast.WithClause) error {
	f.writeKeyword("WITH")
	if with.Recursive {
		f.builder.WriteString(" ")
		f.writeKeyword("RECURSIVE")
	}
	f.builder.WriteString(" ")

	for i, cte := range with.CTEs {
		if i > 0 {
			f.builder.WriteString(", ")
		}
		f.builder.WriteString(cte.Name)

		if len(cte.Columns) > 0 {
			f.builder.WriteString(" (")
			for j, col := range cte.Columns {
				if j > 0 {
					f.builder.WriteString(", ")
				}
				f.builder.WriteString(col)
			}
			f.builder.WriteString(")")
		}

		f.builder.WriteString(" ")
		f.writeKeyword("AS")
		f.builder.WriteString(" (")

		if !f.compact {
			f.writeNewline()
			f.increaseIndent()
		}

		if err := f.formatStatement(cte.Statement); err != nil {
			return err
		}

		if !f.compact {
			f.decreaseIndent()
			f.writeNewline()
		}
		f.builder.WriteString(")")
	}

	return nil
}

// formatJoin formats JOIN clauses
func (f *SQLFormatter) formatJoin(join *ast.JoinClause) error {