add chia dev data sync-time --prefer-db-kv-blob-length (#19693)

Author: altendky

chia/cmds/dev/data.py

@@ -23,7 +23,7 @@
 from chia.cmds.cmd_helpers import NeedsWalletRPC
 from chia.data_layer.data_layer import server_files_path_from_config
 from chia.data_layer.data_layer_util import ServerInfo, Status, Subscription
-from chia.data_layer.data_store import DataStore
+from chia.data_layer.data_store import DataStore, default_prefer_file_kv_blob_length
 from chia.data_layer.download_data import insert_from_delta_file
 from chia.util.chia_logging import initialize_logging
 from chia.util.config import load_config
@@ -77,6 +77,11 @@ class SyncTimeCommand:
     profile_tasks: bool = option("--profile-tasks/--no-profile-tasks")
     restart_all: bool = option("--restart-all/--no-restart-all")
     working_path: Optional[Path] = option("--working-path", default=None)
+    prefer_db_kv_blob_length: int = option(
+        "--prefer-db-kv-blob-length",
+        default=default_prefer_file_kv_blob_length,
+        type=int,
+    )
 
     async def run(self) -> None:
         config = load_config(self.context.root_path, "config.yaml", "data_layer", fill_missing_services=True)
@@ -122,6 +127,7 @@ async def run(self) -> None:
                         database=database_path,
                         merkle_blobs_path=merkle_blob_path,
                         key_value_blobs_path=key_value_blob_path,
+                        prefer_db_kv_blob_length=self.prefer_db_kv_blob_length,
                     )
                 )
 

chia/data_layer/data_store.py

@@ -72,6 +72,8 @@
 
 KeyOrValueId = int64
 
+default_prefer_file_kv_blob_length: int = 4096
+
 
 @dataclass
 class DataStore:
@@ -81,7 +83,7 @@ class DataStore:
     recent_merkle_blobs: LRUCache[bytes32, MerkleBlob]
     merkle_blobs_path: Path
     key_value_blobs_path: Path
-    prefer_db_kv_blob_length: int = 4096
+    prefer_db_kv_blob_length: int = default_prefer_file_kv_blob_length
 
     @classmethod
     @contextlib.asynccontextmanager
@@ -93,6 +95,7 @@ async def managed(
         uri: bool = False,
         sql_log_path: Optional[Path] = None,
         cache_capacity: int = 1,
+        prefer_db_kv_blob_length: int = default_prefer_file_kv_blob_length,
     ) -> AsyncIterator[DataStore]:
         async with DBWrapper2.managed(
             database=database,
@@ -113,6 +116,7 @@ async def managed(
                 recent_merkle_blobs=recent_merkle_blobs,
                 merkle_blobs_path=merkle_blobs_path,
                 key_value_blobs_path=key_value_blobs_path,
+                prefer_db_kv_blob_length=prefer_db_kv_blob_length,
             )
 
             async with db_wrapper.writer() as writer: