137 lines
3.3 KiB
JavaScript
137 lines
3.3 KiB
JavaScript
/*
|
|
* @Descripttion:
|
|
* @version:
|
|
* @Author: WangPeng
|
|
* @Date: 2023-07-10 16:56:01
|
|
* @LastEditors: WangPeng
|
|
* @LastEditTime: 2023-07-12 11:34:50
|
|
*/
|
|
const Router = require("koa-router");
|
|
const douyinRouter = new Router();
|
|
const axios = require("axios");
|
|
const { get, set, del } = require("../utils/cacheData");
|
|
|
|
// 缓存键名
|
|
const cacheKey = "douyinData";
|
|
|
|
// 调用时间
|
|
let updateTime = new Date().toISOString();
|
|
|
|
// 调用路径
|
|
const url = "https://aweme.snssdk.com/aweme/v1/hot/search/list/";
|
|
const HEADERS = {
|
|
'user-agent': 'okhttp3'
|
|
}
|
|
const QUERIES = {
|
|
'device_platform': 'android',
|
|
'version_name': '13.2.0',
|
|
'version_code': '130200',
|
|
'aid': '1128'
|
|
}
|
|
|
|
// 数据处理
|
|
const getData = (data) => {
|
|
if (!data) return [];
|
|
return data.map((v) => {
|
|
return {
|
|
id: v.group_id,
|
|
title: v.word,
|
|
pic: v.word_cover?.url_list[0] || null,
|
|
hot: v.hot_value,
|
|
url: `https://www.douyin.com/hot?modal_id=${v.group_id}`,
|
|
mobileUrl: `snssdk1128://share/video/${v.group_id}`,
|
|
};
|
|
});
|
|
};
|
|
|
|
// 抖音热榜
|
|
douyinRouter.get("/douyin2", async (ctx) => {
|
|
console.log("获取抖音热榜");
|
|
try {
|
|
// 从缓存中获取数据
|
|
let data = await get(cacheKey);
|
|
const from = data ? "cache" : "server";
|
|
if (!data) {
|
|
// 如果缓存中不存在数据
|
|
console.log("从服务端重新获取抖音热榜");
|
|
// 从服务器拉取数据
|
|
const response = await axios.get(url, { headers: HEADERS, params: QUERIES });
|
|
data = getData(response.data.data.word_list);
|
|
updateTime = new Date().toISOString();
|
|
// 将数据写入缓存
|
|
await set(cacheKey, data);
|
|
}
|
|
ctx.body = {
|
|
code: 200,
|
|
message: "获取成功",
|
|
title: "抖音",
|
|
subtitle: "热榜",
|
|
from,
|
|
total: data.length,
|
|
updateTime,
|
|
data,
|
|
};
|
|
} catch (error) {
|
|
console.error(error);
|
|
ctx.body = {
|
|
code: 500,
|
|
title: "抖音",
|
|
subtitle: "热榜",
|
|
message: "获取失败",
|
|
};
|
|
}
|
|
});
|
|
|
|
// 抖音热榜 - 获取最新数据
|
|
douyinRouter.get("/douyin2/new", async (ctx) => {
|
|
console.log("获取抖音热榜 - 最新数据");
|
|
try {
|
|
// 从服务器拉取最新数据
|
|
const response = await axios.get(url, { headers: HEADERS, params: QUERIES });
|
|
const newData = getData(response.data.word_list);
|
|
updateTime = new Date().toISOString();
|
|
console.log("从服务端重新获取抖音热榜");
|
|
|
|
// 返回最新数据
|
|
ctx.body = {
|
|
code: 200,
|
|
message: "获取成功",
|
|
title: "抖音",
|
|
subtitle: "热榜",
|
|
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: "获取成功",
|
|
title: "抖音",
|
|
subtitle: "热榜",
|
|
total: cachedData.length,
|
|
updateTime,
|
|
data: cachedData,
|
|
};
|
|
} else {
|
|
// 如果缓存中也没有数据，则返回错误信息
|
|
ctx.body = {
|
|
code: 500,
|
|
title: "抖音",
|
|
subtitle: "热榜",
|
|
message: "获取失败",
|
|
};
|
|
}
|
|
}
|
|
});
|
|
|
|
module.exports = douyinRouter;
|