Function AddReturning

pkg/transform/returning.go:51–62

AddReturning returns a Rule that appends one or more column names to the RETURNING clause of an INSERT, UPDATE, or DELETE statement. This is the standard PostgreSQL extension for returning row data from DML operations. SQL Server users can achieve a similar result with the OUTPUT clause (not yet cov

(columns ...string)

//
//	transform.Apply(stmt, transform.AddReturning("id", "created_at"))
func AddReturning(columns ...string) Rule {
	return RuleFunc(func(stmt ast.Statement) error {
		ret, err := getReturning(stmt)
		if err != nil {
			return err
		}
		for _, col := range columns {
			*ret = append(*ret, &ast.Identifier{Name: col})
		}
		return nil
	})
}

// RemoveReturning returns a Rule that clears the entire RETURNING clause from
// an INSERT, UPDATE, or DELETE statement. If the clause is already empty the