Julio Biason
3 years ago
1 changed files with 35 additions and 2 deletions
@ -1,16 +1,49 @@ |
|||||||
|
/* |
||||||
|
TIN - Time Tracking Application |
||||||
|
Copyright (C) 2021 Julio Biason |
||||||
|
|
||||||
|
This program is free software: you can redistribute it and/or modify |
||||||
|
it under the terms of the GNU Affero General Public License as published by |
||||||
|
the Free Software Foundation, either version 3 of the License, or |
||||||
|
(at your option) any later version. |
||||||
|
|
||||||
|
This program is distributed in the hope that it will be useful, |
||||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||||
|
GNU Affero General Public License for more details. |
||||||
|
|
||||||
|
You should have received a copy of the GNU Affero General Public License |
||||||
|
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||||
|
*/ |
||||||
|
|
||||||
|
use std::path::Path; |
||||||
|
|
||||||
|
use directories::ProjectDirs; |
||||||
use sqlx::sqlite::Sqlite; |
use sqlx::sqlite::Sqlite; |
||||||
use sqlx::sqlite::SqliteConnectOptions; |
use sqlx::sqlite::SqliteConnectOptions; |
||||||
use sqlx::sqlite::SqlitePoolOptions; |
use sqlx::sqlite::SqlitePoolOptions; |
||||||
use sqlx::Pool; |
use sqlx::Pool; |
||||||
|
|
||||||
async fn connect() -> Result<Pool<Sqlite>, sqlx::Error> { |
async fn connect() -> Result<Pool<Sqlite>, sqlx::Error> { |
||||||
|
connect_with(&default_filename()).await |
||||||
|
} |
||||||
|
|
||||||
|
async fn connect_with(filename: &Path) -> Result<Pool<Sqlite>, sqlx::Error> { |
||||||
let pool = SqlitePoolOptions::new() |
let pool = SqlitePoolOptions::new() |
||||||
.connect_with( |
.connect_with( |
||||||
SqliteConnectOptions::new() |
SqliteConnectOptions::new() |
||||||
.filename("testing.sqlite") |
.filename(filename) |
||||||
.create_if_missing(true), |
.create_if_missing(true), |
||||||
) |
) |
||||||
.await?; |
.await?; |
||||||
sqlx::migrate("migrations").run(&pool).await?; |
sqlx::migrate!("./migrations").run(&pool).await?; |
||||||
Ok(pool) |
Ok(pool) |
||||||
} |
} |
||||||
|
|
||||||
|
/// Return the default filename for the database.
|
||||||
|
fn default_filename() -> Path { |
||||||
|
match ProjectDirs::from("me", "JulioBiason", "tin.sqlite3") { |
||||||
|
Some(project_dir) => Path(project_dir.config_dir()), |
||||||
|
None => Path("tin.sqlite3"), |
||||||
|
} |
||||||
|
} |
||||||
|
Loading…
Reference in new issue