scrapy运行定量爬虫

假设需求

现在大概有三千个scrapy的爬虫文件，10台机器，如何合理的分配爬虫？什么，这么简单的数学题还要问，一台机器分300个爬虫不就行了。确实，这样分配最简单也最直接，但会带来一些问题。就比如，有些站点网页少而有些网站很大，每个爬虫运行的时间是不一样的，最后可能导致一台累死，九台围观。而且一台机器同时运行300个爬虫，在硬件资源上的消耗会很大，也可能会导致很多爬虫无法正常运行，所以即使是这样分配爬虫我们也要限制同时运行的爬虫数量。当某个爬虫运行完了，才执行下一个。

解决方法

可以先创建出一个队列，队列里存放待抓取的爬虫(通常会创建三个，分别是pending(待抓取)、running(正抓取)、finished(已抓取))。然后每台机器开始都取指定数量的爬虫运行，当其中一个运行完，在去任务队列里取，直到队列空了。

实现

这种数量不多的队列使用Redis的集合就行，我们创建pending、running、finished三个集合，然后将所有的爬虫的name字段存放在pending集合中(当然爬虫文件名也可以，只是启动爬虫的方式稍微不一样)，接着我们就可以写个脚本来运行爬虫了。
方式有两种：

一、crawl命令

伪代码：

pending.add(所有爬虫)
while True:
	if  len(running) < 指定数量:
		spider = pending.pop()
		多进程执行：f'scrapy crawl {spider}'
	else:
		time.sleep(指定时间)

然后只要编写一个extension来同步爬虫状态到Redis里

class SpiderCountLimit:

    def __init__(self, count):
        self.spider_count = count
        self.r = redis.Redis(decode_responses=True)

    @classmethod
    def from_crawler(cls, crawler):
        count = crawler.settings.get('SPIDER_COUNT', 20)
        ext = cls(count)
        crawler.signals.connect(ext.spider_closed, signal=signals.spider_closed)
        crawler.signals.connect(ext.spider_opened, signal=signals.spider_opened)
        return ext

    def spider_closed(self, spider, reason):
        self.r.srem('running', spider.name) # 爬虫关闭时，删除running中的爬虫
        self.r.sadd('finished', spider.name) # 加入到已完成队列

    def spider_opened(self, spider):
        self.r.sadd('running', spider.name) # 添加爬虫到running中

这种方法就不多说了，因为我没尝试，直接看第二种

二、Crawler API

伪代码：(不懂怎么用的Crawler API可以看scrapy自定义命令)

for i in range(指定数量):
	crawler_process.crawl(pending.pop())
crawler_process.start()
while True:
	if  len(running) < 指定数量:
		多进程执行：
			crawler_process.crawl(pending.pop())
			crawler_process.start()
	else:
		time.sleep(指定时间)

因为crawler_process.start()这个语句是阻塞的，所以需要多进程来执行。也可以去掉多进程，把extension的内容改成这样：

class SpiderCountLimit:

    def __init__(self, count):
        self.spider_count = count
        self.r = redis.Redis(decode_responses=True)

    @classmethod
    def from_crawler(cls, crawler):
        count = crawler.settings.get('SPIDER_COUNT', 20)
        ext = cls(count)
        crawler.signals.connect(ext.spider_closed, signal=signals.spider_closed)
        crawler.signals.connect(ext.spider_opened, signal=signals.spider_opened)
        return ext

    def spider_closed(self, spider, reason):
        self.r.srem('running', spider.name)
        self.r.sadd('finished', spider.name)

        spider = self.r.spop('pending')
        process = CrawlerProcess()
        process.crawl(spider)
        process.start() 
        
    def spider_opened(self, spider):
        self.r.sadd('running', spider.name)

不过我感觉这个方法不如多进程添加，因为上面提到process.start() 是阻塞的，也就是说spider_closed这个方法一直没有结束，这可能会带来一些无法预见的问题。

至于其他一些细节上的优化就自己思考了，比如改用进程池来管理进程等。