make compatible to private

Author: Shubham8287

crates/core/src/host/host_controller.rs

@@ -904,7 +904,12 @@ impl Host {
             db::Storage::Disk => {
                 // Replay from the local state.
                 let history = relational_db::local_history(&replica_dir, &runtime).await?;
-                let persistence = persistence.persistence(database.database_identity, replica_id).await?;
+                let persistence_db = db::persistence::Database {
+                    id: database.id,
+                    database_identity: database.database_identity,
+                    owner_identity: database.owner_identity,
+                };
+                let persistence = persistence.persistence(&persistence_db, replica_id).await?;
                 // Loading a database from persistent storage involves heavy
                 // blocking I/O. `asyncify` to avoid blocking the async worker.
                 let (db, clients) = asyncify({

crates/engine/src/db/persistence.rs

@@ -85,6 +85,13 @@ pub type Durability = dyn spacetimedb_durability::Durability<TxData = Txdata>;
 /// configured or the database is in follower state.
 pub type DiskSizeFn = Arc<dyn Fn() -> io::Result<SizeOnDisk> + Send + Sync>;
 
+#[derive(Clone, Copy, Debug)]
+pub struct Database {
+    pub id: u64,
+    pub database_identity: Identity,
+    pub owner_identity: Identity,
+}
+
 /// Persistence services for a database.
 pub struct Persistence {
     /// The [Durability] to use, for persisting transactions.
@@ -182,7 +189,7 @@ impl Persistence {
 /// This is an `async_trait` to allow it to be used as a trait object.
 #[async_trait]
 pub trait PersistenceProvider: Send + Sync {
-    async fn persistence(&self, database_identity: Identity, replica_id: u64) -> anyhow::Result<Persistence>;
+    async fn persistence(&self, database: &Database, replica_id: u64) -> anyhow::Result<Persistence>;
 }
 
 /// The standard [PersistenceProvider] for non-replicated databases.
@@ -215,7 +222,8 @@ impl LocalPersistenceProvider {
 
 #[async_trait]
 impl PersistenceProvider for LocalPersistenceProvider {
-    async fn persistence(&self, database_identity: Identity, replica_id: u64) -> anyhow::Result<Persistence> {
+    async fn persistence(&self, database: &Database, replica_id: u64) -> anyhow::Result<Persistence> {
+        let database_identity = database.database_identity;
         let replica_dir = self.data_dir.replica(replica_id);
         let snapshot_dir = replica_dir.snapshots();
         let runtime = Handle::tokio_current();