Mirror/SponsorBlockServer
1
0
Fork 0
mirror of https://github.com/ajayyy/SponsorBlockServer.git synced 2025-12-13 06:57:05 +03:00
Code Issues Packages Projects Releases Wiki Activity

Added youtube api cache (optional)

Browse Source
This commit is contained in:
Joe Dowd
2020-09-08 17:37:29 +01:00
parent b79f740e2b
commit 9d94067cd0
9 changed files with 311 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
src/utils/youtubeAPI.js
View File

@@ -2,6 +2,8 @@ var config = require('../config.js');
// YouTube API
const YouTubeAPI = require("youtube-api");
const redis = require('./redis.js');
const logger = require('./logger.js');
var exportObject;
// If in test mode, return a mocked youtube object
@@ -14,6 +16,44 @@ if (config.mode === "test") {
key: config.youtubeAPIKey
});
exportObject = YouTubeAPI;
// YouTubeAPI.videos.list wrapper with cacheing
exportObject.listVideos = (videoID, part, callback) => {
let redisKey = "youtube.video." + videoID + "." + part;
redis.get(redisKey, (getErr, result) => {
if (getErr || !result) {
logger.debug("redis: no cache for video information: " + videoID);
YouTubeAPI.videos.list({
part,
id: videoID
}, (ytErr, data) => {
if (!ytErr) {
// Only set cache if data returned
if (data.items.length > 0) {
redis.set(redisKey, JSON.stringify(data), (setErr) => {
logger.debug("redis: video information cache set for: " + videoID);
setErr && logger.warn(setErr);
callback(false, data); // don't fail
});
} else {
callback(false, data); // don't fail
}
} else {
callback(ytErr, data)
}
});
} else {
logger.debug("redis: fetched video information from cache: " + videoID);
callback(getErr, JSON.parse(result));
}
});
};
}
/* YouTubeAPI.videos.list({
part: "snippet",
id: videoID
}, function (err, data) {*/
module.exports = exportObject;