(iceberg-rust) branch main updated: support session token (#530)

[xuanwo]

This is an automated email from the ASF dual-hosted git repository.

xuanwo pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/iceberg-rust.git


The following commit(s) were added to refs/heads/main by this push:
     new bbbea97  support session token (#530)
bbbea97 is described below

commit bbbea9751439dea6afb85f5acf0f3689357cf3de
Author: Tobias Pütz <puetztob...@gmail.com>
AuthorDate: Mon Aug 12 21:06:57 2024 +0200

    support session token (#530)
---
 crates/iceberg/src/io/storage_s3.rs | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/crates/iceberg/src/io/storage_s3.rs 
b/crates/iceberg/src/io/storage_s3.rs
index 6e92b90..9abb321 100644
--- a/crates/iceberg/src/io/storage_s3.rs
+++ b/crates/iceberg/src/io/storage_s3.rs
@@ -30,6 +30,9 @@ pub const S3_ENDPOINT: &str = "s3.endpoint";
 pub const S3_ACCESS_KEY_ID: &str = "s3.access-key-id";
 /// S3 secret access key.
 pub const S3_SECRET_ACCESS_KEY: &str = "s3.secret-access-key";
+/// S3 session token.
+/// This is required when using temporary credentials.
+pub const S3_SESSION_TOKEN: &str = "s3.session-token";
 /// S3 region.
 pub const S3_REGION: &str = "s3.region";
 /// S3 Path Style Access.
@@ -56,6 +59,9 @@ pub(crate) fn s3_config_parse(mut m: HashMap<String, String>) 
-> Result<S3Config
     if let Some(secret_access_key) = m.remove(S3_SECRET_ACCESS_KEY) {
         cfg.secret_access_key = Some(secret_access_key);
     };
+    if let Some(session_token) = m.remove(S3_SESSION_TOKEN) {
+        cfg.session_token = Some(session_token);
+    };
     if let Some(region) = m.remove(S3_REGION) {
         cfg.region = Some(region);
     };