error.go

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
}