madsim_tokio_postgres/
cancel_query_raw.rs

1use crate::config::SslMode;
2use crate::tls::TlsConnect;
3use crate::{connect_tls, Error};
4use bytes::BytesMut;
5use postgres_protocol::message::frontend;
6use tokio::io::{AsyncRead, AsyncWrite, AsyncWriteExt};
7
8pub async fn cancel_query_raw<S, T>(
9    stream: S,
10    mode: SslMode,
11    tls: T,
12    process_id: i32,
13    secret_key: i32,
14) -> Result<(), Error>
15where
16    S: AsyncRead + AsyncWrite + Unpin,
17    T: TlsConnect<S>,
18{
19    let mut stream = connect_tls::connect_tls(stream, mode, tls).await?;
20
21    let mut buf = BytesMut::new();
22    frontend::cancel_request(process_id, secret_key, &mut buf);
23
24    stream.write_all(&buf).await.map_err(Error::io)?;
25    stream.flush().await.map_err(Error::io)?;
26    stream.shutdown().await.map_err(Error::io)?;
27
28    Ok(())
29}