make import more robust and with fewer duplicates for stars.
This commit is contained in:
parent
a46a2e8847
commit
c1bea8284c
2 changed files with 40 additions and 19 deletions
|
@ -82,7 +82,7 @@ pub async fn import_imdb_data(w2w_db: &SqlitePool, imdb: &SqlitePool, ids: &mut
|
||||||
.await
|
.await
|
||||||
.unwrap();
|
.unwrap();
|
||||||
|
|
||||||
for batch in iwatches.chunks(2_000) {
|
for batch in iwatches.chunks(5_000) {
|
||||||
let mut tx = w2w_db.acquire().await.unwrap();
|
let mut tx = w2w_db.acquire().await.unwrap();
|
||||||
let mut tx = tx.begin().await.unwrap();
|
let mut tx = tx.begin().await.unwrap();
|
||||||
for iwatch in batch {
|
for iwatch in batch {
|
||||||
|
@ -114,6 +114,9 @@ async fn add_imdb_stars(
|
||||||
|
|
||||||
for row in principals {
|
for row in principals {
|
||||||
let (name_id, cat) = row;
|
let (name_id, cat) = row;
|
||||||
|
let star = if let Some(id) = ids.get(&name_id) {
|
||||||
|
*id
|
||||||
|
} else {
|
||||||
let name_query =
|
let name_query =
|
||||||
"select nconst, primaryName, birthYear, deathYear from names where nconst = ?";
|
"select nconst, primaryName, birthYear, deathYear from names where nconst = ?";
|
||||||
let istar: Option<ImdbStar> = sqlx::query_as(name_query)
|
let istar: Option<ImdbStar> = sqlx::query_as(name_query)
|
||||||
|
@ -125,15 +128,20 @@ async fn add_imdb_stars(
|
||||||
let star = (&star).into();
|
let star = (&star).into();
|
||||||
let star_id = insert_star(&star, w2w_db).await;
|
let star_id = insert_star(&star, w2w_db).await;
|
||||||
ids.insert(name_id, star_id);
|
ids.insert(name_id, star_id);
|
||||||
|
star_id
|
||||||
|
} else {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
let credit = Credit {
|
let credit = Credit {
|
||||||
star: star_id,
|
star,
|
||||||
watch,
|
watch,
|
||||||
credit: Some(cat.to_string()),
|
credit: Some(cat.to_string()),
|
||||||
};
|
};
|
||||||
insert_credit(&credit, w2w_db).await;
|
insert_credit(&credit, w2w_db).await;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
fn show_kind(kind: &str) -> ShowKind {
|
fn show_kind(kind: &str) -> ShowKind {
|
||||||
/*
|
/*
|
||||||
|
|
|
@ -47,6 +47,19 @@ pub async fn insert_credit(credit: &Credit, db: &mut SqliteConnection) {
|
||||||
.bind(credit.credit.as_deref())
|
.bind(credit.credit.as_deref())
|
||||||
.execute(db)
|
.execute(db)
|
||||||
.await
|
.await
|
||||||
|
.map(|_| ())
|
||||||
|
.or_else(|e| match e {
|
||||||
|
sqlx::Error::Database(ref db) => {
|
||||||
|
let exit = db.code().unwrap_or_default().parse().unwrap_or(0u32);
|
||||||
|
// https://www.sqlite.org/rescode.html codes for unique constraint violations:
|
||||||
|
if exit == 2067 || exit == 1555 {
|
||||||
|
Ok(())
|
||||||
|
} else {
|
||||||
|
Err(e)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ => Err(e),
|
||||||
|
})
|
||||||
.unwrap();
|
.unwrap();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue