1 files changed, 2 insertions, 28 deletions
diff --git a/src/main.rs b/src/main.rs
index 9e29a52..d9747c9 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -2,33 +2,7 @@
 #![plugin(rocket_codegen)]
 
 extern crate rocket;
-#[macro_use]
-extern crate diesel;
-#[macro_use]
-extern crate diesel_codegen;
-extern crate dotenv;
-extern crate r2d2_diesel;
-extern crate r2d2;
-
-use diesel::pg::PgConnection;
-use r2d2_diesel::ConnectionManager;
-use dotenv::dotenv;
-use std::env;
-
-// An alias to the type for a pool of Diesel PostgreSql connections.
-type Pool = r2d2::Pool<ConnectionManager<PgConnection>>;
-
-/// Initializes a database pool.
-fn init_pool() -> Pool {
-    dotenv().ok();
-
-    let config = r2d2::Config::default();
-    let dburl = env::var("DATABASE_URL")
-        .expect("DATABASE_URL environment variable must be set");
-
-    let manager = ConnectionManager::<PgConnection>::new(dburl);
-    r2d2::Pool::new(config, manager).expect("db pool")
-}
+extern crate rocket_blog;
 
 #[get("/")]
 fn index() -> &'static str {
@@ -37,7 +11,7 @@ fn index() -> &'static str {
 
 fn main() {
     rocket::ignite()
-        .manage(init_pool())
+        .manage(rocket_blog::init_db_pool())
         .mount("/", routes![index])
         .launch();
 }