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

Rocket example to use the clone interface

Browse Source
This commit is contained in:
Chris Tsang 2021-09-22 16:51:50 +08:00
parent c80621916a
commit aeea07c8ba
2 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -156,9 +156,8 @@ pub fn not_found(req: &Request<'_>) -> Template {
}
async fn run_migrations(rocket: Rocket<Build>) -> fairing::Result {
let db_url = Db::fetch(&rocket).unwrap().db_url.clone();
let conn = sea_orm::Database::connect(&db_url).await.unwrap();
let _ = setup::create_post_table(&conn).await;
let conn = &Db::fetch(&rocket).unwrap().conn;
let _ = setup::create_post_table(conn).await;
Ok(rocket)
}

8
examples/rocket_example/src/pool.rs
View File

@ -3,7 +3,7 @@ use rocket_db_pools::{rocket::figment::Figment, Config};
#[derive(Debug)]
pub struct RocketDbPool {
pub db_url: String,
pub conn: sea_orm::DatabaseConnection,
}
#[async_trait]
@ -14,14 +14,14 @@ impl rocket_db_pools::Pool for RocketDbPool {
async fn init(figment: &Figment) -> Result<Self, Self::Error> {
let config = figment.extract::<Config>().unwrap();
let db_url = config.url;
let conn = sea_orm::Database::connect(&config.url).await.unwrap();
Ok(RocketDbPool {
db_url: db_url.to_owned(),
conn,
})
}
async fn get(&self) -> Result<Self::Connection, Self::Error> {
Ok(sea_orm::Database::connect(&self.db_url).await.unwrap())
Ok(self.conn.clone())
}
}