add diesel migration

hails · Mar 21, 2020 · 2480641 · 2480641
1 parent fb909d9
commit 2480641
Show file tree

Hide file tree

Showing 4 changed files with 45 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -19,3 +19,4 @@ dotenv = "0.15.0"
 uuid = "0.6"
 chrono = "0.4.11"
 md5 = "0.7.0"
+diesel_migrations = "1.4.0"
diff --git a/diesel.toml b/diesel.toml
@@ -0,0 +1,5 @@
+# For documentation on how to configure this file,
+# see diesel.rs/guides/configuring-diesel-cli
+
+[print_schema]
+file = "src/schema.rs"
diff --git a/src/main.rs b/src/main.rs
@@ -3,6 +3,8 @@ extern crate diesel;
 
 use std::env;
 
+#[macro_use]
+extern crate diesel_migrations;
 use futures::StreamExt;
 use telegram_bot::*;
 
@@ -13,8 +15,13 @@ mod process;
 mod scheduler;
 mod schema;
 
+embed_migrations!();
+
 #[tokio::main]
 async fn main() -> Result<(), Error> {
+    let db_connection = database::establish_connection();
+    embedded_migrations::run(&db_connection).unwrap();
+
     let token = env::var("TELEGRAM_BOT_TOKEN").expect("TELEGRAM_BOT_TOKEN not set");
     let api = Api::new(&token);