-
-
Notifications
You must be signed in to change notification settings - Fork 401
/
error.go
70 lines (59 loc) · 1.74 KB
/
error.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
package pg
import (
"net"
"github.com/go-pg/pg/v10/internal"
)
// ErrNoRows is returned by QueryOne and ExecOne when query returned zero rows
// but at least one row is expected.
var ErrNoRows = internal.ErrNoRows
// ErrMultiRows is returned by QueryOne and ExecOne when query returned
// multiple rows but exactly one row is expected.
var ErrMultiRows = internal.ErrMultiRows
// Error represents an error returned by PostgreSQL server
// using PostgreSQL ErrorResponse protocol.
//
// https://www.postgresql.org/docs/10/static/protocol-message-formats.html
type Error interface {
error
// Field returns a string value associated with an error field.
//
// https://www.postgresql.org/docs/10/static/protocol-error-fields.html
Field(field byte) string
// IntegrityViolation reports whether an error is a part of
// Integrity Constraint Violation class of errors.
//
// https://www.postgresql.org/docs/10/static/errcodes-appendix.html
IntegrityViolation() bool
}
var _ Error = (*internal.PGError)(nil)
func isBadConn(err error, allowTimeout bool) (bool, string) {
if err == nil {
return false, ""
}
if _, ok := err.(internal.Error); ok {
return false, ""
}
if pgErr, ok := err.(Error); ok {
switch pgErr.Field('V') {
case "FATAL", "PANIC":
return true, ""
}
switch pgErr.Field('C') {
case "25P02": // current transaction is aborted
return true, "25P02"
case "57014": // canceling statement due to user request
return true, "57014"
}
return false, ""
}
if allowTimeout {
if netErr, ok := err.(net.Error); ok && netErr.Timeout() {
return !netErr.Temporary(), ""
}
}
return true, ""
}
//------------------------------------------------------------------------------
type timeoutError interface {
Timeout() bool
}