selfhosted
/
zer0bin
mirror of https://github.com/zer0bin-dev/zer0bin
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

🐛 fix: Attempted fix, didnt work

This commit is contained in:
Dominic Harris 2022-04-05 07:56:38 -04:00
parent 16f170c382
commit fab8762c1e
No known key found for this signature in database
GPG Key ID: 93CCF85F3E2A4F65
1 changed files with 18 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
backend/src/main.rs
View File

@ -15,8 +15,9 @@ use config::Config;
use sqlx::{
migrate::Migrator,
postgres::{PgPoolOptions, PgRow},
PgPool, Row,
PgPool, Row, error::DatabaseError,
};
use sqlx::migrate::MigrateError::Execute;
use crate::routes::{
get_paste, get_raw_paste, get_stats, get_total_pastes_badge, get_version_badge, new_paste,
@ -29,20 +30,23 @@ pub struct AppState {
}
pub async fn migrations(pool: &PgPool) -> Result<(), sqlx::Error> {
let mut migrator = Migrator::new(Path::new("./migrations")).await?;
let migrator = Migrator::new(Path::new("./migrations")).await?;
for migration in migrator.iter() {
if migration.version == 0 {
let row: bool = sqlx::query("SELECT EXISTS (SELECT 1 FROM information_schema.tables WHERE table_name = 'pastes')")
.try_map(|row: PgRow| row.try_get::<bool, _>("exists"))
.fetch_one(pool).await?;
// The table exists so we dont want to run initial migration
if row == true {}
}
println!("Running migration: {:?}", migration);
}
// TODO: This is a hacky solution but it works for now
// it doesnt work lol
// if let Err(e) = migrator.run(pool).await {
// match e {
// Execute(ex) => {
// // If the error comes from a table exists check then we can return Ok
// if ex.as_database_error().unwrap().code().unwrap() == "42710" {
// return Ok(())
// }
// },
// _ => {
// return Err(e.into());
// }
// }
// };
migrator.run(pool).await?;