client restructured

Author: EugeneSukharev1988

build_and_push.sh

@@ -2,7 +2,7 @@
 
 # Configuration
 PACKAGE_NAME="montycat"
-VERSION="0.1.40"
+VERSION="0.1.41"
 PYPI_TOKEN="${PYPI_TOKEN:-}"
 
 # Exit on any error

montycat/core/engine.py

@@ -95,9 +95,9 @@ async def create_store(self, persistent: bool = False) -> Any:
             bool
         """
         return await self._execute_query_with_credentials([
-            'create-store', "store", self.store, "persistent", "y" if persistent else "n"
+            'create-store', "store", self.store, "persistent", "y" if persistent else "n",
         ])
-
+    
     async def remove_store(self, persistent: bool = False) -> Any:
         """
         Removes an existing data store from the server.

montycat/core/store.py

@@ -80,6 +80,8 @@ class Persistent(generic_kv, persistent_kv):
         Attributes:
             persistent (bool): Always True for persistent stores.
             distributed (bool): Always False for the basic persistent store.
+            cache (Union[int, None]): Optional cache size for the store. Should be setup in MB. If parameter was not set default value (10 MB) will be used.
+            compression (bool): Indicates whether data compression is enabled. Default is False.
 
         Methods:
             __init__: Initializes the persistent store, inheriting from `generic_kv`.

montycat/store_classes/inmemory.py

@@ -171,4 +171,14 @@ async def get_keys(cls):
         cls.command = "get_keys"
 
         query = convert_to_binary_query(cls)
+        return await cls._run_query(query)
+
+    @classmethod  
+    async def create_keyspace(cls):
+
+        query = orjson.dumps({
+            "raw": ["create-keyspace", "store", cls.store, "keyspace", cls.keyspace, "persistent", "y" if cls.persistent else "n"],
+            "credentials": [cls.username, cls.password]
+        })
+
         return await cls._run_query(query)

montycat/store_classes/kv.py

@@ -278,16 +278,6 @@ def connect_engine(cls, engine: Engine) -> None:
         cls.port = engine.port
         cls.store = engine.store
 
-    @classmethod  
-    async def create_keyspace(cls):
-
-        query = orjson.dumps({
-            "raw": ["create-keyspace", "store", cls.store, "keyspace", cls.keyspace, "persistent", "y" if cls.persistent else "n"],
-            "credentials": [cls.username, cls.password]
-        })
-
-        return await cls._run_query(query)
-    
     @classmethod  
     async def remove_keyspace(cls):
 

montycat/store_classes/persistent.py

@@ -1,9 +1,13 @@
 from ..core.engine import send_data
 from ..store_functions.store_generic_functions import convert_to_binary_query, convert_custom_key, handle_limit
 from typing import Union
+import orjson
 
 class persistent_kv:
+
     persistent: bool = True
+    cache: Union[int, None] = None
+    compression: bool = False
 
     @classmethod
     async def _run_query(cls, query: str):
@@ -125,4 +129,30 @@ async def get_keys(cls, limit: Union[list, int] = []):
         cls.command = "get_keys"
 
         query = convert_to_binary_query(cls)
+        return await cls._run_query(query)
+    
+    @classmethod  
+    async def create_keyspace(cls):
+
+        query = orjson.dumps({
+            "raw": ["create-keyspace", "store", cls.store, "keyspace", cls.keyspace, "persistent", "y", "cache", cls.cache if cls.cache else None, "compression", "y" if cls.compression else "n"],
+            "credentials": [cls.username, cls.password]
+        })
+
+        return await cls._run_query(query)
+    
+    @classmethod
+    async def update_cache_and_compression(cls):
+        """
+        Updates the cache size and compression settings for the current store.
+
+        Returns:
+            bool
+        """
+
+        query = orjson.dumps({
+            "raw": ['update-cache-compression', "store", cls.store, "keyspace", cls.keyspace, "persistent", "y", "cache", cls.cache if cls.cache else None, "compression", "y" if cls.compression else "n"],
+            "credentials": [cls.username, cls.password]
+        })
+
         return await cls._run_query(query)

setup.py

@@ -2,7 +2,7 @@
 
 setup(
     name='montycat',
-    version='0.1.40',
+    version='0.1.41',
     description='A Python client for MontyCat, NoSQL store utilizing Data Mesh architecture.',
     packages=find_packages(),
     zip_safe=False,