Skip to main content

A distributed lock implementation based on SQLAlchemy

Project description

SQLAlchemy-DLock

GitHub GitHub tag (latest by date) Python package PyPI PyPI - Status PyPI - License PyPI - Python Version Documentation Status codecov

Distributed lock based on Database and SQLAlchemy.

It currently supports below locks:

Note:

The project is not stable enough and DO NOT use it in production.

Usages

  • Work with SQLAlchemy's Connection object:

    from sqlalchemy import create_engine
    from sqlalchemy_dlock import create_sadlock
    
    key = 'user/001'
    
    engine = create_engine('postgresql://scott:tiger@localhost/')
    conn = engine.connect()
    
    # Create the D-Lock on the connection
    lock = create_sadlock(conn, key)
    
    # it's not lock when constructed
    assert not lock.acquired
    
    # lock
    lock.acquire()
    assert lock.acquired
    
    # un-lock
    lock.release()
    assert not lock.acquired
    
  • Use in with statement

    from contextlib import closing
    
    from sqlalchemy import create_engine
    from sqlalchemy_dlock import create_sadlock
    
    key = 'user/001'
    
    engine = create_engine('postgresql://scott:tiger@localhost/')
    with engine.connect() as conn:
    
        # Create the D-Lock on the connection
        with create_sadlock(conn, key) as lock:
            # It's locked
            assert lock.acquired
    
        # Auto un-locked
        assert not lock.acquired
    
        # If do not want to be locked in `with`, a `closing` wrapper may help
        with closing(create_sadlock(conn, key)) as lock2:
            # It's NOT locked here
            assert not lock2.acquired
            # lock it now:
            lock2.acquire()
            assert lock2.acquired
    
        # Auto un-locked
        assert not lock2.acquired
    
  • Work with SQLAlchemy's ORM session:

    Note:

    According to https://docs.sqlalchemy.org/14/orm/extensions/asyncio.html:

    • The asyncio extension as of SQLAlchemy 1.4.3 can now be considered to be beta level software.
    • The asyncio extension requires at least Python version 3.6
    from sqlalchemy import create_engine
    from sqlalchemy.orm import sessionmaker
    from sqlalchemy_dlock import create_sadlock
    
    key = 'user/001'
    
    engine = create_engine('postgresql://scott:tiger@localhost/')
    Session = sessionmaker(bind=engine)
    
    with Session() as session:
      with create_sadlock(session, key) as lock:
          assert lock.acquired
      assert not lock.acquired
    
  • Work asynchronously

    from sqlalchemy.ext.asyncio import create_async_engine
    from sqlalchemy_dlock.asyncio import create_async_sadlock
    
    key = 'user/001'
    
    engine = create_async_engine('postgresql+asyncpg://scott:tiger@localhost/')
    
    async with engine.begin() as conn:
        async with create_async_sadlock(conn, key) as lock:
            assert lock.locked
            await lock.release()
            assert not lock.locked
            await lock.acquire()
        assert not lock.locked
    

Tests

Just up the docker-compose in tests directory:

(cd tests; docker-compose up --abort-on-container-exit --exit-code-from pytest; docker-compose down)

CHANGELOG

v0.2b2/b3

Date: 2021-03-23

  • Add:
    • More unit tests
    • Optimized CI

v0.2b1

Date: 2021-03-16

  • Add:

    • New unit tests
    • CI by github workflows

v0.2a3

Date: 2021-03-14

  • Change:

    • Drop Python 3.5 support.
    • Remove SQLAlchemy version requires earlier than 1.4 in setup, it's not supported actually.
    • Adjust PostgreSQL lock's constructor arguments order
  • Add:

    • More test cases, and add test/deploy workflow in github actions.
    • Add docker-compose test scripts

v0.2a2

Date: 2021-03-09

  • Change:

    • Rename a lot of function/class:

      • sadlock -> create_sadlock
      • asyncio.sadlock -> asyncio.create_async_sadlock

      and some other ...

v0.2a1

Date: 2021-03-08

v0.1.2

Date: 2021-01-26

Still an early version, not for production.

  • Changes:
    • Arguments and it's default value of acquire now similar to stdlib's multiprossing.Lock, instead of Threading.Lock
    • MySQL lock now accepts float-point value as timeout
  • Adds
    • Several new test cases
  • Other
    • Many other small adjustment

v0.1.1

  • A very early version, maybe not stable enough.
  • Replace black2b with crc64-iso in PostgreSQL key convert function
  • Only named arguments as extra parameters allowed in Lock's implementation class

AUTHORS

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

sqlalchemy-dlock-0.2b3.post4.tar.gz (15.1 kB view hashes)

Uploaded Source

Built Distribution

sqlalchemy_dlock-0.2b3.post4-py3-none-any.whl (17.6 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page