dynco-nym
fbcf44eeb9
* Add /account/{address}
* Don't query vesting info
* Don't query rewards
* Remove unused code
* Fix clippy
* Fix build.rs build on Windows
* Addressing PR feedback
- not cloning nym nodes from cache
- reduced number of nym nodes kept in memory
- reduced number of iterations to read all data
- removed some fields
* Fix total_delegations
* Optimize nym_nodes hashmap
* Split flow into functions
* Remove vesting info
* Add caching for endpoint
* Cache optimizations
* Return early if balance is 0
* Refactor state cloning shenanigans
101 lines
3.0 KiB
Rust
101 lines
3.0 KiB
Rust
use sqlx::{Connection, FromRow, SqliteConnection};
|
|
use std::env;
|
|
|
|
const SQLITE_DB_FILENAME: &str = "nym-api-example.sqlite";
|
|
|
|
// it's fine if compilation fails
|
|
#[allow(clippy::unwrap_used)]
|
|
#[allow(clippy::expect_used)]
|
|
#[tokio::main]
|
|
async fn main() {
|
|
let out_dir = env::var("OUT_DIR").unwrap();
|
|
let database_path = format!("{}/{}", out_dir, SQLITE_DB_FILENAME);
|
|
|
|
#[cfg(target_family = "unix")]
|
|
write_db_path_to_file(&out_dir, SQLITE_DB_FILENAME)
|
|
.await
|
|
.ok();
|
|
|
|
let mut conn = SqliteConnection::connect(&format!("sqlite://{}?mode=rwc", database_path))
|
|
.await
|
|
.expect("Failed to create SQLx database connection");
|
|
|
|
sqlx::migrate!("./migrations")
|
|
.run(&mut conn)
|
|
.await
|
|
.expect("Failed to perform SQLx migrations");
|
|
|
|
#[derive(FromRow)]
|
|
struct Exists {
|
|
exists: bool,
|
|
}
|
|
|
|
// check if it was already run
|
|
let res: Exists = sqlx::query_as("SELECT EXISTS (SELECT 1 FROM v3_migration_info) AS 'exists'")
|
|
.fetch_one(&mut conn)
|
|
.await
|
|
.unwrap();
|
|
|
|
let already_run = res.exists;
|
|
|
|
// execute the manual v3 migration
|
|
// it's performed on an empty storage, so we don't need to actually make any network queries
|
|
if !already_run {
|
|
sqlx::query(
|
|
r#"
|
|
CREATE TABLE gateway_details_temp
|
|
(
|
|
id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
|
|
node_id INTEGER NOT NULL UNIQUE,
|
|
identity VARCHAR NOT NULL UNIQUE
|
|
);
|
|
|
|
DROP TABLE gateway_details;
|
|
ALTER TABLE gateway_details_temp RENAME TO gateway_details;
|
|
|
|
INSERT INTO v3_migration_info(id) VALUES (0);
|
|
"#,
|
|
)
|
|
.execute(&mut conn)
|
|
.await
|
|
.expect("failed to update post v3 migration tables");
|
|
}
|
|
|
|
#[cfg(target_family = "unix")]
|
|
println!("cargo:rustc-env=DATABASE_URL=sqlite://{}", &database_path);
|
|
|
|
#[cfg(target_family = "windows")]
|
|
// for some strange reason we need to add a leading `/` to the windows path even though it's
|
|
// not a valid windows path... but hey, it works...
|
|
println!("cargo:rustc-env=DATABASE_URL=sqlite:///{}", &database_path);
|
|
}
|
|
|
|
/// use `./enter_db.sh` to inspect DB
|
|
#[cfg(target_family = "unix")]
|
|
async fn write_db_path_to_file(out_dir: &str, db_filename: &str) -> anyhow::Result<()> {
|
|
use std::os::unix::fs::PermissionsExt;
|
|
use tokio::{fs::File, io::AsyncWriteExt};
|
|
|
|
if env::var("CI").is_ok() {
|
|
return Ok(());
|
|
}
|
|
let mut file = File::create("settings.sql").await?;
|
|
let settings = ".mode columns
|
|
.headers on";
|
|
file.write_all(settings.as_bytes()).await?;
|
|
|
|
let mut file = File::create("enter_db.sh").await?;
|
|
let contents = format!(
|
|
"#!/bin/sh\n\
|
|
sqlite3 -init settings.sql {}/{}",
|
|
out_dir, db_filename,
|
|
);
|
|
file.write_all(contents.as_bytes()).await?;
|
|
|
|
file.set_permissions(std::fs::Permissions::from_mode(0o755))
|
|
.await
|
|
.map_err(anyhow::Error::from)?;
|
|
|
|
Ok(())
|
|
}
|