NexVeridian/wikidata-to-surrealdb
1
0
Fork 0
mirror of https://github.com/NexVeridian/wikidata-to-surrealdb.git synced 2025-09-02 01:49:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix tokio::fs

Browse source
This commit is contained in:
Elijah McMorris 2024-08-26 00:03:45 -07:00
parent 64d13fff58
commit 057dcb3308
Signed by: NexVeridian
SSH key fingerprint: SHA256:bsA1SKZxuEcEVHAy3gY1HUeM5ykRJl0U0kQHQn0hMg8
2 changed files with 7 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

2
Cargo.toml
View file

@ -9,7 +9,7 @@ anyhow = "1.0"
serde = { version = "1.0", features = ["derive"] }
serde_json = "1.0"
surrealdb-alpha = { version = "2.0.9", features = ["protocol-ws", "kv-mem"] }
tokio = { version = "1.39", features = ["time"] }
tokio = { version = "1.39", features = ["fs", "time"] }
futures = "0.3"
wikidata = "1.1"
bzip2 = { version = "0.4", features = ["tokio"] }

11
src/utils.rs
View file

@ -184,11 +184,12 @@ pub async fn create_db_entities_bulk_filter(
.map(char::from)
.collect();
db_mem
.export(format!("../data/temp/{}.surql", file_name))
.await?;
db.import(format!("../data/temp{}.surql", file_name))
.await?;
let file_path = format!("data/temp/{}.surql", file_name);
db_mem.export(&file_path).await?;
db.import(&file_path).await?;
tokio::fs::remove_file(&file_path).await?;
if let Some(ref p) = pb {
p.inc(batch_size as u64)