本項(xiàng)目爬蟲及服務(wù)端github地址:https://github.com/qiuchendi/cnbeta-node-master
前端github地址:https://github.com/qiuchendi/vue-cnbeta
技術(shù)細(xì)節(jié)
使用 async await 做異步邏輯的處理。
使用 async庫 來做循環(huán)遍歷,以及并發(fā)請(qǐng)求操作。
使用 log4js 來做日志處理
使用 cheerio 來做新聞詳情頁的分析抓取。
使用 mongoose 來連接mongoDB 做數(shù)據(jù)的保存以及操作。
目錄結(jié)構(gòu)
├── bin // 入口
│ ├── article-list.js // 抓取新聞列表邏輯
│ ├── content.js // 抓取新聞內(nèi)容邏輯
│ ├── server.js // 服務(wù)端程序入口
│ └── spider.js // 爬蟲程序入口
├── config // 配置文件
├── dbhelper // 數(shù)據(jù)庫操作方法目錄
├── middleware // koa2 中間件
├── model // mongoDB 集合操作實(shí)例
├── router // koa2 路由文件
├── utils // 工具函數(shù)
├── package.json
方案分析
首先看爬蟲程序入口文件,整體邏輯其實(shí)很簡(jiǎn)單,先抓取新聞列表,存入MongoDB數(shù)據(jù)庫,每十分鐘抓取一次。新聞列表抓取之后,在數(shù)據(jù)庫查詢列表中沒有新聞內(nèi)容的新聞,開始抓取新聞詳情,然后更新到數(shù)據(jù)庫。
const articleListInit = require('./article-list');
const articleContentInit = require('./content');
const logger = require('../config/log');
const start = async() => {
let articleListRes = await articleListInit();
if (!articleListRes) {
logger.warn('news list update failed...');
} else {
logger.info('news list update succeed!');
}
let articleContentRes = await articleContentInit();
if (!articleContentRes) {
logger.warn('article content grab error...');
} else {
logger.info('article content grab succeed!');
}
};
if (typeof articleListInit === 'function') {
start();
}
setInterval(start, 600000);
接著看抓取新聞列表的邏輯,因?yàn)榭梢垣@取到新聞列表的Ajax接口,所以直接調(diào)用接口獲取列表信息。但是也有個(gè)問題,cnBeta新聞列表的縮略圖以及文章里的的圖片是有防盜鏈的,所以你在自己的網(wǎng)站是沒法直接使用它的圖片的,所以我是直接把cnBeta的圖片文件爬下來存到自己的服務(wù)器上。
/**
* 初始化方法 抓取文章列表
* @returns {Promise.<*>}
*/
const articleListInit = async() => {
logger.info('grabbing article list starts...');
const pageUrlList = getPageUrlList(listBaseUrl, totalPage);
if (!pageUrlList) {
return;
}
let res = await getArticleList(pageUrlList);
return res;
}
/**
* 利用分頁接口獲取文章列表
* @param pageUrlList
* @returns {Promise}
*/
const getArticleList = (pageUrlList) => {
return new Promise((resolve, reject) => {
async.mapLimit(pageUrlList, 1, (pageUrl, callback) => {
getCurPage(pageUrl, callback);
}, (err, result) => {
if (err) {
logger.error('get article list error...');
logger.error(err);
reject(false);
return;
}
let articleList = _.flatten(result);
downloadThumbAndSave(articleList, resolve);
})
})
};
/**
* 獲取當(dāng)前頁面的文章列表
* @param pageUrl
* @param callback
* @returns {Promise.<void>}
*/
const getCurPage = async(pageUrl, callback) => {
let num = Math.random() * 1000 + 1000;
await sleep(num);
request(pageUrl, (err, response, body) => {
if (err) {
logger.info('current url went wrong,url address:' + pageUrl);
callback(null, null);
return;
} else {
let responseObj = JSON.parse(body);
if (responseObj.result && responseObj.result.list) {
let newsList = parseObject(articleModel, responseObj.result.list, {
pubTime: 'inputtime',
author: 'aid',
commentCount: 'comments',
});
callback(null, newsList);
return;
}
console.log("出錯(cuò)了");
callback(null, null);
}
});
};
const downloadThumbAndSave = (list, resolve) => {
const host = 'https://static.cnbetacdn.com';
const basepath = './public/data';
if (list.indexOf(null) > -1) {
resolve(false);
} else {
try {
async.eachSeries(list, (item, callback) => {
let thumb_url = item.thumb.replace(host, '');
item.thumb = thumb_url;
if (!fs.exists(thumb_url)) {
mkDirs(basepath + thumb_url.substring(0, thumb_url.lastIndexOf('/')), () => {
request
.get({
url: host + thumb_url,
})
.pipe(fs.createWriteStream(path.join(basepath, thumb_url)))
.on('error', (err) => {
console.log("pipe error", err);
});
callback(null, null);
});
}
}, (err, result) => {
if (!err) {
saveDB(list, resolve);
}
});
}
catch(err) {
console.log(err);
}
}
};
/**
* 將文章列表存入數(shù)據(jù)庫
* @param result
* @param callback
* @returns {Promise.<void>}
*/
const saveDB = async(result, callback) => {
//console.log(result);
let flag = await dbHelper.insertCollection(articleDbModel, result).catch(function (err){
logger.error('data insert falied');
});
if (!flag) {
logger.error('news list save failed');
} else {
logger.info('list saved!total:' + result.length);
}
if (typeof callback === 'function') {
callback(true);
}
};
再來看抓取新聞內(nèi)容的邏輯,這里是直接根據(jù)新聞的sid得到新聞內(nèi)容頁的html,然后利用cheerio庫分析獲取我們需要的新聞內(nèi)容。當(dāng)然這里也是要把文章中的圖片爬下來存入服務(wù)器,并且把存入數(shù)據(jù)庫的新聞內(nèi)容中圖片鏈接替換成自己服務(wù)器中的URL。
/**
* 抓取正文程序入口
* @returns {Promise.<*>}
*/
const articleContentInit = async() => {
logger.info('grabbing article contents starts...');
let uncachedArticleSidList = await getUncachedArticleList(articleDbModel);
// console.log('未緩存的文章:'+ uncachedArticleSidList.join(','));
const res = await batchCrawlArticleContent(uncachedArticleSidList);
if (!res) {
logger.error('grabbing article contents went wrong...');
}
return res;
};
/**
* 查詢新聞列表獲取sid列表
* @param Model
* @returns {Promise.<void>}
*/
const getUncachedArticleList = async(Model) => {
const selectedArticleList = await dbHelper.queryDocList(Model).catch(function (err){
logger.error(err);
});
return selectedArticleList.map(item => item.sid);
// return selectedArticleList.map(item => item._doc.sid);
};
/**
* 批量抓取新聞詳情內(nèi)容
* @param list
* @returns {Promise}
*/
const batchCrawlArticleContent = (list) => {
return new Promise((resolve, reject) => {
async.mapLimit(list, 3, (sid, callback) => {
getArticleContent(sid, callback);
}, (err, result) => {
if (err) {
logger.error(err);
reject(false);
return;
}
resolve(true);
});
});
};
/**
* 抓取單篇文章內(nèi)容
* @param sid
* @param callback
* @returns {Promise.<void>}
*/
const getArticleContent = async(sid, callback) => {
let num = Math.random() * 1000 + 1000;
await sleep(num);
let url = contentBaseUrl + sid + '.htm';
request(url, (err, response, body) => {
if (err) {
logger.error('grabbing article content went wrong,article url:' + url);
callback(null, null);
return;
}
const $ = cheerio.load(body, {
decodeEntities: false
});
const serverAssetPath = `${serverIp}:${serverPort}/data`;
let domainReg = new RegExp('https://static.cnbetacdn.com','g');
let article = {
sid,
source: $('.article-byline span a').html() || $('.article-byline span').html(),
summary: $('.article-summ p').html(),
content: $('.articleCont').html().replace(styleReg.reg, styleReg.replace).replace(scriptReg.reg, scriptReg.replace).replace(domainReg, serverAssetPath),
};
saveContentToDB(article);
let imgList = [];
$('.articleCont img').each((index, dom) => {
imgList.push(dom.attribs.src);
});
downloadImgs(imgList);
callback(null, null);
});
};
/**
* 下載圖片
* @param list
*/
const downloadImgs = (list) => {
const host = 'https://static.cnbetacdn.com';
const basepath = './public/data';
if (!list.length) {
return;
}
try {
async.eachSeries(list, (item, callback) => {
let num = Math.random() * 500 + 500;
sleep(num);
if (item.indexOf(host) === -1) return;
let thumb_url = item.replace(host, '');
item.thumb = thumb_url;
if (!fs.exists(thumb_url)) {
mkDirs(basepath + thumb_url.substring(0, thumb_url.lastIndexOf('/')), () => {
request
.get({
url: host + thumb_url,
})
.pipe(fs.createWriteStream(path.join(basepath, thumb_url)))
.on("error", (err) => {
console.log("pipe error", err);
});
callback(null, null);
});
}
});
}
catch(err) {
console.log(err);
}
};
/**
* 保存到文章內(nèi)容到數(shù)據(jù)庫
* @param article
*/
const saveContentToDB = (item) => {
let flag = dbHelper.updateCollection(articleDbModel, item);
if (flag) {
logger.info('grabbing article content succeeded:' + item.sid);
}
};
爬蟲部分差不多就是這樣,還有一點(diǎn)就自己服務(wù)器存儲(chǔ)的爬取的圖片每天都會(huì)有上百張,時(shí)間一長(zhǎng),圖片占用的存儲(chǔ)空間就會(huì)特別大,所以需要定時(shí)清理一下,有興趣的可以看看項(xiàng)目里面的clear-expire.js文件。
總結(jié)
其實(shí),雖然這個(gè)項(xiàng)目整體并不復(fù)雜,但是一套前后端系統(tǒng)搭建起來的過程中,自己的收獲還是挺不少的,很多問題的解決需要自己去實(shí)踐和思考的,對(duì)于性能優(yōu)化考量也是一個(gè)重要的方面。
下面截圖就是我最終完成得m站,界面很清爽,體驗(yàn)上確實(shí)比cnBeta官網(wǎng)要好很多。這樣是平時(shí)看科技新聞也確實(shí)方便很多。
