Clone storage factories...

crates/librqbit/src/storage/examples/inmemory.rs

@@ -6,7 +6,7 @@ use parking_lot::RwLock;
 
 use crate::type_aliases::FileInfos;
 
-use crate::storage::{StorageFactory, TorrentStorage};
+use crate::storage::{StorageFactory, StorageFactoryExt, TorrentStorage};
 
 struct InMemoryPiece {
     bytes: Box<[u8]>,
@@ -19,7 +19,7 @@ impl InMemoryPiece {
     }
 }
 
-#[derive(Default)]
+#[derive(Default, Clone)]
 pub struct InMemoryExampleStorageFactory {}
 
 impl StorageFactory for InMemoryExampleStorageFactory {
@@ -31,6 +31,10 @@ impl StorageFactory for InMemoryExampleStorageFactory {
     ) -> anyhow::Result<InMemoryExampleStorage> {
         InMemoryExampleStorage::new(info.lengths, info.file_infos.clone())
     }
+
+    fn clone_box(&self) -> crate::storage::BoxStorageFactory {
+        self.clone().boxed()
+    }
 }
 
 pub struct InMemoryExampleStorage {

crates/librqbit/src/storage/examples/mmap.rs

@@ -3,11 +3,11 @@ use memmap2::{MmapMut, MmapOptions};
 use parking_lot::RwLock;
 
 use crate::{
-    storage::{StorageFactory, TorrentStorage},
+    storage::{StorageFactory, StorageFactoryExt, TorrentStorage},
     FileInfos, ManagedTorrentInfo,
 };
 
-#[derive(Default)]
+#[derive(Default, Clone)]
 pub struct MmapStorageFactory {}
 
 pub struct MmapStorage {
@@ -28,6 +28,10 @@ impl StorageFactory for MmapStorageFactory {
             file_infos: info.file_infos.clone(),
         })
     }
+
+    fn clone_box(&self) -> crate::storage::BoxStorageFactory {
+        self.clone().boxed()
+    }
 }
 
 impl TorrentStorage for MmapStorage {

crates/librqbit/src/storage/filesystem/fs.rs

@@ -6,13 +6,13 @@ use std::{
 
 use anyhow::Context;
 
-use crate::torrent_state::ManagedTorrentInfo;
+use crate::{storage::StorageFactoryExt, torrent_state::ManagedTorrentInfo};
 
 use crate::storage::{StorageFactory, TorrentStorage};
 
 use super::opened_file::OpenedFile;
 
-#[derive(Default)]
+#[derive(Default, Clone, Copy)]
 pub struct FilesystemStorageFactory {}
 
 impl StorageFactory for FilesystemStorageFactory {
@@ -59,6 +59,10 @@ impl StorageFactory for FilesystemStorageFactory {
             opened_files: files,
         })
     }
+
+    fn clone_box(&self) -> crate::storage::BoxStorageFactory {
+        self.boxed()
+    }
 }
 
 pub struct FilesystemStorage {

crates/librqbit/src/storage/filesystem/mmap.rs

@@ -6,11 +6,11 @@ use parking_lot::RwLock;
 
 use crate::torrent_state::ManagedTorrentInfo;
 
-use crate::storage::{StorageFactory, TorrentStorage};
+use crate::storage::{StorageFactory, StorageFactoryExt, TorrentStorage};
 
 use super::{FilesystemStorage, FilesystemStorageFactory};
 
-#[derive(Default)]
+#[derive(Default, Clone, Copy)]
 pub struct MmapFilesystemStorageFactory {}
 
 type OpenedMmap = RwLock<MmapMut>;
@@ -36,6 +36,10 @@ impl StorageFactory for MmapFilesystemStorageFactory {
             fs: fs_storage,
         })
     }
+
+    fn clone_box(&self) -> crate::storage::BoxStorageFactory {
+        self.boxed()
+    }
 }
 
 pub struct MmapFilesystemStorage {

crates/librqbit/src/storage/middleware/slow.rs

@@ -2,8 +2,9 @@ use std::time::Duration;
 
 use rand_distr::Distribution;
 
-use crate::storage::{StorageFactory, TorrentStorage};
+use crate::storage::{StorageFactory, StorageFactoryExt, TorrentStorage};
 
+#[derive(Clone)]
 pub struct SlowStorageFactory<U> {
     underlying_factory: U,
 }
@@ -16,7 +17,7 @@ impl<U: StorageFactory> SlowStorageFactory<U> {
     }
 }
 
-impl<U: StorageFactory> StorageFactory for SlowStorageFactory<U> {
+impl<U: StorageFactory + Clone> StorageFactory for SlowStorageFactory<U> {
     type Storage = SlowStorage<U::Storage>;
 
     fn init_storage(&self, info: &crate::ManagedTorrentInfo) -> anyhow::Result<Self::Storage> {
@@ -28,6 +29,10 @@ impl<U: StorageFactory> StorageFactory for SlowStorageFactory<U> {
     fn is_type_id(&self, type_id: std::any::TypeId) -> bool {
         self.underlying_factory.is_type_id(type_id)
     }
+
+    fn clone_box(&self) -> crate::storage::BoxStorageFactory {
+        self.clone().boxed()
+    }
 }
 
 pub struct SlowStorage<U> {

crates/librqbit/src/storage/middleware/timing.rs

@@ -1,5 +1,6 @@
-use crate::storage::{StorageFactory, TorrentStorage};
+use crate::storage::{StorageFactory, StorageFactoryExt, TorrentStorage};
 
+#[derive(Clone)]
 pub struct TimingStorageFactory<U> {
     name: String,
     underlying_factory: U,
@@ -14,7 +15,7 @@ impl<U> TimingStorageFactory<U> {
     }
 }
 
-impl<U: StorageFactory> StorageFactory for TimingStorageFactory<U> {
+impl<U: StorageFactory + Clone> StorageFactory for TimingStorageFactory<U> {
     type Storage = TimingStorage<U::Storage>;
 
     fn init_storage(&self, info: &crate::ManagedTorrentInfo) -> anyhow::Result<Self::Storage> {
@@ -27,6 +28,10 @@ impl<U: StorageFactory> StorageFactory for TimingStorageFactory<U> {
     fn is_type_id(&self, type_id: std::any::TypeId) -> bool {
         self.underlying_factory.is_type_id(type_id)
     }
+
+    fn clone_box(&self) -> crate::storage::BoxStorageFactory {
+        self.clone().boxed()
+    }
 }
 
 pub struct TimingStorage<U> {

crates/librqbit/src/storage/mod.rs

@@ -16,6 +16,7 @@ pub trait StorageFactory: Send + Sync + Any {
     fn is_type_id(&self, type_id: TypeId) -> bool {
         Self::type_id(self) == type_id
     }
+    fn clone_box(&self) -> BoxStorageFactory;
 }
 
 pub type BoxStorageFactory = Box<dyn StorageFactory<Storage = Box<dyn TorrentStorage>>>;
@@ -41,6 +42,10 @@ impl<SF: StorageFactory> StorageFactoryExt for SF {
             fn is_type_id(&self, type_id: TypeId) -> bool {
                 self.sf.type_id() == type_id
             }
+
+            fn clone_box(&self) -> BoxStorageFactory {
+                self.sf.clone_box()
+            }
         }
 
         Box::new(Wrapper { sf: self })
@@ -53,6 +58,10 @@ impl<U: StorageFactory + ?Sized> StorageFactory for Box<U> {
     fn init_storage(&self, info: &ManagedTorrentInfo) -> anyhow::Result<U::Storage> {
         (**self).init_storage(info)
     }
+
+    fn clone_box(&self) -> BoxStorageFactory {
+        (**self).clone_box()
+    }
 }
 
 pub trait TorrentStorage: Send + Sync {