lamassu-server/packages/server/lib/tx-batching.js
2025-05-12 15:35:00 +01:00

89 lines
2.7 KiB
JavaScript

const _ = require('lodash/fp')
const pgp = require('pg-promise')()
const uuid = require('uuid')
const BN = require('./bn')
const db = require('./db')
function addTransactionToBatch(tx) {
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`
return db.oneOrNone(sql, [tx.cryptoCode]).then(batch => {
if (_.isNil(batch)) {
return db.tx(t => {
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])
})
}
function closeTransactionBatch(batch) {
const sql = `UPDATE transaction_batches SET status='ready', closed_at=now() WHERE id=$1`
return db.none(sql, [batch.id])
}
function confirmSentBatch(batch, tx) {
return db.tx(t => {
const q1 = t.none(
`UPDATE transaction_batches SET status='sent', error_message=NULL WHERE id=$1`,
[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(batch, errorMsg) {
const sql = `UPDATE transaction_batches SET status='failed', error_message=$1 WHERE id=$2`
return db.none(sql, [errorMsg, batch.id])
}
function getBatchTransactions(batch) {
const sql = `SELECT * FROM cash_in_txs WHERE batch_id=$1`
return db
.manyOrNone(sql, [batch.id])
.then(res => _.map(_.mapKeys(_.camelCase), res))
}
function getBatchesByStatus(statuses) {
const sql = `SELECT *, EXTRACT(EPOCH FROM (now() - created_at)) as time_elapsed FROM transaction_batches WHERE status in ($1^)`
return db.manyOrNone(sql, [_.map(pgp.as.text, statuses).join(',')])
}
function getOpenBatchCryptoValue(cryptoCode) {
const sql = `SELECT * FROM transaction_batches WHERE crypto_code=$1 AND status='open' ORDER BY created_at DESC LIMIT 1`
return db
.oneOrNone(sql, [cryptoCode])
.then(batch => {
if (_.isNil(batch)) return Promise.resolve([])
return db.any(`SELECT * FROM cash_in_txs WHERE batch_id=$1`, [batch.id])
})
.then(txs => _.reduce((acc, tx) => acc.plus(tx.crypto_atoms), BN(0), txs))
}
module.exports = {
addTransactionToBatch,
closeTransactionBatch,
confirmSentBatch,
setErroredBatch,
getBatchTransactions,
getBatchesByStatus,
getOpenBatchCryptoValue,
}