wait_until_first_data_arrives added to MyNoSqlDataReader

MyJetTools · Oct 4, 2023 · d286924 · d286924
1 parent c4a7d2a
commit d286924
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 13 deletions.
diff --git a/my-no-sql-tcp-reader/src/subscribers/my_no_sql_data_reader.rs b/my-no-sql-tcp-reader/src/subscribers/my_no_sql_data_reader.rs
@@ -29,4 +29,6 @@ pub trait MyNoSqlDataReader<TMyNoSqlEntity: MyNoSqlEntity + Sync + Send + 'stati
     ) -> GetEntityBuilder<TMyNoSqlEntity>;
 
     async fn has_partition(&self, partition_key: &str) -> bool;
+
+    async fn wait_until_first_data_arrives(&self);
 }
diff --git a/my-no-sql-tcp-reader/src/subscribers/my_no_sql_data_reader_tcp.rs b/my-no-sql-tcp-reader/src/subscribers/my_no_sql_data_reader_tcp.rs
@@ -175,19 +175,6 @@ where
 
         result
     }
-
-    pub async fn wait_until_first_data_arrives(&self) {
-        loop {
-            {
-                let reader = self.inner.data.read().await;
-                if reader.has_entities_at_all().await {
-                    return;
-                }
-            }
-
-            tokio::time::sleep(Duration::from_millis(100)).await;
-        }
-    }
 }
 
 #[async_trait]
@@ -263,4 +250,17 @@ where
     async fn has_partition(&self, partition_key: &str) -> bool {
         self.has_partition(partition_key).await
     }
+
+    async fn wait_until_first_data_arrives(&self) {
+        loop {
+            {
+                let reader = self.inner.data.read().await;
+                if reader.has_entities_at_all().await {
+                    return;
+                }
+            }
+
+            tokio::time::sleep(Duration::from_millis(100)).await;
+        }
+    }
 }