lamassu/lamassu-server
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: batching db transactions

Browse source
This commit is contained in:
Sérgio Salgado 2021-11-24 18:07:54 +00:00
parent f6e793ea4d
commit 0132227251
1 changed files with 39 additions and 46 deletions
Download patch file Download diff file Expand all files Collapse all files

71
lib/tx-batching.js
View file

@ -6,88 +6,81 @@ const BN = require('./bn')
const db = require('./db') const db = require('./db')
const wallet = require('./wallet') const wallet = require('./wallet')
function createTransactionBatch (cryptoCode) {
const sql = `INSERT INTO transaction_batches (id, crypto_code) VALUES ($1, $2) RETURNING *`
return db.one(sql, [uuid.v4(), cryptoCode])
}
function addTransactionToBatch (tx) { function addTransactionToBatch (tx) {
const sql = `SELECT * FROM transaction_batches WHERE crypto_code=$1 AND status='open' ORDER BY created_at DESC LIMIT 1` const sql = `SELECT * FROM transaction_batches WHERE crypto_code=$1 AND status='open' ORDER BY created_at DESC LIMIT 1`
const sql2 = `UPDATE cash_in_txs SET batch_id=$1 WHERE id=$2` const sql2 = `UPDATE cash_in_txs SET batch_id=$1 WHERE id=$2`
return db.oneOrNone(sql, [tx.cryptoCode]) return db.oneOrNone(sql, [tx.cryptoCode])
.then(batch => { .then(batch => {
if (_.isNil(batch)) if (_.isNil(batch)) {
return createTransactionBatch(tx.cryptoCode) return db.tx(t => {
return Promise.resolve(batch) const newBatchId = uuid.v4()
const q1 = t.none(`INSERT INTO transaction_batches (id, crypto_code) VALUES ($1, $2)`, [newBatchId, tx.cryptoCode])
const q2 = t.none(sql2, [newBatchId, tx.id])
return t.batch([q1, q2])
})
}
return db.none(sql2, [batch.id, tx.id])
}) })
.then(batch => db.none(sql2, [batch.id, tx.id]))
} }
function closeTransactionBatch (t, batch) { function closeTransactionBatch (batch) {
const sql = `UPDATE transaction_batches SET status='ready', closed_at=now() WHERE id=$1` const sql = `UPDATE transaction_batches SET status='ready', closed_at=now() WHERE id=$1`
return t.none(sql, [batch.id]) return db.none(sql, [batch.id])
} }
function confirmSentBatch (t, batch, tx) { function confirmSentBatch (batch, tx) {
return t.none(`UPDATE transaction_batches SET status='sent', error_message=NULL WHERE id=$1`, [batch.id]) return db.tx(t => {
.then(() => const q1 = t.none(`UPDATE transaction_batches SET status='sent', error_message=NULL WHERE id=$1`, [batch.id])
t.none(`UPDATE cash_in_txs SET tx_hash=$1, fee=$2, send=true, send_confirmed=true, send_time=now(), send_pending=false, error=NULL, error_code=NULL WHERE batch_id=$3`, [tx.txid, tx.fee.toString(), batch.id]) const q2 = t.none(`UPDATE cash_in_txs SET tx_hash=$1, fee=$2, send=true, send_confirmed=true, send_time=now(), send_pending=false, error=NULL, error_code=NULL WHERE batch_id=$3`, [tx.txid, tx.fee.toString(), batch.id])
)
return t.batch([q1, q2])
})
} }
function setErroredBatch (t, batch, errorMsg) { function setErroredBatch (batch, errorMsg) {
const sql = `UPDATE transaction_batches SET status='failed', error_message=$1 WHERE id=$2` const sql = `UPDATE transaction_batches SET status='failed', error_message=$1 WHERE id=$2`
return t.none(sql, [errorMsg, batch.id]) return db.none(sql, [errorMsg, batch.id])
} }
function getBatchTransactions (t, batch) { function getBatchTransactions (batch) {
const sql = `SELECT * FROM cash_in_txs WHERE batch_id=$1` const sql = `SELECT * FROM cash_in_txs WHERE batch_id=$1`
return t.manyOrNone(sql, [batch.id]) return db.manyOrNone(sql, [batch.id])
} }
function getBatchesByStatus (t, statuses) { function getBatchesByStatus (statuses) {
const sql = `SELECT *, EXTRACT(EPOCH FROM (now() - created_at)) as time_elapsed FROM transaction_batches WHERE status in ($1^)` const sql = `SELECT *, EXTRACT(EPOCH FROM (now() - created_at)) as time_elapsed FROM transaction_batches WHERE status in ($1^)`
return t.manyOrNone(sql, [_.map(pgp.as.text, statuses).join(',')]) return db.manyOrNone(sql, [_.map(pgp.as.text, statuses).join(',')])
} }
function submitBatch (t, settings, batch) { function submitBatch (settings, batch) {
getBatchTransactions(t, batch) getBatchTransactions(batch)
.then(txs => { .then(txs => {
wallet.sendCoinsBatch(settings, txs, batch.crypto_code) wallet.sendCoinsBatch(settings, txs, batch.crypto_code)
.then(res => confirmSentBatch(t, batch, res)) .then(res => confirmSentBatch(batch, res))
.catch(err => setErroredBatch(t, batch, err.message)) .catch(err => setErroredBatch(batch, err.message))
}) })
} }
function processBatches (settings, lifecycle) { function processBatches (settings, lifecycle) {
const transaction = t => { return getBatchesByStatus(['open'])
getBatchesByStatus(t, ['open', 'failed', 'ready'])
.then(batches => { .then(batches => {
_.each(batch => { _.each(batch => {
const elapsedMS = batch.time_elapsed * 1000 const elapsedMS = batch.time_elapsed * 1000
if (elapsedMS >= lifecycle) { if (elapsedMS >= lifecycle) {
return closeTransactionBatch(t, batch) return closeTransactionBatch(batch)
.then(() => submitBatch(t, settings, batch)) .then(() => submitBatch(settings, batch))
} }
}, batches) }, batches)
}) })
}
return db.tx(
new pgp.txMode.TransactionMode({ tiLevel: pgp.txMode.isolationLevel }),
transaction
)
} }
module.exports = { module.exports = {
createTransactionBatch,
closeTransactionBatch,
addTransactionToBatch, addTransactionToBatch,
processBatches processBatches
} }