From 70abd5785286f062140faf17ab6d43b20dd75397 Mon Sep 17 00:00:00 2001
From: Mauricio Colli <mauriciocolli@outlook.com>
Date: Sat, 7 Mar 2020 16:46:00 -0300
Subject: [PATCH] [YouTube] Cache commonly used stream type result

---
 .../extractors/YoutubeStreamInfoItemExtractor.java    | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/extractor/src/main/java/org/schabi/newpipe/extractor/services/youtube/extractors/YoutubeStreamInfoItemExtractor.java b/extractor/src/main/java/org/schabi/newpipe/extractor/services/youtube/extractors/YoutubeStreamInfoItemExtractor.java
index 00aa8ae2b..2330130eb 100644
--- a/extractor/src/main/java/org/schabi/newpipe/extractor/services/youtube/extractors/YoutubeStreamInfoItemExtractor.java
+++ b/extractor/src/main/java/org/schabi/newpipe/extractor/services/youtube/extractors/YoutubeStreamInfoItemExtractor.java
@@ -40,6 +40,7 @@ public class YoutubeStreamInfoItemExtractor implements StreamInfoItemExtractor {
 
     private JsonObject videoInfo;
     private final TimeAgoParser timeAgoParser;
+    private StreamType cachedStreamType;
 
     /**
      * Creates an extractor of StreamInfoItems from a YouTube page.
@@ -54,11 +55,15 @@ public class YoutubeStreamInfoItemExtractor implements StreamInfoItemExtractor {
 
     @Override
     public StreamType getStreamType() {
+        if (cachedStreamType != null) {
+            return cachedStreamType;
+        }
+
         try {
             JsonArray badges = videoInfo.getArray("badges");
             for (Object badge : badges) {
                 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)
                     .getObject("thumbnailOverlayTimeStatusRenderer").getString("style");
             if (style.equalsIgnoreCase("LIVE")) {
-                return StreamType.LIVE_STREAM;
+                return cachedStreamType = StreamType.LIVE_STREAM;
             }
         } catch (Exception ignored) {}
 
-        return StreamType.VIDEO_STREAM;
+        return cachedStreamType = StreamType.VIDEO_STREAM;
     }
 
     @Override