format: postgres_migrations

This commit is contained in:
thesimplekid 2023-02-06 00:52:17 -05:00 committed by Greg Heartsfield
parent d7cceab8fc
commit 1ca5d652de

View File

@ -39,7 +39,10 @@ pub async fn run_migrations(db: &PostgresPool) -> crate::error::Result<usize> {
} }
async fn current_version(db: &PostgresPool) -> i64 { async fn current_version(db: &PostgresPool) -> i64 {
sqlx::query_scalar("SELECT max(serial_number) FROM migrations;").fetch_one(db).await.unwrap() sqlx::query_scalar("SELECT max(serial_number) FROM migrations;")
.fetch_one(db)
.await
.unwrap()
} }
async fn prepare_migrations_table(db: &PostgresPool) { async fn prepare_migrations_table(db: &PostgresPool) {
@ -138,15 +141,15 @@ CREATE INDEX user_verification_name_idx ON user_verification USING btree (name);
} }
mod m002 { mod m002 {
use async_std::stream::StreamExt;
use indicatif::{ProgressBar, ProgressStyle};
use sqlx::Row;
use std::time::Instant; use std::time::Instant;
use tracing::info; use tracing::info;
use async_std::stream::StreamExt;
use sqlx::Row;
use indicatif::{ProgressBar, ProgressStyle};
use crate::repo::postgres_migration::{Migration, SimpleSqlMigration}; use crate::event::{single_char_tagname, Event};
use crate::repo::postgres::PostgresPool; use crate::repo::postgres::PostgresPool;
use crate::event::{Event, single_char_tagname}; use crate::repo::postgres_migration::{Migration, SimpleSqlMigration};
use crate::utils::is_lower_hex; use crate::utils::is_lower_hex;
pub const VERSION: i64 = 2; pub const VERSION: i64 = 2;
@ -173,23 +176,31 @@ CREATE INDEX tag_value_hex_idx ON tag USING btree (value_hex);
let mut tx = db.begin().await.unwrap(); let mut tx = db.begin().await.unwrap();
let mut update_tx = db.begin().await.unwrap(); let mut update_tx = db.begin().await.unwrap();
// Clear out table // Clear out table
sqlx::query("DELETE FROM tag;").execute(&mut update_tx).await?; sqlx::query("DELETE FROM tag;")
.execute(&mut update_tx)
.await?;
{ {
let event_count: i64 = let event_count: i64 = sqlx::query_scalar("SELECT COUNT(*) from event;")
sqlx::query_scalar("SELECT COUNT(*) from event;")
.fetch_one(&mut tx) .fetch_one(&mut tx)
.await .await
.unwrap(); .unwrap();
let bar = ProgressBar::new(event_count.try_into().unwrap()).with_message("rebuilding tags table"); let bar = ProgressBar::new(event_count.try_into().unwrap())
bar.set_style(ProgressStyle::with_template("[{elapsed_precise}] {bar:40.white/blue} {pos:>7}/{len:7} [{percent}%] {msg}").unwrap()); .with_message("rebuilding tags table");
let mut events = sqlx::query("SELECT id, content FROM event ORDER BY id;").fetch(&mut tx); bar.set_style(
ProgressStyle::with_template(
"[{elapsed_precise}] {bar:40.white/blue} {pos:>7}/{len:7} [{percent}%] {msg}",
)
.unwrap(),
);
let mut events =
sqlx::query("SELECT id, content FROM event ORDER BY id;").fetch(&mut tx);
while let Some(row) = events.next().await { while let Some(row) = events.next().await {
bar.inc(1); bar.inc(1);
// get the row id and content // get the row id and content
let row = row.unwrap(); let row = row.unwrap();
let event_id: Vec<u8> = row.get(0); let event_id: Vec<u8> = row.get(0);
let event_bytes: Vec<u8> = row.get(1); let event_bytes: Vec<u8> = row.get(1);
let event:Event = serde_json::from_str(&String::from_utf8(event_bytes).unwrap())?; let event: Event = serde_json::from_str(&String::from_utf8(event_bytes).unwrap())?;
for t in event.tags.iter().filter(|x| x.len() > 1) { for t in event.tags.iter().filter(|x| x.len() > 1) {
let tagname = t.get(0).unwrap(); let tagname = t.get(0).unwrap();
@ -203,12 +214,21 @@ CREATE INDEX tag_value_hex_idx ON tag USING btree (value_hex);
// this means it needs to be even length and lowercase. // this means it needs to be even length and lowercase.
if (tagval.len() % 2 == 0) && is_lower_hex(tagval) { if (tagval.len() % 2 == 0) && is_lower_hex(tagval) {
let q = "INSERT INTO tag (event_id, \"name\", value_hex) VALUES ($1, $2, $3) ON CONFLICT DO NOTHING;"; let q = "INSERT INTO tag (event_id, \"name\", value_hex) VALUES ($1, $2, $3) ON CONFLICT DO NOTHING;";
sqlx::query(q).bind(&event_id).bind(&tagname).bind(hex::decode(tagval).ok()).execute(&mut update_tx).await?; sqlx::query(q)
.bind(&event_id)
.bind(&tagname)
.bind(hex::decode(tagval).ok())
.execute(&mut update_tx)
.await?;
} else { } else {
let q = "INSERT INTO tag (event_id, \"name\", value) VALUES ($1, $2, $3) ON CONFLICT DO NOTHING;"; let q = "INSERT INTO tag (event_id, \"name\", value) VALUES ($1, $2, $3) ON CONFLICT DO NOTHING;";
sqlx::query(q).bind(&event_id).bind(&tagname).bind(tagval.as_bytes()).execute(&mut update_tx).await?; sqlx::query(q)
.bind(&event_id)
.bind(&tagname)
.bind(tagval.as_bytes())
.execute(&mut update_tx)
.await?;
} }
} }
} }
update_tx.commit().await?; update_tx.commit().await?;
@ -221,7 +241,7 @@ CREATE INDEX tag_value_hex_idx ON tag USING btree (value_hex);
mod m003 { mod m003 {
use crate::repo::postgres_migration::{Migration, SimpleSqlMigration}; use crate::repo::postgres_migration::{Migration, SimpleSqlMigration};
pub const VERSION: i64 = 3; pub const VERSION: i64 = 3;
pub fn migration() -> impl Migration { pub fn migration() -> impl Migration {
@ -235,4 +255,4 @@ ALTER TABLE tag ADD CONSTRAINT unique_constraint_name UNIQUE (event_id, "name",
], ],
} }
} }
} }