Function renderUpdate

pkg/formatter/render.go:439–498

(u *ast.UpdateStatement, opts ast.FormatOptions)

}

func renderUpdate(u *ast.UpdateStatement, opts ast.FormatOptions) string {
	if u == nil {
		return ""
	}
	f := newNodeFormatter(opts)
	sb := f.sb

	if u.With != nil {
		sb.WriteString(renderWith(u.With, f))
		sb.WriteString(f.clauseSep())
	}

	sb.WriteString(f.kw("UPDATE"))
	sb.WriteString(" ")
	sb.WriteString(u.TableName)
	if u.Alias != "" {
		sb.WriteString(" ")
		sb.WriteString(u.Alias)
	}

	sb.WriteString(f.clauseSep())
	sb.WriteString(f.kw("SET"))
	sb.WriteString(" ")
	upds := make([]string, len(u.Assignments))
	for i, upd := range u.Assignments {
		upds[i] = exprSQL(upd.Column) + " = " + exprSQL(upd.Value)
	}
	sb.WriteString(strings.Join(upds, ", "))

	if len(u.From) > 0 {
		sb.WriteString(f.clauseSep())
		sb.WriteString(f.kw("FROM"))
		sb.WriteString(" ")
		froms := make([]string, len(u.From))
		for i := range u.From {
			froms[i] = tableRefSQL(&u.From[i], f)
		}
		sb.WriteString(strings.Join(froms, ", "))
	}

	if u.Where != nil {
		sb.WriteString(f.clauseSep())
		sb.WriteString(f.kw("WHERE"))
		sb.WriteString(" ")
		sb.WriteString(FormatExpression(u.Where, opts))
	}

	if len(u.Returning) > 0 {
		sb.WriteString(f.clauseSep())
		sb.WriteString(f.kw("RETURNING"))
		sb.WriteString(" ")
		sb.WriteString(exprListSQL(u.Returning))
	}

	if opts.AddSemicolon {
		sb.WriteString(";")
	}