Method formatUpdate

cmd/gosqlx/cmd/sql_formatter.go:301–335

formatUpdate formats UPDATE statements

(stmt *ast.UpdateStatement)

// formatUpdate formats UPDATE statements
func (f *SQLFormatter) formatUpdate(stmt *ast.UpdateStatement) error {
	f.writeKeyword("UPDATE")
	f.builder.WriteString(" " + stmt.TableName)

	if stmt.Alias != "" {
		f.builder.WriteString(" " + stmt.Alias)
	}

	if len(stmt.Assignments) > 0 {
		f.writeNewline()
		f.writeKeyword("SET")
		f.builder.WriteString(" ")

		for i, update := range stmt.Assignments {
			update := update // G601: Create local copy to avoid memory aliasing
			if i > 0 {
				f.builder.WriteString(", ")
			}
			if err := f.formatUpdateExpression(&update); err != nil {
				return err
			}
		}
	}

	if stmt.Where != nil {
		f.writeNewline()
		f.writeKeyword("WHERE")
		f.builder.WriteString(" ")
		if err := f.formatExpression(stmt.Where); err != nil {
			return err
		}
	}

	return nil
}

// formatDelete formats DELETE statements
func (f *SQLFormatter) formatDelete(stmt *ast.DeleteStatement) error {