From 386c066ce01e7b0c69dcc73ba8f509cb704bca75 Mon Sep 17 00:00:00 2001
From: Matheus Alcantara <mths.dev@pm.me>
Date: Fri, 16 Aug 2024 09:34:32 -0300
Subject: [PATCH] io: add support to set assume role session name

Partial fix for #527
---
 crates/iceberg/src/io/storage_s3.rs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/crates/iceberg/src/io/storage_s3.rs b/crates/iceberg/src/io/storage_s3.rs
index 7c5400c5f..115640131 100644
--- a/crates/iceberg/src/io/storage_s3.rs
+++ b/crates/iceberg/src/io/storage_s3.rs
@@ -52,6 +52,8 @@ pub const S3_SSE_MD5: &str = "s3.sse.md5";
 pub const S3_ASSUME_ROLE_ARN: &str = "client.assume-role.arn";
 /// Optional external ID used to assume an IAM role.
 pub const S3_ASSUME_ROLE_EXTERNAL_ID: &str = "client.assume-role.external-id";
+/// Optional session name used to assume an IAM role.
+pub const S3_ASSUME_ROLE_SESSION_NAME: &str = "client.assume-role.session-name";
 
 /// Parse iceberg props to s3 config.
 pub(crate) fn s3_config_parse(mut m: HashMap<String, String>) -> Result<S3Config> {
@@ -76,13 +78,15 @@ pub(crate) fn s3_config_parse(mut m: HashMap<String, String>) -> Result<S3Config
             cfg.enable_virtual_host_style = true;
         }
     };
-
     if let Some(arn) = m.remove(S3_ASSUME_ROLE_ARN) {
         cfg.role_arn = Some(arn);
     }
     if let Some(external_id) = m.remove(S3_ASSUME_ROLE_EXTERNAL_ID) {
         cfg.external_id = Some(external_id);
     };
+    if let Some(session_name) = m.remove(S3_ASSUME_ROLE_SESSION_NAME) {
+        cfg.role_session_name = Some(session_name);
+    };
     let s3_sse_key = m.remove(S3_SSE_KEY);
     if let Some(sse_type) = m.remove(S3_SSE_TYPE) {
         match sse_type.to_lowercase().as_str() {