Method redis_client

langroid/cachedb/redis_cachedb.py:65–74

Cleanly open and close a redis client, avoids max clients exceeded error

(self)

    @contextmanager  # type: ignore
    def redis_client(self) -> AbstractContextManager[T]:  # type: ignore
        """Cleanly open and close a redis client, avoids max clients exceeded error"""
        if isinstance(self.pool, fakeredis.FakeStrictRedis):
            yield self.pool
        else:
            client: T = redis.Redis(connection_pool=self.pool)
            try:
                yield client
            finally:
                client.close()

    def close_all_connections(self) -> None:
        with self.redis_client() as client:  # type: ignore