account loading with deduplication

warehouse/sql/migrations/1_init.sql

@@ -8,6 +8,6 @@ CREATE TABLE balance (
     account_address CHAR(64) REFERENCES users(account_address) ON DELETE CASCADE,
     balance BIGINT NOT NULL,
     chain_timestamp TIMESTAMP NOT NULL,
-    height BIGINT NOT NULL,
+    db_version BIGINT NOT NULL,
     epoch_number BIGINT NOT NULL
 );

warehouse/src/lib.rs

@@ -3,3 +3,4 @@ pub mod scan;
 pub mod warehouse_cli;
 pub mod extract;
 pub mod table_structs;
+pub mod load;

warehouse/src/load.rs

@@ -0,0 +1,27 @@
+use crate::table_structs::{WarehouseAccount, WarehouseState};
+use anyhow::Result;
+use sqlx::{sqlite::SqliteQueryResult, SqlitePool};
+
+pub async fn load_account_state(pool: &SqlitePool, accounts: Vec<WarehouseState>) -> Result<()> {
+    // insert missing accounts
+    for ws in accounts.iter() {
+        insert_one_account(pool, &ws.account).await?;
+    }
+
+    // increment the balance changes
+    Ok(())
+}
+
+pub async fn insert_one_account(pool: &SqlitePool, acc: &WarehouseAccount) -> Result<SqliteQueryResult> {
+
+    let res = sqlx::query(r#"
+      INSERT INTO users (account_address, is_legacy)
+      VALUES ($1,$2)
+    "#)
+    .bind(acc.address.to_string())
+    .bind(true)
+    .execute(pool)
+    .await?;
+
+    Ok(res)
+}

warehouse/tests/test_load.rs

@@ -0,0 +1,19 @@
+use libra_types::exports::AccountAddress;
+use libra_warehouse::table_structs::WarehouseAccount;
+use sqlx::SqlitePool;
+
+#[sqlx::test]
+async fn insert_one_account(pool: SqlitePool) -> anyhow::Result<()> {
+    libra_warehouse::migrate::maybe_init(&pool).await?;
+    let marlon = AccountAddress::random();
+    let acc = WarehouseAccount {
+      address: marlon
+    };
+
+    libra_warehouse::load::insert_one_account(&pool, &acc).await?;
+
+    // second time should error if we are using the same account
+    assert!(libra_warehouse::load::insert_one_account(&pool, &acc).await.is_err());
+
+    Ok(())
+}

warehouse/tests/test_migrate.rs

@@ -20,7 +20,7 @@ async fn can_init(pool: SqlitePool) -> anyhow::Result<()> {
 
     let id = sqlx::query(
         r#"
-      INSERT INTO balance (account_address, balance, chain_timestamp, height, epoch_number)
+      INSERT INTO balance (account_address, balance, chain_timestamp, db_version, epoch_number)
       VALUES ("00000000000000000000000000000000e8953084617dd5c6071cf2918215e183", 11, 22222222, 600, 1)
       "#,
     )