#[derive(Debug, Clone, Copy)]
pub struct Threads;
#[derive(Debug, Clone, PartialEq)]
pub struct Thread {
pub id: i32,
pub title: String,
pub first_message_id: Option<i32>,
pub rel: Rel,
}
#[derive(Debug, Clone, PartialEq, Default)]
pub struct Rel {
pub first_message: Option<Box<super::messages::Message>>,
pub messages: Vec<super::messages::Message>,
}
impl keelson_exec::FromRow for Thread {
fn from_row(row: &mut keelson_exec::Row) -> Result<Self, keelson_exec::ExecError> {
Ok(Thread {
id: row.take("id")?,
title: row.take("title")?,
first_message_id: row.take("first_message_id")?,
rel: Rel::default(),
})
}
}
#[derive(Debug, Clone, Default)]
pub struct Setter {
pub id: keelson_models::Set<i32>,
pub title: keelson_models::Set<String>,
pub first_message_id: keelson_models::Set<i32>,
}
pub fn table() -> Threads {
Threads
}
pub fn id() -> keelson_models::Column<i32> {
keelson_models::Column::new("threads", "id")
}
pub fn title() -> keelson_models::Column<String> {
keelson_models::Column::new("threads", "title")
}
pub fn first_message_id() -> keelson_models::Column<i32> {
keelson_models::Column::new("threads", "first_message_id")
}
#[allow(clippy::type_complexity)]
fn all_columns() -> (
keelson_models::Column<i32>,
keelson_models::Column<String>,
keelson_models::Column<i32>,
) {
(id(), title(), first_message_id())
}
impl keelson_models::View for Threads {
type Row = Thread;
type Select = keelson_mysql::SelectQuery;
fn base_select() -> Self::Select {
keelson_mysql::select((
keelson_mysql::select::columns(all_columns()),
keelson_mysql::select::from(keelson_mysql::quote("threads")),
))
}
}
impl keelson_models::Table for Threads {
type Pk = i32;
type Setter = Setter;
type Insert = keelson_mysql::InsertQuery;
type Update = keelson_mysql::UpdateQuery;
type Delete = keelson_mysql::DeleteQuery;
fn insert_query(s: Setter) -> Self::Insert {
let mut cols: Vec<&'static str> = Vec::new();
let mut vals: Vec<keelson_core::expr::Expr> = Vec::new();
s.id.push_into("id", &mut cols, &mut vals);
s.title.push_into("title", &mut cols, &mut vals);
s.first_message_id.push_into("first_message_id", &mut cols, &mut vals);
let mut q = keelson_mysql::insert(
keelson_mysql::insert::into(keelson_mysql::quote("threads")).columns(cols),
);
if !vals.is_empty() {
q.apply(keelson_mysql::insert::values(vals));
}
q
}
fn update_query() -> Self::Update {
keelson_mysql::update(
keelson_mysql::update::table(keelson_mysql::quote("threads")),
)
}
fn apply_setter(s: Setter, q: &mut Self::Update) {
if let Some(v) = s.id.into_expr() {
q.apply(keelson_mysql::update::set_col("id").to(v));
}
if let Some(v) = s.title.into_expr() {
q.apply(keelson_mysql::update::set_col("title").to(v));
}
if let Some(v) = s.first_message_id.into_expr() {
q.apply(keelson_mysql::update::set_col("first_message_id").to(v));
}
}
fn delete_query() -> Self::Delete {
keelson_mysql::delete(
keelson_mysql::delete::from(keelson_mysql::quote("threads")),
)
}
fn pk(row: &Thread) -> Self::Pk {
row.id
}
}
impl Threads {
pub fn query(
self,
mods: impl keelson_core::Mod<keelson_models::ModelSelect<Threads>>,
) -> keelson_models::ModelSelect<Threads> {
keelson_models::ModelTable::<Threads>::new().query(mods)
}
pub fn insert(self, setter: Setter) -> Insert {
Insert { setter, mods: Vec::new() }
}
pub fn update(
self,
setter: Setter,
mods: impl keelson_core::Mod<keelson_models::ModelUpdate<Threads>>,
) -> Update {
Update(keelson_models::ModelTable::<Threads>::new().update(setter, mods))
}
pub fn delete(
self,
mods: impl keelson_core::Mod<keelson_models::ModelDelete<Threads>>,
) -> Delete {
Delete(keelson_models::ModelTable::<Threads>::new().delete(mods))
}
}
pub struct Insert {
setter: Setter,
#[allow(clippy::type_complexity)]
mods: Vec<Box<dyn FnOnce(&mut keelson_mysql::InsertQuery) + Send>>,
}
impl std::fmt::Debug for Insert {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
f.debug_struct("Insert")
.field("setter", &self.setter)
.field("mods", &self.mods.len())
.finish()
}
}
impl Insert {
#[must_use]
pub fn with(
mut self,
mods: impl keelson_core::Mod<keelson_mysql::InsertQuery> + Send + 'static,
) -> Self {
self.mods.push(Box::new(move |q| mods.apply(q)));
self
}
pub async fn one(
self,
db: &dyn keelson_exec::Executor,
) -> Result<Thread, keelson_exec::ExecError> {
use keelson_exec::Execute as _;
let Insert { mut setter, mods } = self;
<Threads as keelson_models::Table>::before_insert(db, &mut setter).await?;
let k0 = match &setter.id {
keelson_models::Set::Value(v) => Some(*v),
_ => None,
};
let mut q = <Threads as keelson_models::Table>::insert_query(setter);
for m in mods {
m(&mut q);
}
let done = q.execute(db).await?;
let k0: i32 = k0
.or_else(|| {
done
.last_insert_id
.and_then(|id| {
<i32 as std::convert::TryFrom<i64>>::try_from(id).ok()
})
})
.ok_or_else(|| keelson_exec::ExecError::other(
"threads: the INSERT set no `id` and MySQL reported no last_insert_id, so the inserted row cannot be read back",
))?;
let row: Thread = by_pk(k0).fetch_one(db).await?;
<Threads as keelson_models::Table>::after_insert(db, std::slice::from_ref(&row))
.await?;
Ok(row)
}
pub async fn exec(
self,
db: &dyn keelson_exec::Executor,
) -> Result<keelson_exec::ExecResult, keelson_exec::ExecError> {
use keelson_exec::Execute as _;
let Insert { mut setter, mods } = self;
<Threads as keelson_models::Table>::before_insert(db, &mut setter).await?;
let mut q = <Threads as keelson_models::Table>::insert_query(setter);
for m in mods {
m(&mut q);
}
let done = q.execute(db).await?;
<Threads as keelson_models::Table>::after_insert(db, &[]).await?;
Ok(done)
}
}
pub fn by_pk(k0: i32) -> keelson_mysql::SelectQuery {
keelson_mysql::select((
keelson_mysql::select::columns(all_columns()),
keelson_mysql::select::from(keelson_mysql::quote("threads")),
id().eq(k0),
))
}
#[derive(Debug)]
pub struct Update(keelson_models::ModelUpdate<Threads>);
impl Update {
pub fn apply(&mut self, mods: impl keelson_core::Mod<keelson_mysql::UpdateQuery>) {
self.0.apply(mods);
}
pub async fn exec(
self,
db: &dyn keelson_exec::Executor,
) -> Result<keelson_exec::ExecResult, keelson_exec::ExecError> {
self.0.exec(db).await
}
}
#[derive(Debug)]
pub struct Delete(keelson_models::ModelDelete<Threads>);
impl Delete {
pub fn apply(&mut self, mods: impl keelson_core::Mod<keelson_mysql::DeleteQuery>) {
self.0.apply(mods);
}
pub async fn exec(
self,
db: &dyn keelson_exec::Executor,
) -> Result<keelson_exec::ExecResult, keelson_exec::ExecError> {
self.0.exec(db).await
}
}
pub mod preload {
pub fn first_message() -> impl keelson_core::Mod<
keelson_models::ModelSelect<super::Threads>,
> {
keelson_core::mod_fn(|q: &mut keelson_models::ModelSelect<super::Threads>| {
use keelson_mysql::Chain as _;
use keelson_mysql::Mod as _;
(
keelson_mysql::select::left_join(keelson_mysql::quote("messages"))
.on(
keelson_mysql::quote(("messages", "id"))
.eq(keelson_mysql::quote(("threads", "first_message_id"))),
),
keelson_mysql::select::preload_columns((
keelson_mysql::quote(("messages", "id")).as_("first_message.id"),
keelson_mysql::quote(("messages", "thread_id"))
.as_("first_message.thread_id"),
keelson_mysql::quote(("messages", "body")).as_("first_message.body"),
)),
)
.apply(q);
q.add_mapper_mod(
keelson_models::mapper_mod(|row, parent: &mut super::Thread| {
parent.rel.first_message = first_message_from_preload(row)?
.map(Box::new);
Ok(())
}),
);
})
}
pub fn first_message_from_preload(
row: &mut keelson_exec::Row,
) -> Result<Option<super::super::messages::Message>, keelson_exec::ExecError> {
if matches!(
row.value("first_message.id"), None | Some(keelson_mysql::Value::Null)
) {
return Ok(None);
}
Ok(
Some(super::super::messages::Message {
id: row.take("first_message.id")?,
thread_id: row.take("first_message.thread_id")?,
body: row.take("first_message.body")?,
rel: super::super::messages::Rel::default(),
}),
)
}
}
pub mod then_load {
pub fn first_message() -> keelson_models::ThenLoad<
super::Threads,
super::super::messages::Messages,
i32,
> {
keelson_models::ThenLoad::new(
|rows: &[super::Thread]| {
rows.iter().filter_map(|r| r.first_message_id).collect()
},
|keys, q| keelson_core::Mod::apply(
super::super::messages::id().in_(keys),
q,
),
|rows: &mut [super::Thread], related| {
keelson_models::attach_to_one(
rows,
related,
|r| r.first_message_id,
|c| Some(c.id),
|r, c| {
r.rel.first_message = c.map(Box::new);
},
);
},
)
}
pub fn messages() -> keelson_models::ThenLoad<
super::Threads,
super::super::messages::Messages,
i32,
> {
keelson_models::ThenLoad::new(
|rows: &[super::Thread]| rows.iter().map(|r| r.id).collect(),
|keys, q| keelson_core::Mod::apply(
super::super::messages::thread_id().in_(keys),
q,
),
|rows: &mut [super::Thread], related| {
keelson_models::attach_to_many(
rows,
related,
|r| r.id,
|c| c.thread_id,
|r, cs| {
r.rel.messages = cs;
},
);
},
)
}
}