Nodejs爬虫进阶=>异步并发控制

之前写了个现在看来很不完美的小爬虫，很多地方没有处理好，比如说在知乎点开一个问题的时候，它的所有回答并不是全部加载好了的，当你拉到回答的尾部时，点击加载更多，回答才会再加载一部分，所以说如果直接发送一个问题的请求链接，取得的页面是不完整的。还有就是我们通过访问链接下载图片的时候，是一张一张来下的，如果图片数量太多的话，真的是会下到你睡完觉它还在下。

这次的的爬虫是上次那个的升级版，爬虫代码在我的github上可以找到=>NodeSpider。

整个爬虫的思路是这样的：在一开始我们通过请求问题的链接抓取到部分页面数据，接下来我们在代码中模拟ajax请求截取剩余页面的数据，当然在这里也是可以通过异步来实现并发的，对于小规模的异步流程控制，可以用这个模块=>eventproxy，但这里我就没有用啦！我们通过分析获取到的页面从中截取出所有图片的链接，再通过异步并发来实现对这些图片的批量下载。

抓取页面初始的数据很简单啊，这里就不做多解释了

/*获取首屏所有图片链接*/
var getInitUrlList=function(){
    request.get("https://www.zhihu.com/question/34937418")
            .end(function(err,res){
                if(err){
                    console.log(err);
                }else{
                    var $=cheerio.load(res.text);
                    var answerList=$(".zm-item-answer");
                    answerList.map(function(i,answer){
                        var images=$(answer).find('.zm-item-rich-text img');
                        images.map(function(i,image){
                            photos.push($(image).attr("src"));
                        });
                    });
                    console.log("已成功抓取"+photos.length+"张图片的链接");
                    getIAjaxUrlList(20);
                }
            });
}

模拟ajax请求获取完整页面

接下来就是怎么去模拟点击加载更多时发出的ajax请求了，去知乎看一下吧！

 

有了这些信息，就可以来模拟发送相同的请求来获得这些数据啦。

/*每隔100毫秒模拟发送ajax请求，并获取请求结果中所有的图片链接*/
var getIAjaxUrlList=function(offset){
    request.post("https://www.zhihu.com/node/QuestionAnswerListV2")
            .set(config)
                .send("method=next&params=%7B%22url_token%22%3A34937418%2C%22pagesize%22%3A20%2C%22offset%22%3A" +offset+ "%7D&_xsrf=98360a2df02783902146dee374772e51")
                    .end(function(err,res){
                        if(err){
                            console.log(err);
                        }else{
                            var response=JSON.parse(res.text);/*想用json的话对json序列化即可，提交json的话需要对json进行反序列化*/
                            if(response.msg&&response.msg.length){
                                var $=cheerio.load(response.msg.join(""));/*把所有的数组元素拼接在一起，以空白符分隔，不要这样join()，它会默认数组元素以逗号分隔*/
                                var answerList=$(".zm-item-answer");
                                answerList.map(function(i,answer){
                                    var images=$(answer).find('.zm-item-rich-text img');
                                    images.map(function(i,image){
                                        photos.push($(image).attr("src"));
                                    });
                                });
                                setTimeout(function(){
                                    offset+=20;
                                    console.log("已成功抓取"+photos.length+"张图片的链接");
                                    getIAjaxUrlList(offset);
                                },100);
                            }else{
                                console.log("图片链接全部获取完毕，一共有"+photos.length+"条图片链接");
                                // console.log(photos);
                                return downloadImg(50);
                            }
                        }
                    });
}

在代码中post这条请求https://www.zhihu.com/node/QuestionAnswerListV2，把原请求头和请求参数复制下来，作为我们的请求头和请求参数，superagent的set方法可用来设置请求头，send方法可以用来发送请求参数。我们把请求参数中的offset初始为20，每隔一定时间offset再加20，再重新发送请求，这样就相当于我们每隔一定时间发送了一条ajax请求，获取到最新的20条数据，每获取到了数据，我们再对这些数据进行一定的处理，让它们变成一整段的html，便于后面的提取链接处理。

异步并发控制下载图片
再获取完了所有的图片链接之后，即判定response.msg为空时，我们就要对这些图片进行下载了，不可能一条一条下对不对，因为如你所看到的，我们的图片足足有

没错，2万多张，不过幸好nodejs拥有神奇的单线程异步特性，我们可以同时对这些图片进行下载。但这个时候问题来了，听说同时发送请求太多的话会被网站封ip的啊！所有我们肯定不能同时并发下载这两万多张图片，这个时候就需要对异步并发数量进行一些控制了。

在这里用到了一个神奇的模块=>async，它不仅能帮我们拜托难以维护的回调金字塔恶魔，还能轻松的帮我们进行异步流程的管理。具体看文档啦，这里就只用到了一个强大的async.mapLimit方法。

var requestAndwrite=function(url,callback){
    request.get(url).end(function(err,res){
        if(err){
            console.log(err);
            console.log("有一张图片请求失败啦...");
        }else{
            var fileName=path.basename(url);
            fs.writeFile("./img1/"+fileName,res.body,function(err){
                if(err){
                    console.log(err);
                    console.log("有一张图片写入失败啦...");
                }else{
                    console.log("图片下载成功啦");
                    callback(null,"successful !");
                    /*callback貌似必须调用，第二个参数将传给下一个回调函数的result，result是一个数组*/
                }
            });
        }
    });
}

var downloadImg=function(asyncNum){
    /*有一些图片链接地址不完整没有“http:”头部,帮它们拼接完整*/
    for(var i=0;i<photos.length;i++){
        if(photos[i].indexOf("http")===-1){
            photos[i]="http:"+photos[i];
        }
    }
    console.log("即将异步并发下载图片，当前并发数为:"+asyncNum);
    async.mapLimit(photos,asyncNum,function(photo,callback){
        console.log("已有"+asyncNum+"张图片进入下载队列");
        requestAndwrite(photo,callback);
    },function(err,result){
        if(err){
            console.log(err);
        }else{
            // console.log(result);<=会输出一个有2万多个“successful”字符串的数组
            console.log("全部已下载完毕！");
        }
    });

};

先看这里=>

mapLimit方法的第一个参数photos是所有图片链接的数组，也是我们并发请求的对象，asyncNum是限制并发请求的数量。当我们有这个参数时，比如它的值是10，则它一次就只会帮我们从数组中取10条链接，执行并发的请求，这10条请求都得到响应后，再发送下10条请求。

结尾

哦哦~，明天就是除夕了~