pgw 0.2.17

auto reconnect tokio postgres
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

pub use tokio_postgres::{self, Error, Statement, ToStatement};

use crate::Sql;

pub trait IntoStatement<T: ToStatement> {
  fn into(self) -> impl std::future::Future<Output = Result<T, Error>>;
}

impl<T: ToStatement> IntoStatement<T> for T {
  async fn into(self) -> Result<T, Error> {
    Ok(self)
  }
}

pub async fn into_statement(sql: &Sql) -> Result<Statement, Error> {
  let sql = &sql.0;
  loop {
    if let Some(st) = sql.st.read().await.as_ref() {
      return Ok(st.clone());
    }

    let st = sql.pg.prepare(&*sql.sql).await?;
    *sql.st.write().await = Some(st);
  }
}

impl IntoStatement<Statement> for Sql {
  async fn into(self) -> Result<Statement, Error> {
    into_statement(&self).await
  }
}