make get_table_metadata_location_error transparent

Author: Jan Kaul

catalogs/iceberg-s3tables-catalog/src/error.rs

@@ -4,6 +4,7 @@ use aws_sdk_s3tables::{
     operation::{
         create_namespace::CreateNamespaceError, create_table::CreateTableError,
         delete_table::DeleteTableError, get_table::GetTableError,
+        get_table_metadata_location::GetTableMetadataLocationError,
         list_namespaces::ListNamespacesError, list_tables::ListTablesError,
         update_table_metadata_location::UpdateTableMetadataLocationError,
     },
@@ -26,7 +27,11 @@ pub enum Error {
     #[error(transparent)]
     GetTable(#[from] SdkError<GetTableError, HttpResponse>),
     #[error(transparent)]
-    DeletaTable(#[from] SdkError<DeleteTableError, HttpResponse>),
+    DeleteTable(#[from] SdkError<DeleteTableError, HttpResponse>),
+    #[error(transparent)]
+    SdkError(#[from] SdkError<GetTableMetadataLocationError, HttpResponse>),
+    #[error(transparent)]
+    GetTableMetadataLocation(#[from] GetTableMetadataLocationError),
     #[error(transparent)]
     CreateTable(#[from] SdkError<CreateTableError, HttpResponse>),
     #[error(transparent)]

catalogs/iceberg-s3tables-catalog/src/lib.rs

@@ -6,7 +6,10 @@ use std::{
 use async_trait::async_trait;
 use aws_config::SdkConfig;
 
-use aws_sdk_s3tables::{types::OpenTableFormat, Client};
+use aws_sdk_s3tables::{
+    error::SdkError, operation::get_table_metadata_location::GetTableMetadataLocationError,
+    types::OpenTableFormat, Client,
+};
 use iceberg_rust::{
     catalog::{
         commit::{
@@ -236,7 +239,15 @@ impl Catalog for S3TablesCatalog {
             .name(identifier.name())
             .send()
             .await
-            .map_err(|_| IcebergError::CatalogNotFound)?;
+            .map_err(|err| match err {
+                SdkError::ServiceError(err) => match err.into_err() {
+                    GetTableMetadataLocationError::NotFoundException(_) => {
+                        IcebergError::CatalogNotFound
+                    }
+                    x => Error::from(x).into(),
+                },
+                x => Error::from(x).into(),
+            })?;
 
         let metadata_location = table.metadata_location.ok_or(Error::Text(format!(
             "Table {} not found.",