DailyHotApi/routes/weread.js

139 lines
3.5 KiB
JavaScript
Raw Normal View History

2023-06-01 09:35:41 +00:00
const Router = require("koa-router");
const wereadRouter = new Router();
const axios = require("axios");
const getWereadID = require("../utils/getWereadID");
2023-06-01 09:35:41 +00:00
const { get, set, del } = require("../utils/cacheData");
2023-07-03 08:14:40 +00:00
// 接口信息
const routerInfo = {
name: "weread",
2023-07-03 08:14:40 +00:00
title: "微信读书",
subtitle: "飙升榜",
};
2023-06-01 09:35:41 +00:00
// 缓存键名
const cacheKey = "wereadData";
// 调用时间
let updateTime = new Date().toISOString();
// 调用路径
2023-07-03 08:14:40 +00:00
const url = "https://weread.qq.com/web/bookListInCategory/rising?rank=1";
2023-06-01 09:35:41 +00:00
// 数据处理
const getData = (data) => {
if (!data) return [];
return data.map((v) => {
const book = v.bookInfo;
return {
2023-07-03 08:14:40 +00:00
id: book.bookId,
title: book.title,
desc: book.intro,
pic: book.cover.replace("s_", "t9_"),
hot: v.readingCount,
author: book.author,
url: `https://weread.qq.com/web/bookDetail/${getWereadID(book.bookId)}`,
mobileUrl: `https://weread.qq.com/web/bookDetail/${getWereadID(book.bookId)}`,
2023-06-01 09:35:41 +00:00
};
});
};
// 微信读书
wereadRouter.get("/weread", async (ctx) => {
console.log("获取微信读书");
try {
// 从缓存中获取数据
let data = await get(cacheKey);
const from = data ? "cache" : "server";
if (!data) {
// 如果缓存中不存在数据
console.log("从服务端重新获取微信读书");
// 从服务器拉取数据
2023-07-03 08:14:40 +00:00
const response = await axios.get(url, {
Headers: {
"User-Agent":
"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0 Safari/537.36 Edg/114.0.1823.67",
},
});
2023-06-01 09:35:41 +00:00
data = getData(response.data.books);
updateTime = new Date().toISOString();
if (!data) {
ctx.body = {
code: 500,
2023-07-03 08:14:40 +00:00
...routerInfo,
2023-06-01 09:35:41 +00:00
message: "获取失败",
};
return false;
}
// 将数据写入缓存
await set(cacheKey, data);
}
ctx.body = {
code: 200,
message: "获取成功",
2023-07-03 08:14:40 +00:00
...routerInfo,
2023-06-01 09:35:41 +00:00
from,
total: data.length,
updateTime,
data,
};
} catch (error) {
console.error(error);
ctx.body = {
code: 500,
message: "获取失败",
};
}
});
// 微信读书 - 获取最新数据
wereadRouter.get("/weread/new", async (ctx) => {
console.log("获取微信读书 - 最新数据");
try {
// 从服务器拉取最新数据
const response = await axios.get(url);
2023-07-03 08:14:40 +00:00
const newData = getData(response.data.books);
2023-06-01 09:35:41 +00:00
updateTime = new Date().toISOString();
console.log("从服务端重新获取微信读书");
// 返回最新数据
ctx.body = {
code: 200,
message: "获取成功",
2023-07-03 08:14:40 +00:00
...routerInfo,
2023-06-01 09:35:41 +00:00
total: newData.length,
updateTime,
data: newData,
};
// 删除旧数据
await del(cacheKey);
// 将最新数据写入缓存
await set(cacheKey, newData);
} catch (error) {
// 如果拉取最新数据失败,尝试从缓存中获取数据
console.error(error);
const cachedData = await get(cacheKey);
if (cachedData) {
ctx.body = {
code: 200,
message: "获取成功",
2023-07-03 08:14:40 +00:00
...routerInfo,
2023-06-01 09:35:41 +00:00
total: cachedData.length,
updateTime,
data: cachedData,
};
} else {
// 如果缓存中也没有数据,则返回错误信息
ctx.body = {
code: 500,
2023-07-03 08:14:40 +00:00
...routerInfo,
2023-06-01 09:35:41 +00:00
message: "获取失败",
};
}
}
});
2023-07-03 08:14:40 +00:00
wereadRouter.info = routerInfo;
2023-06-01 09:35:41 +00:00
module.exports = wereadRouter;