iframe-proxy | Sunbelt Computer Software

This repository was archived by the owner on Jul 16, 2024. It is now read-only.

Repository navigation

'''

Benchmark downloading files from S3

'''

import boto3

import json

import time

# create an S3 session

s3 = boto3.resource('s3')

s3_client = boto3.client('s3')

def lambda_handler(event, context):

start_time = time.time()

src_bucket = event['bucket']

src_keys = event['keys']

total_bytes = 0.0

# Download and process all keys

for key in src_keys:

response = s3_client.get_object(Bucket=src_bucket,Key=key)

total_bytes += response['ContentLength']

contents = response['Body'].read()

time_in_secs = (time.time() - start_time)

print "Time taken (s)", time_in_secs

print "Size (MB)", total_bytes / 1024/1024

return time_in_secs

'''

ev = {

"bucket": "smallya-useast-1",

"keys": ["pavlo.sample"],

}

lambda_handler(ev, {});

'''