Mini Query

A mini ORM to quickly find / insert / retrieve records with tokio-postgres.

Generates the following functions on the struct:

If #[mini_query(primary_key)] is set:

• MyStruct::get(id: &T) -> Result<Option<T>>

For all fields marked with #[mini_query(find_by)]:

• MyStruct::find_by_{x}(client: &impl GenericClient, val: &T) -> Result<Option<T>>

For all fields marked with #[mini_query(get_by)]:

• MyStruct::get_by_{x}(client: &impl GenericClient, val: &T) -> Result<Vec<T>>

And of course, support for inserting and updating.

• MyStruct::quick_insert(&self, client: &impl GenericClient) -> Result<Self>
• MyStruct::quick_insert_no_return(&self, client: &impl GenericClient) -> Result<()>
• MyStruct::quick_update(&self, client: &impl GenericClient) -> Result<Self>

This macro also implements the From<Row> trait for your struct. Making this possible:

  let user: User = client.query_one("SELECT * FROM users WHERE id = $1", &[&1]).await?.into();

Here's an example for a "users" table

use mini_query::MiniQuery;
use tokio_postgres::{Row, GenericClient};
use anyhow::Result;

#[derive(MiniQuery, Default)]
#[mini_query(table_name = "users")]
struct User {
  #[mini_query(primary_key)]
  pub id: i32,
  #[mini_query(find_by)]
  pub email: String,
  #[mini_query(skip)]
  pub raw_password: Option<String>,
  #[mini_query(rename = "password")] // renames field to "password" when saving
  enc_password: String,
  #[mini_query(cast = i16, get_by)] // this column is represented by a smallint in postgres
  pub role: UserRole
  pub created_at: DateTime<Utc>,
  pub updated_at: DateTime<Utc>,
}
impl User {
  fn encrypt_password(&mut self) {
    let Some(raw_password) = &self.raw_password else {
      return;
    }
    self.enc_password = format!("{raw_password} - tada, I am encrypted");
  }
}

#[derive(Default)]
#[repr(i16)]
enum UserRole {
  #[default]
  User = 0,
  Admin = 1
}
impl From<i16> for UserRole {
  fn from(val: i16) -> Self {
    match val {
      0 => UserRole::User,
      1 => UserRole::Admin,
      _ => unimplemented!(),
    }
  }
}

#[tokio::main]
async fn main() -> Result<()> {
  let (client, connection) =
    tokio_postgres::connect("postgresql://postgres@localhost/mydb-dev", NoTls).await?;
  tokio::spawn(async move { connection.await });

  let mut user = User {
    email: "foo@dog.com".to_owned(),
    raw_password: "I am bad password".to_owned(),
    role: UserRole::Admin,
    ..Default::default()
  };
  user.encrypt_password();

  // fn is prefixed with "quick_" to avoid naming collisions, in case you wish to write a validation wrapper.
  user.quick_insert(&client).await?;

  // find user by email
  let same_user = User::find_by_email(&client, "foo@dog.com")?;
  assert_eq!(user.email, same_user.email);

  // get all the admins
  let admins = User::get_by_role(&client, &UserRole::Admin);
  assert_eq!(vec![same_user], admins);

  // get user by id and update
  let mut user = User::get(&client, &same_user.id);
  user.email = "bar@dog.com".to_owned();
  user.quick_update(&client).await?;

  // assert it saved
  assert_eq!(&User::get(&client, &user.id).email, "bar@dog.com");

  Ok(())
}

Only supports tokio-postgres right now.