Commit720897d

committed

feat: add support for quick active health checks on connections

1 parent8488a46 commit720897dCopy full SHA for 720897d

File tree

+22

-0

lines changed

+22

-0

lines changed

Lines changed: 6 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -531,6 +531,12 @@ impl Client {`
`531`	`531`	`simple_query::batch_execute(self.inner(), query).await`
`532`	`532`	`}`
`533`	`533`
	`534`	`+/// Check the connection is alive and wait for the confirmation.`
	`535`	`+pubasyncfncheck_connection(&self) ->Result<(),Error>{`
	`536`	`+// sync is a very quick message to test the connection health.`
	`537`	`+ query::sync(self.inner()).await`
	`538`	`+}`
	`539`	`+`
`534`	`540`	`/// Begins a new database transaction.`
`535`	`541`	`///`
`536`	`542`	/// The transaction will roll back by default - use the `commit` method to commit it.

Lines changed: 10 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -323,3 +323,13 @@ impl RowStream {`
`323`	`323`	`self.rows_affected`
`324`	`324`	`}`
`325`	`325`	`}`
	`326`	`+`
	`327`	`+pubasyncfnsync(client:&InnerClient) ->Result<(),Error>{`
	`328`	`+let buf =Bytes::from_static(b"S\0\0\0\x04");`
	`329`	`+letmut responses = client.send(RequestMessages::Single(FrontendMessage::Raw(buf)))?;`
	`330`	`+`
	`331`	`+match responses.next().await?{`
	`332`	`+Message::ReadyForQuery(_) =>Ok(()),`
	`333`	`+ _ =>Err(Error::unexpected_message()),`
	`334`	`+}`
	`335`	`+}`

Lines changed: 6 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -147,6 +147,12 @@ async fn scram_password_ok() {`
`147`	`147`	`connect("user=scram_user password=password dbname=postgres").await;`
`148`	`148`	`}`
`149`	`149`
	`150`	`+#[tokio::test]`
	`151`	`+asyncfnsync(){`
	`152`	`+let client =connect("user=postgres").await;`
	`153`	`+ client.check_connection().await.unwrap();`
	`154`	`+}`
	`155`	`+`
`150`	`156`	`#[tokio::test]`
`151`	`157`	`asyncfnpipelined_prepare(){`
`152`	`158`	`let client =connect("user=postgres").await;`

Comments

(0)