ErrNullValueNotAllowed and ErrRaisedException

Author: ungerik

errors.go

@@ -53,8 +53,18 @@ const (
 	// that are are only allowed within DB transactions
 	// when the DB connection is not a transaction.
 	ErrNotWithinTransaction sentinelError = "not within a transaction"
+
+	ErrNullValueNotAllowed sentinelError = "null value not allowed"
 )
 
+type ErrRaisedException struct {
+	Message string
+}
+
+func (e ErrRaisedException) Error() string {
+	return "raised exception: " + e.Message
+}
+
 type ErrIntegrityConstraintViolation struct {
 	Constraint string
 }

pqconn/errors.go

@@ -17,6 +17,8 @@ func WrapKnownErrors(err error) error {
 	var e *pq.Error
 	if errors.As(err, &e) {
 		switch e.Code {
+		case "22004":
+			return errors.Join(sqldb.ErrNullValueNotAllowed, err)
 		case "23000":
 			return errors.Join(sqldb.ErrIntegrityConstraintViolation{Constraint: e.Constraint}, err)
 		case "23001":
@@ -31,6 +33,8 @@ func WrapKnownErrors(err error) error {
 			return errors.Join(sqldb.ErrCheckViolation{Constraint: e.Constraint}, err)
 		case "23P01":
 			return errors.Join(sqldb.ErrExclusionViolation{Constraint: e.Constraint}, err)
+		case "P0001":
+			return errors.Join(sqldb.ErrRaisedException{Message: e.Message}, err)
 		}
 	}
 	return err