python - 具有多个 request.session 的 grequests 池？

我想向 REST 网络服务发送大量 url 请求。通常在 75-90k 之间。但是，我需要限制与 Web 服务的并发连接数。

我开始以下列方式使用 grequests，但很快开始咀嚼打开的套接字。

concurrent_limit = 30
urllist = buildUrls()
hdrs = {'Host' : 'hostserver'}
g_requests = (grequests.get(url, headers=hdrs) for url in urls)
g_responses = grequests.map(g_requests, size=concurrent_limit)

由于这运行了一分钟左右，我遇到了“达到最大套接字数”错误。
据我所知，grequests 中的每个 requests.get 调用都使用它自己的 session ，这意味着为每个请求打开一个新套接字。

我在 github 上找到了一条说明如何使 grequest 使用单个 session 。但这似乎有效地将所有请求限制在一个共享池中。这似乎违背了异步 http 请求的目的。

s = requests.session()
rs = [grequests.get(url, session=s) for url in urls]
grequests.map(rs)

是否可以以创建多个 session 的方式使用 grequests 或 gevent.Pool？

换句话说:如何通过排队或连接池发出许多并发 http 请求？

最佳答案

我最终没有使用 grequests 来解决我的问题。我仍然希望这可能是可能的。

我使用线程:

class MyAwesomeThread(Thread):
    """
    Threading wrapper to handle counting and processing of tasks
    """
    def __init__(self, session, q):
        self.q = q
        self.count = 0
        self.session = session
        self.response = None
        Thread.__init__(self)

    def run(self):
        """TASK RUN BY THREADING"""
        while True:
            url, host = self.q.get()
            httpHeaders = {'Host' : host}
            self.response = session.get(url, headers=httpHeaders)
            # handle response here
            self.count+= 1
            self.q.task_done()
        return

q=Queue()
threads = []
for i in range(CONCURRENT):
    session = requests.session()
    t=MyAwesomeThread(session,q)
    t.daemon=True # allows us to send an interrupt
    threads.append(t)


## build urls and add them to the Queue
for url in buildurls():
    q.put_nowait((url,host))

## start the threads
for t in threads:
    t.start()

关于python - 具有多个 request.session 的 grequests 池？，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/19820901/