[Feature] [Fix] Load Mooncake config from dict, when lack params, load from env config file.

Author: propanone1006

docs/source/getting-started/example/mooncake_conn.md

@@ -53,7 +53,7 @@ To use the Mooncake connector, you need to configure the `connector_config` dict
   The metadata server of the mooncake transfer engine.
 - `master_server_address`:  
   The IP address and the port of the master daemon process of MooncakeStore.
-- `protocl`  *(optional)*:  
+- `protocol`  *(optional)*:  
   If not provided, it defaults to **tcp**.
 - `device_name`  *(optional)*:  
   The device to be used for data transmission, it is required when “protocol” is set to “rdma”. If multiple NIC devices are used, they can be separated by commas such as “erdma_0,erdma_1”. Please note that there are no spaces between them.

unifiedcache/ucm_connector/ucm_mooncake.py

@@ -14,8 +14,8 @@
 from unifiedcache.ucm_connector import Task, UcmKVStoreBase
 
 TIMEOUT_S_THR: int = 60 * 60
-DEFAULT_GLOBAL_SEGMENT_SIZE: int = 1 * 1024 * 1024 * 1024  # 3.125 GiB
-DEFAULT_LOCAL_BUFFER_SIZE: int = 1 * 1024 * 1024 * 1024  # 1.0 GiB
+DEFAULT_GLOBAL_SEGMENT_SIZE: int = 3355443200  # 3.125 GiB
+DEFAULT_LOCAL_BUFFER_SIZE: int = 1073741824  # 1.0 GiB
 
 logger = init_logger(__name__)
 
@@ -111,11 +111,10 @@ def __init__(self, config: Dict = {}):
 
         try:
             self.store = MooncakeDistributedStore()
-            if config != {}:
-                mooncake_config = MooncakeStoreConfig.load_from_dict(config)
-            else:
-                mooncake_config = MooncakeStoreConfig.load_from_env()
-            logger.info("Mooncake Configuration loaded successfully.")
+
+            mooncake_config = MooncakeStoreConfig.load_from_dict(config)
+            logger.info("Mooncake Configuration loaded from dict successfully.")
+
             self.store.setup(
                 mooncake_config.local_hostname,
                 mooncake_config.metadata_server,
@@ -129,6 +128,34 @@ def __init__(self, config: Dict = {}):
         except ValueError as e:
             logger.error("Configuration loading failed: %s", e)
             raise
+        except TypeError:
+            logger.warning(
+                "Lack of configuration, loading Mooncake configuration from environment variables instead."
+            )
+            try:
+                mooncake_config = MooncakeStoreConfig.load_from_env()
+                logger.info("Mooncake Configuration loaded from env successfully.")
+                self.store.setup(
+                    mooncake_config.local_hostname,
+                    mooncake_config.metadata_server,
+                    mooncake_config.global_segment_size,
+                    mooncake_config.local_buffer_size,
+                    mooncake_config.protocol,
+                    mooncake_config.device_name,
+                    mooncake_config.master_server_address,
+                )
+            except ValueError as e:
+                logger.error(
+                    "Configuration loading failed: %s \n Please check the dict params or edit the config file and set path.",
+                    e,
+                )
+                raise
+            except Exception as exc:
+                logger.error(
+                    "An error occurred while loading the configuration: %s", exc
+                )
+                raise
+
         except Exception as exc:
             logger.error("An error occurred while loading the configuration: %s", exc)
             raise