caisin/sea-orm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Tweak rocket example

Browse Source
This commit is contained in:
Chris Tsang 2021-09-06 23:09:01 +08:00
parent a5b230f67c
commit 4a701d5e2d
1 changed files with 13 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
examples/rocket_example/src/main.rs
View File

@ -11,7 +11,7 @@ use rocket_db_pools::{sqlx, Connection, Database};
use rocket_dyn_templates::{context, Template}; use rocket_dyn_templates::{context, Template};
use sea_orm::entity::*; use sea_orm::entity::*;
use sea_orm::QueryOrder; use sea_orm::query::*;
mod pool; mod pool;
use pool::RocketDbPool; use pool::RocketDbPool;
@ -34,11 +34,11 @@ fn new() -> Template {
#[post("/", data = "<post_form>")] #[post("/", data = "<post_form>")]
async fn create(conn: Connection<Db>, post_form: Form<post::Model>) -> Flash<Redirect> { async fn create(conn: Connection<Db>, post_form: Form<post::Model>) -> Flash<Redirect> {
let post = post_form.into_inner(); let form = post_form.into_inner();
let _post = post::ActiveModel { post::ActiveModel {
title: Set(post.title.to_owned()), title: Set(form.title.to_owned()),
text: Set(post.text.to_owned()), text: Set(form.text.to_owned()),
..Default::default() ..Default::default()
} }
.save(&conn) .save(&conn)
@ -57,12 +57,12 @@ async fn update(conn: Connection<Db>, id: i32, post_form: Form<post::Model>) ->
.unwrap() .unwrap()
.into(); .into();
let post_data = post_form.into_inner(); let form = post_form.into_inner();
let _edited_post = post::ActiveModel { post::ActiveModel {
id: post.id, id: post.id,
title: Set(post_data.title.to_owned()), title: Set(form.title.to_owned()),
text: Set(post_data.text.to_owned()), text: Set(form.text.to_owned()),
} }
.save(&conn) .save(&conn)
.await .await
@ -113,14 +113,15 @@ async fn delete(conn: Connection<Db>, id: i32) -> Flash<Redirect> {
.unwrap() .unwrap()
.unwrap() .unwrap()
.into(); .into();
let _result = post.delete(&conn).await.unwrap();
post.delete(&conn).await.unwrap();
Flash::success(Redirect::to("/"), "Post successfully deleted.") Flash::success(Redirect::to("/"), "Post successfully deleted.")
} }
#[delete("/")] #[delete("/")]
async fn destroy(conn: Connection<Db>) -> Result<()> { async fn destroy(conn: Connection<Db>) -> Result<()> {
let _result = Post::delete_many().exec(&conn).await.unwrap(); Post::delete_many().exec(&conn).await.unwrap();
Ok(()) Ok(())
} }
@ -137,7 +138,7 @@ pub fn not_found(req: &Request<'_>) -> Template {
async fn run_migrations(rocket: Rocket<Build>) -> fairing::Result { async fn run_migrations(rocket: Rocket<Build>) -> fairing::Result {
let db_url = Db::fetch(&rocket).unwrap().db_url.clone(); let db_url = Db::fetch(&rocket).unwrap().db_url.clone();
let conn = sea_orm::Database::connect(&db_url).await.unwrap(); let conn = sea_orm::Database::connect(&db_url).await.unwrap();
let _create_post_table = setup::create_post_table(&conn).await; setup::create_post_table(&conn).await;
Ok(rocket) Ok(rocket)
} }