feat(sqlx-core): add table function to database error (#2616)

launchbadge · Jul 14, 2023 · 3db1675 · 3db1675
1 parent afb6b10
commit 3db1675
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/sqlx-core/src/error.rs b/sqlx-core/src/error.rs
@@ -212,6 +212,14 @@ pub trait DatabaseError: 'static + Send + Sync + StdError {
         None
     }
 
+    /// Returns the name of the table that was affected by the error, if applicable.
+    ///
+    /// ### Note
+    /// Currently only populated by the Postgres driver.
+    fn table(&self) -> Option<&str> {
+        None
+    }
+
     /// Returns the kind of the error, if supported.
     ///
     /// ### Note

diff --git a/sqlx-postgres/src/error.rs b/sqlx-postgres/src/error.rs
@@ -204,6 +204,10 @@ impl DatabaseError for PgDatabaseError {
         self.constraint()
     }
 
+    fn table(&self) -> Option<&str> {
+        self.table()
+    }
+
     fn kind(&self) -> ErrorKind {
         match self.code() {
             error_codes::UNIQUE_VIOLATION => ErrorKind::UniqueViolation,