xiaoxue_admin
/
RSSHub
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: youtube key retry

This commit is contained in:
DIYgod 2020-09-25 12:01:01 +08:00
parent 03d6e7f04e
commit 7e777bbb56
No known key found for this signature in database
GPG Key ID: D159328F47A80DCA
1 changed files with 47 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

73
lib/routes/youtube/utils.js
View File

@ -1,12 +1,10 @@
const { google } = require('googleapis');
const config = require('@/config').value;
let getYoutube = () => null;
let count = 0;
const youtube = {};
if (config.youtube.key) {
const keys = config.youtube.key.split(',');
const youtube = {};
let count = 0;
let index = -1;
keys.forEach((key, index) => {
if (key) {
@ -17,44 +15,67 @@ if (config.youtube.key) {
count = index + 1;
}
});
getYoutube = () => {
index++;
return youtube[index % count];
};
}
let index = -1;
const exec = async (func) => {
let result;
for (let i = 0; i < count; i++) {
index++;
try {
// eslint-disable-next-line no-await-in-loop
result = await func(youtube[index % count]);
break;
} catch (error) {
// console.error(error);
}
}
return result;
};
const youtubeUtils = {
getPlaylistItems: async (id, part) => {
const res = await getYoutube().playlistItems.list({
part,
playlistId: id,
maxResults: 50, // youtube api param value default is 5
});
const res = await exec(
async (youtube) =>
await youtube.playlistItems.list({
part,
playlistId: id,
maxResults: 50, // youtube api param value default is 5
})
);
return res;
},
getPlaylist: async (id, part, cache) =>
await cache.tryGet('getPlaylist' + id, async () => {
const res = await getYoutube().playlists.list({
part,
id: id,
});
const res = await exec(
async (youtube) =>
await youtube.playlists.list({
part,
id: id,
})
);
return res;
}),
getChannelWithId: async (id, part, cache) =>
await cache.tryGet('getChannelWithId' + id, async () => {
const res = await getYoutube().channels.list({
part,
id: id,
});
const res = await exec(
async (youtube) =>
await youtube.channels.list({
part,
id: id,
})
);
return res;
}),
getChannelWithUsername: async (username, part, cache) =>
await cache.tryGet('getPlaylist' + username, async () => {
const res = await getYoutube().channels.list({
part,
forUsername: username,
});
const res = await exec(
async (youtube) =>
await youtube.channels.list({
part,
forUsername: username,
})
);
return res;
}),
getThumbnail: (thumbnails) => thumbnails.maxres || thumbnails.standard || thumbnails.high || thumbnails.medium || thumbnails.default,