[YouTube] Cache commonly used stream type result

This commit is contained in:
Mauricio Colli 2020-03-07 16:46:00 -03:00
parent 1ef706f567
commit 70abd57852
No known key found for this signature in database
GPG Key ID: F200BFD6F29DDD85

View File

@ -40,6 +40,7 @@ public class YoutubeStreamInfoItemExtractor implements StreamInfoItemExtractor {
private JsonObject videoInfo; private JsonObject videoInfo;
private final TimeAgoParser timeAgoParser; private final TimeAgoParser timeAgoParser;
private StreamType cachedStreamType;
/** /**
* Creates an extractor of StreamInfoItems from a YouTube page. * Creates an extractor of StreamInfoItems from a YouTube page.
@ -54,11 +55,15 @@ public class YoutubeStreamInfoItemExtractor implements StreamInfoItemExtractor {
@Override @Override
public StreamType getStreamType() { public StreamType getStreamType() {
if (cachedStreamType != null) {
return cachedStreamType;
}
try { try {
JsonArray badges = videoInfo.getArray("badges"); JsonArray badges = videoInfo.getArray("badges");
for (Object badge : badges) { for (Object badge : badges) {
if (((JsonObject) badge).getObject("metadataBadgeRenderer").getString("label").equals("LIVE NOW")) { if (((JsonObject) badge).getObject("metadataBadgeRenderer").getString("label").equals("LIVE NOW")) {
return StreamType.LIVE_STREAM; return cachedStreamType = StreamType.LIVE_STREAM;
} }
} }
@ -68,11 +73,11 @@ public class YoutubeStreamInfoItemExtractor implements StreamInfoItemExtractor {
final String style = videoInfo.getArray("thumbnailOverlays").getObject(0) final String style = videoInfo.getArray("thumbnailOverlays").getObject(0)
.getObject("thumbnailOverlayTimeStatusRenderer").getString("style"); .getObject("thumbnailOverlayTimeStatusRenderer").getString("style");
if (style.equalsIgnoreCase("LIVE")) { if (style.equalsIgnoreCase("LIVE")) {
return StreamType.LIVE_STREAM; return cachedStreamType = StreamType.LIVE_STREAM;
} }
} catch (Exception ignored) {} } catch (Exception ignored) {}
return StreamType.VIDEO_STREAM; return cachedStreamType = StreamType.VIDEO_STREAM;
} }
@Override @Override