# DLock
A lease based locking implementations for distributed clients with support for a fencing token to prevent usage of stale locks.
[![Crates.io][crates-badge]][crates-url]
[![MIT licensed][mit-badge]][mit-url]
[![Build Status][actions-badge]][actions-url]
[crates-badge]: https://img.shields.io/crates/v/dlock
[crates-url]: https://crates.io/crates/dlock
[mit-badge]: https://img.shields.io/badge/license-MIT-blue.svg
[mit-url]: https://github.com/kodebooth/dlink/blob/main/LICENSE
[actions-badge]: https://github.com/kodebooth/dlock/workflows/CI/badge.svg
[actions-url]: https://github.com/kodebooth/dlock/actions?query=workflow%3ACI+branch%3Amain
# Overview
`DLock` is a lease based distributed lock implementation. It supports recovering from a dead lease holder as well as provides a fencing token to prevent usage of stale tokens.
The implementation has a generic frontend called `DLock` which accepts a provider backend to support the actual locking implementation. Currently it only provides a AWS DynamoDB provider backend.
## Algorithm
The following diagram shows a basic lock contention where Client A has already acquired the lock when Client B attempts to get the lock.
```mermaid
sequenceDiagram
participant A as Client A
participant P as Provider
participant B as Client B
activate A
A ->> P: Acquire (prev_uuid=None, new_uuid=A1, duration=D1)
P ->> A: Acquired (uuid=A1, token=0)
A ->> A: Do synchronized work
activate B
loop
B ->> P: Acquire (prev_uuid=None, new_uuid=B1, duration=D2)
P ->> B: Failure (uuid=A1, duration=D1)
end
A ->> P: Release (uuid=A1)
P ->> A: Released (uuid=A1)
deactivate A
B ->> P: Acquire (prev_uuid=None, new_uuid=B1, duration=D2)
P ->> B: Acquired (uuid=B1, duration=D2, token=0)
B ->> B: Do synchronized work
B ->> P: Release (uuid=B1)
P ->> B: Released (uuid=B1)
deactivate B
```
The following diagram shows a dead lease recovery scenario where Client A has died after acquiring the lock.
```mermaid
sequenceDiagram
participant A as Client A
participant P as Provider
participant B as Client B
activate A
A ->> P: Acquire (prev_uuid=None, new_uuid=A1, duration=D1)
P ->> A: Acquired (uuid=A1, token=0)
A ->> A: Crash!
deactivate A
activate B
B ->> P: Acquire (prev_uuid=None, new_uuid=B1, duration=D2)
P ->> B: Failure (uuid=A1, duration=D1)
loop D1
B ->> P: Acquire (prev_uuid=None, new_uuid=B1, duration=D2)
P ->> B: Failure (uuid=A1, duration=D1)
end
B ->> P: Acquire (prev_uuid=A1, new_uuid=B1, duration=D2)
P ->> B: Acquired (uuid=B1, duration=D2, token=1)
B ->> B: Synchronized Work
B ->> P: Release (uuid=B1)
P ->> B: Released (uuid=B1)
deactivate B
```
The following diagram shows a paused lease holder scenario where Client A is interrupted for a long time and the fencing token is used to ensure correct behavior.
```mermaid
sequenceDiagram
participant A as Client A
participant P as Provider
participant S as Storage
participant B as Client B
activate A
A ->> P: Acquire (prev_uuid=None, new_uuid=A1, duration=D1)
P ->> A: Acquired (uuid=A1, token=0)
activate B
B ->> P: Acquire (prev_uuid=None, new_uuid=B1, duration=D2)
P ->> B: Failure (uuid=A1, duration=D1)
loop D1
B ->> P: Acquire (prev_uuid=None, new_uuid=B1, duration=D2)
P ->> B: Failure (uuid=A1, duration=D1)
end
B ->> P: Acquire (prev_uuid=A1, new_uuid=B1, duration=D2)
P ->> B: Acquired (uuid=B1, duration=D2, token=1)
B ->> S: Synchronized Work (token=1)
S ->> B: Success (token=1)
B ->> P: Release (uuid=B1)
P ->> B: Released (uuid=B1)
deactivate B
A ->> S: Synchronized Work (token=0)
S ->> S: Token Invalid 0 < 1
S ->> A: Failure (token=0)
A ->> P: Release (uuid=A1)
P ->> A: Already Released ()
deactivate A
```
# Usage
Add `DLock` to your crate:
```toml
[dependencies]
dlock = "0.1"
```
Create a provider for the lock backend:
```rust
let client = ... // Create a aws_sdk_dynamodb::Client based on your project configuration
let provider = DynamodbProvider::builder()
.client(Arc::new(client))
.table_name("dynamodb_table".to_string())
.build();
```
You can use the automatic lease renewal mechanism built into `DLock` or manually manage the lease yourself:
## Automatic
`with` will acquire, automatically renew, and release the lease. For more information see the function documentation.
```rust
let lock = DLock::builder()
.name("my_lock".to_string())
.owner("me".to_string())
.duration(Duration::from_secs(1))
.provider(provider)
.build();
).await?;
```
## Manually
You are responsible for acquiring, renewing, and releasing the lease.
```rust
let lock = DLock::builder()
.name("my_lock".to_string())
.owner("me".to_string())
.duration(Duration::from_secs(10))
.provider(provider)
.build();
let lease = lock.acquire().await?;
// Synchronized Work
lease.release().await?;
```