Add tcp keepalive feature flag to botocore

VERSION

@@ -1 +1 @@
-0.0.9
+0.0.10

tests/.wickerconfig.test.json

@@ -5,7 +5,8 @@
     "boto_config": {
       "max_pool_connections":10,
       "read_timeout_s": 140,
-      "connect_timeout_s": 140
+      "connect_timeout_s": 140,
+      "tcp_keepalive": false
     }
   },
   "dynamodb_config": {

tests/test_s3_storage.py

@@ -147,7 +147,7 @@ def test_get_column_concatenated_bytes_files_path(self, mock_get_config: mock.Mo
             "aws_s3_config": {
                 "s3_datasets_path": "s3://dummy_bucket/wicker/",
                 "region": "us-east-1",
-                "boto_config": {"max_pool_connections": 10, "read_timeout_s": 140, "connect_timeout_s": 140},
+                "boto_config": {"max_pool_connections": 10, "read_timeout_s": 140, "connect_timeout_s": 140, "tcp_keepalive": False},
             },
             "dynamodb_config": {"table_name": "fake-table-name", "region": "us-west-2"},
             "storage_download_config": {

wicker/core/config.py

@@ -28,13 +28,15 @@ class BotoS3Config:
     max_pool_connections: int
     read_timeout_s: int
     connect_timeout_s: int
+    tcp_keepalive: bool
 
     @classmethod
     def from_json(cls, data: Dict[str, Any]) -> BotoS3Config:
         return cls(
             max_pool_connections=data["max_pool_connections"],
             read_timeout_s=data["read_timeout_s"],
             connect_timeout_s=data["connect_timeout_s"],
+            tcp_keepalive=data["tcp_keepalive"]
         )
 
 

wicker/core/storage.py

@@ -42,6 +42,7 @@ def __init__(self, session: Optional[boto3.session.Session] = None) -> None:
             max_pool_connections=boto_config.max_pool_connections,
             read_timeout=boto_config.read_timeout_s,
             connect_timeout=boto_config.connect_timeout_s,
+            tcp_keepalive= boto_config.tcp_keepalive
         )
         self.session = boto3.session.Session() if session is None else session
         self.client = self.session.client("s3", config=boto_client_config)