Struct PostgresSaver 

pub struct PostgresSaver { /* private fields */ }

Async Postgres checkpoint saver using sqlx.

Implementations

impl PostgresSaver

pub fn new(pool: PgPool) -> Self

Create a new PostgresSaver from a connection pool.

pub fn with_serde(pool: PgPool, serde: Arc<dyn SerializerProtocol>) -> Self

Create a new PostgresSaver with a custom serializer.

pub async fn from_conn_string( conn_string: &str, ) -> Result<Self, CheckpointError>

Create a PostgresSaver from a connection string.

pub async fn setup(&self) -> Result<(), CheckpointError>

Run migrations to set up the checkpoint schema.

pub fn pool(&self) -> &PgPool

Get the underlying connection pool.

impl PostgresSaver

Async list method for PostgresSaver.

pub async fn alist( &self, config: Option<&RunnableConfig>, filter: Option<&HashMap<String, JsonValue>>, before: Option<&RunnableConfig>, limit: Option<usize>, ) -> Result<Vec<CheckpointTuple>, CheckpointError>

Trait Implementations

impl BaseCheckpointSaver for PostgresSaver

fn get_tuple( &self, config: &RunnableConfig, ) -> Result<Option<CheckpointTuple>, CheckpointError>

Get a checkpoint tuple by config.

fn list( &self, config: Option<&RunnableConfig>, filter: Option<&HashMap<String, JsonValue>>, before: Option<&RunnableConfig>, limit: Option<usize>, ) -> Result<Vec<CheckpointTuple>, CheckpointError>

List checkpoint tuples.

fn put( &self, config: &RunnableConfig, checkpoint: &Checkpoint, metadata: &CheckpointMetadata, new_versions: &ChannelVersions, ) -> Result<RunnableConfig, CheckpointError>

Store a checkpoint.

fn put_writes( &self, config: &RunnableConfig, writes: &[(String, String, JsonValue)], task_id: &str, task_path: &str, ) -> Result<(), CheckpointError>

Store pending writes for a checkpoint.

fn delete_thread(&self, thread_id: &str) -> Result<(), CheckpointError>

Delete all checkpoints for a thread.

fn aget_tuple<'life0, 'life1, 'async_trait>( &'life0 self, config: &'life1 RunnableConfig, ) -> Pin<Box<dyn Future<Output = Result<Option<CheckpointTuple>, CheckpointError>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

fn aput<'life0, 'life1, 'life2, 'life3, 'life4, 'async_trait>( &'life0 self, config: &'life1 RunnableConfig, checkpoint: &'life2 Checkpoint, metadata: &'life3 CheckpointMetadata, new_versions: &'life4 ChannelVersions, ) -> Pin<Box<dyn Future<Output = Result<RunnableConfig, CheckpointError>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait, 'life3: 'async_trait, 'life4: 'async_trait,

fn aput_writes<'life0, 'life1, 'async_trait>( &'life0 self, config: &'life1 RunnableConfig, writes: Vec<(String, String, JsonValue)>, task_id: String, task_path: String, ) -> Pin<Box<dyn Future<Output = Result<(), CheckpointError>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

fn adelete_thread<'life0, 'async_trait>( &'life0 self, thread_id: String, ) -> Pin<Box<dyn Future<Output = Result<(), CheckpointError>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

fn get_next_version(&self, current: Option<&Value>) -> Value

Get the next version for a channel.