From dabd9cbe1e5b89e234c1e0c9fa87f7110945d34d Mon Sep 17 00:00:00 2001
From: Jack Robison <jackrobison@lbry.io>
Date: Fri, 1 Feb 2019 11:37:51 -0500
Subject: [PATCH] faster stream manager startup

---
 lbrynet/stream/stream_manager.py | 44 ++++++++++++++++++--------------
 1 file changed, 25 insertions(+), 19 deletions(-)

diff --git a/lbrynet/stream/stream_manager.py b/lbrynet/stream/stream_manager.py
index 66e9cdf06..39a13baf0 100644
--- a/lbrynet/stream/stream_manager.py
+++ b/lbrynet/stream/stream_manager.py
@@ -12,7 +12,6 @@ from lbrynet.extras.daemon.storage import lbc_to_dewies
 if typing.TYPE_CHECKING:
     from lbrynet.conf import Config
     from lbrynet.blob.blob_manager import BlobFileManager
-    from lbrynet.dht.peer import KademliaPeer
     from lbrynet.dht.node import Node
     from lbrynet.extras.daemon.storage import SQLiteStorage
     from lbrynet.extras.wallet import LbryWalletManager
@@ -64,25 +63,32 @@ class StreamManager:
         claim_info = await self.storage.get_content_claim(stream.stream_hash)
         stream.set_claim(claim_info, smart_decode(claim_info['value']))
 
+    async def add_stream(self, sd_hash: str, file_name: str, download_directory: str, status: str, claim):
+        sd_blob = self.blob_manager.get_blob(sd_hash)
+        if sd_blob.get_is_verified():
+            descriptor = await self.blob_manager.get_stream_descriptor(sd_blob.blob_hash)
+            downloader = StreamDownloader(
+                self.loop, self.config, self.blob_manager, descriptor.sd_hash,
+                download_directory,
+                file_name
+            )
+            stream = ManagedStream(
+                self.loop, self.blob_manager, descriptor,
+                download_directory,
+                file_name,
+                downloader, status, claim
+            )
+            self.streams.add(stream)
+            self.storage.content_claim_callbacks[stream.stream_hash] = lambda: self._update_content_claim(stream)
+
     async def load_streams_from_database(self):
-        infos = await self.storage.get_all_lbry_files()
-        for file_info in infos:
-            sd_blob = self.blob_manager.get_blob(file_info['sd_hash'])
-            if sd_blob.get_is_verified():
-                descriptor = await self.blob_manager.get_stream_descriptor(sd_blob.blob_hash)
-                downloader = StreamDownloader(
-                    self.loop, self.config, self.blob_manager, descriptor.sd_hash,
-                    binascii.unhexlify(file_info['download_directory']).decode(),
-                    binascii.unhexlify(file_info['file_name']).decode()
-                )
-                stream = ManagedStream(
-                    self.loop, self.blob_manager, descriptor,
-                    binascii.unhexlify(file_info['download_directory']).decode(),
-                    binascii.unhexlify(file_info['file_name']).decode(),
-                    downloader, file_info['status'], file_info['claim']
-                )
-                self.streams.add(stream)
-                self.storage.content_claim_callbacks[stream.stream_hash] = lambda: self._update_content_claim(stream)
+        file_infos = await self.storage.get_all_lbry_files()
+        await asyncio.gather(*[
+            self.add_stream(
+                file_info['sd_hash'], binascii.unhexlify(file_info['file_name']).decode(),
+                binascii.unhexlify(file_info['download_directory']).decode(), file_info['status'], file_info['claim']
+            ) for file_info in file_infos
+        ])
 
     async def resume(self):
         if not self.node: