从S3存储桶下载300万个对象的最快方法

时间:2011-01-18 05:20:24

标签: python linux amazon-s3 boto eventlet

我尝试过使用Python + boto +多处理,S3cmd和J3tset但是他们都在苦苦挣扎。

任何建议,也许是您一直在使用的现成脚本或其他我不知道的方式?

修改

如下所述,

eventlet + boto是一个有价值的解决方案。在这里http://web.archive.org/web/20110520140439/http://teddziuba.com/2010/02/eventlet-asynchronous-io-for-g.html

找到了一篇很好的eventlet参考文章

我已经添加了我正在使用的python脚本。

2 个答案:

答案 0 :(得分:33)

好的,我找到了一个基于@Matt Billenstien暗示的解决方案。它使用eventlet库。第一步是最重要的(猴子修补标准IO库)。

使用nohup在后台运行此脚本,您已完成设置。

from eventlet import *
patcher.monkey_patch(all=True)

import os, sys, time
from boto.s3.connection import S3Connection
from boto.s3.bucket import Bucket

import logging

logging.basicConfig(filename="s3_download.log", level=logging.INFO)


def download_file(key_name):
    # Its imp to download the key from a new connection
    conn = S3Connection("KEY", "SECRET")
    bucket = Bucket(connection=conn, name="BUCKET")
    key = bucket.get_key(key_name)

    try:
        res = key.get_contents_to_filename(key.name)
    except:
        logging.info(key.name+":"+"FAILED")

if __name__ == "__main__":
    conn = S3Connection("KEY", "SECRET")
    bucket = Bucket(connection=conn, name="BUCKET")

    logging.info("Fetching bucket list")
    bucket_list = bucket.list(prefix="PREFIX")

    logging.info("Creating a pool")
    pool = GreenPool(size=20)

    logging.info("Saving files in bucket...")
    for key in bucket.list():
        pool.spawn_n(download_file, key.key)
    pool.waitall()

答案 1 :(得分:5)

使用eventlet为您提供I / O并行性,编写一个简单的函数来使用urllib下载一个对象,然后使用GreenPile将其映射到输入URL列表 - 一堆50到100个greenlets应该做..