本篇文章給大家介紹一下node中借助第三方開源庫輕松實(shí)現(xiàn)網(wǎng)站爬取功能的方法,希望對大家有所幫助!
nodejs實(shí)現(xiàn)網(wǎng)站爬取功能
第三方庫介紹
-
request 對網(wǎng)絡(luò)請求的封裝
-
cheerio node 版本的 jQuery
-
mkdirp 創(chuàng)建多層的文件夾目錄
實(shí)現(xiàn)思路
-
通過
request
獲取指定 url 內(nèi)容 -
通過
cheerio
找到頁面中跳轉(zhuǎn)的路徑(去重) -
通過
mkdirp
創(chuàng)建目錄 -
通過
fs
創(chuàng)建文件,將讀取的內(nèi)容寫入 -
拿到?jīng)]有訪問的路徑重復(fù)以上執(zhí)行步驟
代碼實(shí)現(xiàn)
const fs = require("fs"); const path = require("path"); const request = require("request"); const cheerio = require("cheerio"); const mkdirp = require("mkdirp"); // 定義入口url const homeUrl = "https://www.baidu.com"; // 定義set存儲已經(jīng)訪問過的路徑,避免重復(fù)訪問 const set = new Set([homeUrl]); function grab(url) { // 校驗(yàn)url規(guī)范性 if (!url) return; // 去空格 url = url.trim(); // 自動補(bǔ)全url路徑 if (url.endsWith("/")) { url += "index.html"; } const chunks = []; // url可能存在一些符號或者中文,可以通過encodeURI編碼 request(encodeURI(url)) .on("error", (e) => { // 打印錯誤信息 console.log(e); }) .on("data", (chunk) => { // 接收響應(yīng)內(nèi)容 chunks.push(chunk); }) .on("end", () => { // 將相應(yīng)內(nèi)容轉(zhuǎn)換成文本 const html = Buffer.concat(chunks).toString(); // 沒有獲取到內(nèi)容 if (!html) return; // 解析url let { host, origin, pathname } = new URL(url); pathname = decodeURI(pathname); // 通過cheerio解析html const $ = cheerio.load(html); // 將路徑作為目錄 const dir = path.dirname(pathname); // 創(chuàng)建目錄 mkdirp.sync(path.join(__dirname, dir)); // 往文件寫入內(nèi)容 fs.writeFile(path.join(__dirname, pathname), html, "utf-8", (err) => { // 打印錯誤信息 if (err) { console.log(err); return; } console.log(`[${url}]保存成功`); }); // 獲取到頁面中所有a元素 const aTags = $("a"); Array.from(aTags).forEach((aTag) => { // 獲取到a標(biāo)簽中的路徑 const href = $(aTag).attr("href"); // 此處可以校驗(yàn)href的合法或者控制爬去的網(wǎng)站范圍,比如必須都是某個域名下的 // 排除空標(biāo)簽 if (!href) return; // 排除錨點(diǎn)連接 if (href.startsWith("#")) return; if (href.startsWith("mailto:")) return; // 如果不想要保存圖片可以過濾掉 // if (/.(jpg|jpeg|png|gif|bit)$/.test(href)) return; // href必須是入口url域名 let reg = new RegExp(`^https?://${host}`); if (/^https?:///.test(href) && !reg.test(href)) return; // 可以根據(jù)情況增加