entc/gen/template: remove irrelevant code from templates (#481)

This commit is contained in:
Ariel Mashraki
2020-05-09 14:32:54 +03:00
committed by GitHub
parent 08289b3570
commit 7f260c3ae0
26 changed files with 5 additions and 980 deletions

View File

@@ -5,36 +5,4 @@ in the LICENSE file in the root directory of this source tree.
*/}}
{{/* custom globals and helpers for sql dialects */}}
{{ define "dialect/sql/globals" }}
// insertLastID invokes the insert query on the transaction and returns the LastInsertID.
func insertLastID(ctx context.Context, tx dialect.Tx, insert *sql.InsertBuilder) (int64, error) {
query, args := insert.Query()
// PostgreSQL does not support the LastInsertId() method of sql.Result
// on Exec, and should be extracted manually using the `RETURNING` clause.
if insert.Dialect() == dialect.Postgres {
rows := &sql.Rows{}
if err := tx.Query(ctx, query, args, rows); err != nil {
return 0, err
}
defer rows.Close()
if !rows.Next() {
return 0, fmt.Errorf("no rows found for query: %v", query)
}
var id int64
if err := rows.Scan(&id); err != nil {
return 0, err
}
return id, nil
}
// MySQL, SQLite, etc.
var res sql.Result
if err := tx.Exec(ctx, query, args, &res); err != nil {
return 0, err
}
id, err := res.LastInsertId()
if err != nil {
return 0, err
}
return id, nil
}
{{ end }}
{{ define "dialect/sql/globals" }}{{ end }}