blockworks-foundation
/
BankingStageErrorsTrackingSidecar
mirror of https://github.com/blockworks-foundation/BankingStageErrorsTrackingSidecar.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

redesign create_accounts_for_transaction

This commit is contained in:
GroovieGermanikus 2024-01-31 12:12:45 +01:00
parent 3465af2a13
commit 1569a6cd2f
No known key found for this signature in database
GPG Key ID: 5B6EB831A5CD2015
1 changed files with 11 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
src/postgres.rs
View File

@ -244,30 +244,16 @@ impl PostgresSession {
&self,
accounts: HashSet<String>,
) -> anyhow::Result<()> {
// create temp table
let temp_table = self.get_new_temp_table();
self.client
.execute(
format!(
"CREATE TEMP TABLE {}(
let statement = "CREATE TEMP TABLE temp_new_accounts_raw(
key TEXT
);",
temp_table
)
.as_str(),
&[],
)
.await?;
)";
self.client.execute(statement, &[]).await?;
let statement = format!(
r#"
COPY {}(
let statement = r#"
COPY temp_new_accounts_raw(
key
) FROM STDIN BINARY
"#,
temp_table
);
"#.to_string();
let started_at = Instant::now();
let sink: CopyInSink<bytes::Bytes> = self.copy_in(statement.as_str()).await?;
let writer = BinaryCopyInWriter::new(sink, &[Type::TEXT]);
@ -282,22 +268,18 @@ impl PostgresSession {
started_at.elapsed().as_millis()
);
let statement = format!(
r#"
INSERT INTO banking_stage_results_2.accounts(account_key) SELECT key from {}
ON CONFLICT DO NOTHING
"#,
temp_table
);
let statement = r#"
INSERT INTO banking_stage_results_2.accounts(account_key) SELECT key FROM temp_new_accounts_raw
ON CONFLICT DO NOTHING
"#;
let started_at = Instant::now();
self.client.execute(statement.as_str(), &[]).await?;
self.client.execute(statement, &[]).await?;
debug!(
"inserted {} account keys into accounts table in {}ms",
num_rows,
started_at.elapsed().as_millis()
);
self.drop_temp_table(temp_table).await?;
Ok(())
}