quasardb 3.14.2.dev6__cp311-cp311-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl

quasardb/pool.py

@@ -0,0 +1,305 @@
+import logging
+import quasardb
+import threading
+import functools
+import weakref
+
+logger = logging.getLogger("quasardb.pool")
+
+
+def _create_conn(**kwargs):
+    return quasardb.Cluster(**kwargs)
+
+
+class SessionWrapper(object):
+    def __init__(self, pool, conn):
+        self._conn = conn
+        self._pool = pool
+
+    def __getattr__(self, attr):
+        # This hack copies all the quasardb.Cluster() properties, functions and
+        # whatnot, and pretends that this class is actually a quasardb.Cluster.
+        #
+        # The background is that when someone does this:
+        #
+        # with pool.connect() as conn:
+        #   ...
+        #
+        # we don't want the the connection to be closed near the end, but rather
+        # released back onto the pool.
+        #
+        # Now, my initial approach was to build a pool.Session class which inherited
+        # from quasardb.Cluster, and just overload the __exit__ function there. But,
+        # we want people to have the flexibility to pass in an external `get_conn` callback
+        # in the pool, which establishes a connection, because they may have to look up
+        # some dynamic security tokens. This function should then, in turn, return a vanilla
+        # quasardb.Cluster() object.
+        #
+        # And this is why we end up with the solution below.
+        if attr in self.__dict__:
+            return getattr(self, attr)
+
+        return getattr(self._conn, attr)
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self._pool.release(self._conn)
+
+
+class Pool(object):
+    """
+    A connection pool. This class should not be initialized directly, but
+    rather the subclass `SingletonPool` should be initialized.
+
+    The constructor either accepts all regular `quasardb.Cluster()` connection parameters,
+    or a `get_conn` parameter which is invoked any time a new connection should be
+    created.
+
+    Example usage:
+    --------------
+
+    Initialize the pool by passthrough of `quasardb.Cluster()` arguments:
+    ```
+    import quasardb.pool as pool
+
+    pool.SingletonPool(uri='qdb://127.0.0.1:2836',
+                       cluster_public_key='...',
+                       user_private_key='...',
+                       user_name='...')
+    ```
+
+    Initialize pool by providing a callback function
+    ```
+    import quasardb
+    import quasardb.pool as pool
+
+    def my_qdb_connection_create():
+       # This function is invoked any time the pool needs to allocate
+       # a new connection.
+       return quasardb.Cluster(uri='qdb://127.0.0.1:2836',
+                               cluster_public_key='...',
+                               user_private_key='...',
+                               user_name='...')
+
+    pool.SingletonPool(get_conn=my_qdb_connection_create)
+    ```
+
+    """
+
+    def __init__(self, get_conn=None, **kwargs):
+        self._all_connections = []
+
+        if get_conn is None:
+            get_conn = functools.partial(_create_conn, **kwargs)
+
+        if not callable(get_conn):
+            raise TypeError("get_conn must be callable")
+
+        self._get_conn = get_conn
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.close()
+
+    def _create_conn(self):
+        return SessionWrapper(self, self._get_conn())
+
+    def close(self):
+        """
+        Close this connection pool, and all associated connections. This function
+        is automatically invoked when used in a with-block or when using the global
+        `instance()` singleton.
+        """
+        for conn in self._all_connections:
+            logger.debug("closing connection {}".format(conn))
+            conn.close()
+
+    def connect(self) -> quasardb.Cluster:
+        """
+        Acquire a new connection from the pool. Returned connection must either
+        be explicitly released using `pool.release()`, or should be wrapped in a
+        with-block.
+
+        Returns:
+        --------
+        `quasardb.Cluster`
+        """
+        logger.info("Acquiring connection from pool")
+        return self._do_connect()
+
+    def release(self, conn: quasardb.Cluster):
+        """
+        Put a connection back into the pool
+        """
+        logger.info("Putting connection back onto pool")
+        return self._do_release(conn)
+
+
+class SingletonPool(Pool):
+    """
+    Implementation of our connection pool that maintains just a single connection
+    per thread.
+
+    Example usage:
+    --------
+
+    Using pool.connect() in a with-block:
+    ```
+    import quasardb.pool as pool
+
+    with pool.Pool(uri='qdb://127.0.0.1:2836') as pool:
+      with pool.connect() as conn:
+        conn.query(...)
+    ```
+
+    Explicitly releasing the connection using `Pool.release()`:
+    ```
+    import quasardb.pool as pool
+
+    with pool.Pool(uri='qdb://127.0.0.1:2836') as pool:
+      conn = pool.connect()
+      try:
+        conn.query(...)
+      finally:
+        pool.release(conn)
+    ```
+    """
+
+    def __init__(self, **kwargs):
+        Pool.__init__(self, **kwargs)
+        self._conn = threading.local()
+
+    def _do_connect(self):
+        try:
+            c = self._conn.current()
+            if c:
+                return c
+        except AttributeError:
+            pass
+
+        c = self._create_conn()
+        self._conn.current = weakref.ref(c)
+        self._all_connections.append(c)
+
+        return c
+
+    def _do_release(self, conn):
+        # Thread-local connections do not have to be 'released'.
+        pass
+
+
+__instance = None
+
+
+def initialize(*args, **kwargs):
+    """
+    Initialize a new global SingletonPool. Forwards all arguments to the constructor of
+    `SingletonPool()`.
+
+    After initialization, the instance can be used by invoking `instance()`.
+
+    Example usage:
+    --------------
+
+    ```
+    import quasardb.pool as pool
+
+    pool.initialize(cluster='qdb://127.0.0.1:2836')
+
+    # ...
+
+    def myfunction()
+      with pool.instance().connect() as conn:
+        conn.query(...)
+    ```
+    """
+    global __instance
+    __instance = SingletonPool(*args, **kwargs)
+
+
+def instance() -> SingletonPool:
+    """
+    Singleton accessor. Instance must have been initialized using initialize()
+    prior to invoking this function.
+
+    Returns:
+    --------
+    `SingletonPool`
+
+    """
+    global __instance
+    assert (
+        __instance is not None
+    ), "Global connection pool uninitialized: please initialize by calling the initialize() function."
+    return __instance
+
+
+def _inject_conn_arg(conn, arg, args, kwargs):
+    """
+    Decorator utility function. Takes the argument provided to the decorator
+    that configures how we should inject the pool into the args to the callback
+    function, and injects it in the correct place.
+    """
+    if isinstance(arg, int):
+        # Invoked positional such as `@with_pool(arg=1)`, put the pool into the
+        # correct position.
+        #
+        # Because positional args are always a tuple, and tuples don't have an
+        # easy 'insert into position' function, we just cast to and from a list.
+        args = list(args)
+        args.insert(arg, conn)
+        args = tuple(args)
+    else:
+        assert isinstance(arg, str) == True
+        # If not a number, we assume it's a kwarg, which makes things easier
+        kwargs[arg] = conn
+
+    return (args, kwargs)
+
+
+def with_conn(_fn=None, *, arg=0):
+    """
+    Decorator function that handles connection assignment, release and invocation for you.
+    Should be used in conjuction with the global singleton accessor, see also: `initialize()`.
+
+    By default, the decorator function injects the connection as the first argument to the
+    function:
+    ```
+    import quasardb.pool as pool
+    pool.initialize(...)
+
+    @pool.with_conn()
+    def myfunction(conn, arg1, arg2):
+       conn.query(...)
+    ```
+
+    You can optionally provide an `arg` parameter to denote which named keyword to provide
+    it as:
+    ```
+    import quasardb.pool as pool
+    pool.initialize(...)
+
+    @pool.with_conn(arg='conn')
+    def myfunction(arg1, arg2, conn=None):
+       conn.query(...)
+    ```
+    """
+
+    def inner(fn):
+        def wrapper(*args, **kwargs):
+            pool = instance()
+
+            with pool.connect() as conn:
+                (args, kwargs) = _inject_conn_arg(conn, arg, args, kwargs)
+                return fn(*args, **kwargs)
+
+        return wrapper
+
+    if _fn is None:
+        return inner
+    else:
+        return inner(_fn)

quasardb/pybind11/CMakeFiles/CMakeDirectoryInformation.cmake

@@ -0,0 +1,16 @@
+# CMAKE generated file: DO NOT EDIT!
+# Generated by "Unix Makefiles" Generator, CMake Version 4.0
+
+# Relative path conversion top directories.
+set(CMAKE_RELATIVE_PATH_TOP_SOURCE "/mnt/TeamCity/work/938b0bdf6727d1ad/thirdparty")
+set(CMAKE_RELATIVE_PATH_TOP_BINARY "/mnt/TeamCity/work/938b0bdf6727d1ad/build/lib.linux-aarch64-cpython-311/quasardb")
+
+# Force unix paths in dependencies.
+set(CMAKE_FORCE_UNIX_PATHS 1)
+
+
+# The C and CXX include file regular expressions for this directory.
+set(CMAKE_C_INCLUDE_REGEX_SCAN "^.*$")
+set(CMAKE_C_INCLUDE_REGEX_COMPLAIN "^$")
+set(CMAKE_CXX_INCLUDE_REGEX_SCAN ${CMAKE_C_INCLUDE_REGEX_SCAN})
+set(CMAKE_CXX_INCLUDE_REGEX_COMPLAIN ${CMAKE_C_INCLUDE_REGEX_COMPLAIN})

quasardb/pybind11/CMakeFiles/progress.marks

@@ -0,0 +1 @@
+0

quasardb/pybind11/Makefile

@@ -0,0 +1,189 @@
+# CMAKE generated file: DO NOT EDIT!
+# Generated by "Unix Makefiles" Generator, CMake Version 4.0
+
+# Default target executed when no arguments are given to make.
+default_target: all
+.PHONY : default_target
+
+# Allow only one "make -f Makefile2" at a time, but pass parallelism.
+.NOTPARALLEL:
+
+#=============================================================================
+# Special targets provided by cmake.
+
+# Disable implicit rules so canonical targets will work.
+.SUFFIXES:
+
+# Disable VCS-based implicit rules.
+% : %,v
+
+# Disable VCS-based implicit rules.
+% : RCS/%
+
+# Disable VCS-based implicit rules.
+% : RCS/%,v
+
+# Disable VCS-based implicit rules.
+% : SCCS/s.%
+
+# Disable VCS-based implicit rules.
+% : s.%
+
+.SUFFIXES: .hpux_make_needs_suffix_list
+
+# Command-line flag to silence nested $(MAKE).
+$(VERBOSE)MAKESILENT = -s
+
+#Suppress display of executed commands.
+$(VERBOSE).SILENT:
+
+# A target that is always out of date.
+cmake_force:
+.PHONY : cmake_force
+
+#=============================================================================
+# Set environment variables for the build.
+
+# The shell in which to execute make rules.
+SHELL = /bin/sh
+
+# The CMake executable.
+CMAKE_COMMAND = /mnt/TeamCity/temp/buildTmp/build-env-rhd3fmms/lib/python3.11/site-packages/cmake/data/bin/cmake
+
+# The command to remove a file.
+RM = /mnt/TeamCity/temp/buildTmp/build-env-rhd3fmms/lib/python3.11/site-packages/cmake/data/bin/cmake -E rm -f
+
+# Escaping for special characters.
+EQUALS = =
+
+# The top-level source directory on which CMake was run.
+CMAKE_SOURCE_DIR = /mnt/TeamCity/work/938b0bdf6727d1ad/quasardb
+
+# The top-level build directory on which CMake was run.
+CMAKE_BINARY_DIR = /mnt/TeamCity/work/938b0bdf6727d1ad/build/temp.linux-aarch64-cpython-311
+
+#=============================================================================
+# Targets provided globally by CMake.
+
+# Special rule for the target edit_cache
+edit_cache:
+	@$(CMAKE_COMMAND) -E cmake_echo_color "--switch=$(COLOR)" --cyan "No interactive CMake dialog available..."
+	/mnt/TeamCity/temp/buildTmp/build-env-rhd3fmms/lib/python3.11/site-packages/cmake/data/bin/cmake -E echo No\ interactive\ CMake\ dialog\ available.
+.PHONY : edit_cache
+
+# Special rule for the target edit_cache
+edit_cache/fast: edit_cache
+.PHONY : edit_cache/fast
+
+# Special rule for the target rebuild_cache
+rebuild_cache:
+	@$(CMAKE_COMMAND) -E cmake_echo_color "--switch=$(COLOR)" --cyan "Running CMake to regenerate build system..."
+	/mnt/TeamCity/temp/buildTmp/build-env-rhd3fmms/lib/python3.11/site-packages/cmake/data/bin/cmake --regenerate-during-build -S$(CMAKE_SOURCE_DIR) -B$(CMAKE_BINARY_DIR)
+.PHONY : rebuild_cache
+
+# Special rule for the target rebuild_cache
+rebuild_cache/fast: rebuild_cache
+.PHONY : rebuild_cache/fast
+
+# Special rule for the target list_install_components
+list_install_components:
+	@$(CMAKE_COMMAND) -E cmake_echo_color "--switch=$(COLOR)" --cyan "Available install components are: \"Unspecified\""
+.PHONY : list_install_components
+
+# Special rule for the target list_install_components
+list_install_components/fast: list_install_components
+.PHONY : list_install_components/fast
+
+# Special rule for the target install
+install: preinstall
+	@$(CMAKE_COMMAND) -E cmake_echo_color "--switch=$(COLOR)" --cyan "Install the project..."
+	/mnt/TeamCity/temp/buildTmp/build-env-rhd3fmms/lib/python3.11/site-packages/cmake/data/bin/cmake -P cmake_install.cmake
+.PHONY : install
+
+# Special rule for the target install
+install/fast: preinstall/fast
+	@$(CMAKE_COMMAND) -E cmake_echo_color "--switch=$(COLOR)" --cyan "Install the project..."
+	/mnt/TeamCity/temp/buildTmp/build-env-rhd3fmms/lib/python3.11/site-packages/cmake/data/bin/cmake -P cmake_install.cmake
+.PHONY : install/fast
+
+# Special rule for the target install/local
+install/local: preinstall
+	@$(CMAKE_COMMAND) -E cmake_echo_color "--switch=$(COLOR)" --cyan "Installing only the local directory..."
+	/mnt/TeamCity/temp/buildTmp/build-env-rhd3fmms/lib/python3.11/site-packages/cmake/data/bin/cmake -DCMAKE_INSTALL_LOCAL_ONLY=1 -P cmake_install.cmake
+.PHONY : install/local
+
+# Special rule for the target install/local
+install/local/fast: preinstall/fast
+	@$(CMAKE_COMMAND) -E cmake_echo_color "--switch=$(COLOR)" --cyan "Installing only the local directory..."
+	/mnt/TeamCity/temp/buildTmp/build-env-rhd3fmms/lib/python3.11/site-packages/cmake/data/bin/cmake -DCMAKE_INSTALL_LOCAL_ONLY=1 -P cmake_install.cmake
+.PHONY : install/local/fast
+
+# Special rule for the target install/strip
+install/strip: preinstall
+	@$(CMAKE_COMMAND) -E cmake_echo_color "--switch=$(COLOR)" --cyan "Installing the project stripped..."
+	/mnt/TeamCity/temp/buildTmp/build-env-rhd3fmms/lib/python3.11/site-packages/cmake/data/bin/cmake -DCMAKE_INSTALL_DO_STRIP=1 -P cmake_install.cmake
+.PHONY : install/strip
+
+# Special rule for the target install/strip
+install/strip/fast: preinstall/fast
+	@$(CMAKE_COMMAND) -E cmake_echo_color "--switch=$(COLOR)" --cyan "Installing the project stripped..."
+	/mnt/TeamCity/temp/buildTmp/build-env-rhd3fmms/lib/python3.11/site-packages/cmake/data/bin/cmake -DCMAKE_INSTALL_DO_STRIP=1 -P cmake_install.cmake
+.PHONY : install/strip/fast
+
+# The main all target
+all: cmake_check_build_system
+	cd /mnt/TeamCity/work/938b0bdf6727d1ad/build/temp.linux-aarch64-cpython-311 && $(CMAKE_COMMAND) -E cmake_progress_start /mnt/TeamCity/work/938b0bdf6727d1ad/build/temp.linux-aarch64-cpython-311/CMakeFiles /mnt/TeamCity/work/938b0bdf6727d1ad/build/lib.linux-aarch64-cpython-311/quasardb/pybind11//CMakeFiles/progress.marks
+	cd /mnt/TeamCity/work/938b0bdf6727d1ad/build/temp.linux-aarch64-cpython-311 && $(MAKE) $(MAKESILENT) -f CMakeFiles/Makefile2 /mnt/TeamCity/work/938b0bdf6727d1ad/build/lib.linux-aarch64-cpython-311/quasardb/pybind11/all
+	$(CMAKE_COMMAND) -E cmake_progress_start /mnt/TeamCity/work/938b0bdf6727d1ad/build/temp.linux-aarch64-cpython-311/CMakeFiles 0
+.PHONY : all
+
+# The main clean target
+clean:
+	cd /mnt/TeamCity/work/938b0bdf6727d1ad/build/temp.linux-aarch64-cpython-311 && $(MAKE) $(MAKESILENT) -f CMakeFiles/Makefile2 /mnt/TeamCity/work/938b0bdf6727d1ad/build/lib.linux-aarch64-cpython-311/quasardb/pybind11/clean
+.PHONY : clean
+
+# The main clean target
+clean/fast: clean
+.PHONY : clean/fast
+
+# Prepare targets for installation.
+preinstall: all
+	cd /mnt/TeamCity/work/938b0bdf6727d1ad/build/temp.linux-aarch64-cpython-311 && $(MAKE) $(MAKESILENT) -f CMakeFiles/Makefile2 /mnt/TeamCity/work/938b0bdf6727d1ad/build/lib.linux-aarch64-cpython-311/quasardb/pybind11/preinstall
+.PHONY : preinstall
+
+# Prepare targets for installation.
+preinstall/fast:
+	cd /mnt/TeamCity/work/938b0bdf6727d1ad/build/temp.linux-aarch64-cpython-311 && $(MAKE) $(MAKESILENT) -f CMakeFiles/Makefile2 /mnt/TeamCity/work/938b0bdf6727d1ad/build/lib.linux-aarch64-cpython-311/quasardb/pybind11/preinstall
+.PHONY : preinstall/fast
+
+# clear depends
+depend:
+	cd /mnt/TeamCity/work/938b0bdf6727d1ad/build/temp.linux-aarch64-cpython-311 && $(CMAKE_COMMAND) -S$(CMAKE_SOURCE_DIR) -B$(CMAKE_BINARY_DIR) --check-build-system CMakeFiles/Makefile.cmake 1
+.PHONY : depend
+
+# Help Target
+help:
+	@echo "The following are some of the valid targets for this Makefile:"
+	@echo "... all (the default if no target is provided)"
+	@echo "... clean"
+	@echo "... depend"
+	@echo "... edit_cache"
+	@echo "... install"
+	@echo "... install/local"
+	@echo "... install/strip"
+	@echo "... list_install_components"
+	@echo "... rebuild_cache"
+.PHONY : help
+
+
+
+#=============================================================================
+# Special targets to cleanup operation of make.
+
+# Special rule to run CMake to check the build system integrity.
+# No rule that depends on this can have commands that come from listfiles
+# because they might be regenerated.
+cmake_check_build_system:
+	cd /mnt/TeamCity/work/938b0bdf6727d1ad/build/temp.linux-aarch64-cpython-311 && $(CMAKE_COMMAND) -S$(CMAKE_SOURCE_DIR) -B$(CMAKE_BINARY_DIR) --check-build-system CMakeFiles/Makefile.cmake 0
+.PHONY : cmake_check_build_system
+

quasardb/pybind11/cmake_install.cmake

@@ -0,0 +1,50 @@
+# Install script for directory: /mnt/TeamCity/work/938b0bdf6727d1ad/thirdparty/pybind11
+
+# Set the install prefix
+if(NOT DEFINED CMAKE_INSTALL_PREFIX)
+  set(CMAKE_INSTALL_PREFIX "/usr/local")
+endif()
+string(REGEX REPLACE "/$" "" CMAKE_INSTALL_PREFIX "${CMAKE_INSTALL_PREFIX}")
+
+# Set the install configuration name.
+if(NOT DEFINED CMAKE_INSTALL_CONFIG_NAME)
+  if(BUILD_TYPE)
+    string(REGEX REPLACE "^[^A-Za-z0-9_]+" ""
+           CMAKE_INSTALL_CONFIG_NAME "${BUILD_TYPE}")
+  else()
+    set(CMAKE_INSTALL_CONFIG_NAME "Release")
+  endif()
+  message(STATUS "Install configuration: \"${CMAKE_INSTALL_CONFIG_NAME}\"")
+endif()
+
+# Set the component getting installed.
+if(NOT CMAKE_INSTALL_COMPONENT)
+  if(COMPONENT)
+    message(STATUS "Install component: \"${COMPONENT}\"")
+    set(CMAKE_INSTALL_COMPONENT "${COMPONENT}")
+  else()
+    set(CMAKE_INSTALL_COMPONENT)
+  endif()
+endif()
+
+# Install shared libraries without execute permission?
+if(NOT DEFINED CMAKE_INSTALL_SO_NO_EXE)
+  set(CMAKE_INSTALL_SO_NO_EXE "0")
+endif()
+
+# Is this installation the result of a crosscompile?
+if(NOT DEFINED CMAKE_CROSSCOMPILING)
+  set(CMAKE_CROSSCOMPILING "FALSE")
+endif()
+
+# Set path to fallback-tool for dependency-resolution.
+if(NOT DEFINED CMAKE_OBJDUMP)
+  set(CMAKE_OBJDUMP "/opt/rh/gcc-toolset-14/root/usr/bin/objdump")
+endif()
+
+string(REPLACE ";" "\n" CMAKE_INSTALL_MANIFEST_CONTENT
+       "${CMAKE_INSTALL_MANIFEST_FILES}")
+if(CMAKE_INSTALL_LOCAL_ONLY)
+  file(WRITE "/mnt/TeamCity/work/938b0bdf6727d1ad/build/lib.linux-aarch64-cpython-311/quasardb/pybind11/install_local_manifest.txt"
+     "${CMAKE_INSTALL_MANIFEST_CONTENT}")
+endif()

quasardb/range-v3/CMakeFiles/CMakeDirectoryInformation.cmake

@@ -0,0 +1,16 @@
+# CMAKE generated file: DO NOT EDIT!
+# Generated by "Unix Makefiles" Generator, CMake Version 4.0
+
+# Relative path conversion top directories.
+set(CMAKE_RELATIVE_PATH_TOP_SOURCE "/mnt/TeamCity/work/938b0bdf6727d1ad/thirdparty")
+set(CMAKE_RELATIVE_PATH_TOP_BINARY "/mnt/TeamCity/work/938b0bdf6727d1ad/build/lib.linux-aarch64-cpython-311/quasardb")
+
+# Force unix paths in dependencies.
+set(CMAKE_FORCE_UNIX_PATHS 1)
+
+
+# The C and CXX include file regular expressions for this directory.
+set(CMAKE_C_INCLUDE_REGEX_SCAN "^.*$")
+set(CMAKE_C_INCLUDE_REGEX_COMPLAIN "^$")
+set(CMAKE_CXX_INCLUDE_REGEX_SCAN ${CMAKE_C_INCLUDE_REGEX_SCAN})
+set(CMAKE_CXX_INCLUDE_REGEX_COMPLAIN ${CMAKE_C_INCLUDE_REGEX_COMPLAIN})