有多个请求会话的grequests池？

Question

我想对REST webserivce做很多url请求。通常在75-90k之间。但是，我需要限制到Web服务的并发连接数。

我开始用下面的方式玩灰球，但很快就开始咀嚼打开的球窝。

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中的每个request s.get调用都使用自己的会话，这意味着为每个请求打开一个新套接字。

我在github上找到了一个注释，其中提到了如何使grequest使用单个会话。但这似乎有效地将所有请求限制在一个共享池中。这似乎违背了异步http请求的目的。

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

是否可以使用grequests或gevent.Pool来创建多个会话？

换言之：如何通过排队或连接池发出多个并发http请求？

Answer 1

rs是一个异步请求列表。每个异步请求都有自己的会话。

rs = [grequests.get(url) for url in urls]
grequests.map(rs)
for ar in rs:
    print(ar.session.cookies)

Answer 2

像这样的：

NUM_SESSIONS = 50
sessions = [requests.Session() for i in range(NUM_SESSIONS)]
reqs = []
i = 0
for url in urls:
    reqs.append(grequests.get(url, session=sessions[i % NUM_SESSIONS]
    i+=1
responses = grequests.map(reqs, size=NUM_SESSIONS*5)

这将把请求分散到50个不同的会话中。

Answer 3

我最终没有用社交游戏来解决我的问题。我仍然希望这是可能的。

我用了线程：

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()