You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
76 lines
1.6 KiB
76 lines
1.6 KiB
package pgdriver
|
|
|
|
import (
|
|
"database/sql/driver"
|
|
"fmt"
|
|
"net"
|
|
)
|
|
|
|
// Error represents an error returned by PostgreSQL server
|
|
// using PostgreSQL ErrorResponse protocol.
|
|
//
|
|
// https://www.postgresql.org/docs/current/static/protocol-message-formats.html
|
|
type Error struct {
|
|
m map[byte]string
|
|
}
|
|
|
|
// Field returns a string value associated with an error field.
|
|
//
|
|
// https://www.postgresql.org/docs/current/static/protocol-error-fields.html
|
|
func (err Error) Field(k byte) string {
|
|
return err.m[k]
|
|
}
|
|
|
|
// IntegrityViolation reports whether the error is a part of
|
|
// Integrity Constraint Violation class of errors.
|
|
//
|
|
// https://www.postgresql.org/docs/current/static/errcodes-appendix.html
|
|
func (err Error) IntegrityViolation() bool {
|
|
switch err.Field('C') {
|
|
case "23000", "23001", "23502", "23503", "23505", "23514", "23P01":
|
|
return true
|
|
default:
|
|
return false
|
|
}
|
|
}
|
|
|
|
// StatementTimeout reports whether the error is a statement timeout error.
|
|
func (err Error) StatementTimeout() bool {
|
|
return err.Field('C') == "57014"
|
|
}
|
|
|
|
func (err Error) Error() string {
|
|
return fmt.Sprintf("%s: %s (SQLSTATE=%s)",
|
|
err.Field('S'), err.Field('M'), err.Field('C'))
|
|
}
|
|
|
|
func isBadConn(err error, allowTimeout bool) bool {
|
|
switch err {
|
|
case nil:
|
|
return false
|
|
case driver.ErrBadConn:
|
|
return true
|
|
}
|
|
|
|
if err, ok := err.(Error); ok {
|
|
switch err.Field('V') {
|
|
case "FATAL", "PANIC":
|
|
return true
|
|
}
|
|
switch err.Field('C') {
|
|
case "25P02", // current transaction is aborted
|
|
"57014": // canceling statement due to user request
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
if allowTimeout {
|
|
if err, ok := err.(net.Error); ok && err.Timeout() {
|
|
return !err.Temporary()
|
|
}
|
|
}
|
|
|
|
return true
|
|
}
|