mirror of
https://github.com/ent/ent.git
synced 2026-05-24 09:31:56 +03:00
4323141fe2
Reviewed By: alexsn Differential Revision: D17149292 fbshipit-source-id: 837de5fad988de1e54438b47584701f2fc35326d
27 lines
967 B
Cheetah
27 lines
967 B
Cheetah
{{/*
|
|
Copyright 2019-present Facebook Inc. All rights reserved.
|
|
This source code is licensed under the Apache 2.0 license found
|
|
in the LICENSE file in the root directory of this source tree.
|
|
*/}}
|
|
|
|
{{/* custom errors and errors handlers from sql dialects */}}
|
|
{{ define "dialect/sql/errors" }}
|
|
func isSQLConstraintError(err error) (*ErrConstraintFailed, bool) {
|
|
// Error number 1062 is ER_DUP_ENTRY in mysql, and "UNIQUE constraint failed" is SQLite prefix.
|
|
if msg := err.Error(); strings.HasPrefix(msg, "Error 1062") || strings.HasPrefix(msg, "UNIQUE constraint failed") {
|
|
return &ErrConstraintFailed{msg, err}, true
|
|
}
|
|
return nil, false
|
|
}
|
|
|
|
// rollback calls to tx.Rollback and wraps the given error with the rollback error if occurred.
|
|
func rollback(tx dialect.Tx, err error) error {
|
|
if rerr := tx.Rollback(); rerr != nil {
|
|
err = fmt.Errorf("%s: %v", err.Error(), rerr)
|
|
}
|
|
if err, ok := isSQLConstraintError(err); ok {
|
|
return err
|
|
}
|
|
return err
|
|
}
|
|
{{ end }} |