use crate::diesel::*;
use crate::schema::*;
use diesel::QueryResult;
use serde::{Deserialize, Serialize};
type Connection = diesel::r2d2::PooledConnection<diesel::r2d2::ConnectionManager<diesel::PgConnection>>;
#[derive(Debug, Serialize, Deserialize, Clone, Queryable, Insertable, AsChangeset, Selectable)]
#[diesel(table_name=users, primary_key(name,address))]
pub struct User {
pub name: String,
pub address: String,
pub secret: String,
}
#[derive(Debug, Serialize, Deserialize, Clone, Queryable, Insertable, AsChangeset)]
#[diesel(table_name=users)]
pub struct CreateUser {
pub name: String,
pub address: String,
pub secret: String,
}
#[derive(Debug, Serialize, Deserialize, Clone, Queryable, Insertable, AsChangeset)]
#[diesel(table_name=users)]
pub struct UpdateUser {
pub secret: Option<String>,
}
#[derive(Debug, Serialize)]
pub struct PaginationResult<T> {
pub items: Vec<T>,
pub total_items: i64,
pub page: i64,
pub page_size: i64,
pub num_pages: i64,
}
impl User {
pub fn create(db: &mut Connection, item: &CreateUser) -> QueryResult<Self> {
use crate::schema::users::dsl::*;
insert_into(users).values(item).get_result::<Self>(db)
}
pub fn read(db: &mut Connection, param_name: String, param_address: String) -> QueryResult<Self> {
use crate::schema::users::dsl::*;
users.filter(name.eq(param_name)).filter(address.eq(param_address)).first::<Self>(db)
}
pub fn paginate(db: &mut Connection, page: i64, page_size: i64) -> QueryResult<PaginationResult<Self>> {
use crate::schema::users::dsl::*;
let page_size = if page_size < 1 { 1 } else { page_size };
let total_items = users.count().get_result(db)?;
let items = users.limit(page_size).offset(page * page_size).load::<Self>(db)?;
Ok(PaginationResult {
items,
total_items,
page,
page_size,
num_pages: total_items / page_size + i64::from(total_items % page_size != 0)
})
}
pub fn update(db: &mut Connection, param_name: String, param_address: String, item: &UpdateUser) -> QueryResult<Self> {
use crate::schema::users::dsl::*;
diesel::update(users.filter(name.eq(param_name)).filter(address.eq(param_address))).set(item).get_result(db)
}
pub fn delete(db: &mut Connection, param_name: String, param_address: String) -> QueryResult<usize> {
use crate::schema::users::dsl::*;
diesel::delete(users.filter(name.eq(param_name)).filter(address.eq(param_address))).execute(db)
}
}