Alexis Hernandez
6 years ago
9 changed files with 331 additions and 304 deletions
@ -0,0 +1,83 @@ |
|||
package com.xsn.explorer.data.anorm.dao |
|||
|
|||
import java.sql.Connection |
|||
|
|||
import anorm._ |
|||
import com.xsn.explorer.data.anorm.parsers.TransactionParsers._ |
|||
import com.xsn.explorer.models.{AddressTransactionDetails, Transaction, TransactionId} |
|||
|
|||
class AddressTransactionDetailsPostgresDAO { |
|||
|
|||
def batchInsertDetails(transaction: Transaction)(implicit conn: Connection): Option[Unit] = { |
|||
val received = transaction |
|||
.outputs |
|||
.groupBy(_.address) |
|||
.mapValues { outputs => outputs.map(_.value).sum } |
|||
.map { case (address, value) => AddressTransactionDetails(address, transaction.id, time = transaction.time, received = value) } |
|||
|
|||
val sent = transaction |
|||
.inputs |
|||
.groupBy(_.address) |
|||
.mapValues { inputs => inputs.map(_.value).sum } |
|||
.map { case (address, value) => AddressTransactionDetails(address, transaction.id, time = transaction.time, sent = value) } |
|||
|
|||
val details = (received ++ sent) |
|||
.groupBy(_.address) |
|||
.mapValues { |
|||
case head :: list => list.foldLeft(head) { (acc, current) => |
|||
current.copy(received = current.received + acc.received, sent = current.sent + acc.sent) |
|||
} |
|||
} |
|||
.values |
|||
|
|||
batchInsertDetails(details.toList) |
|||
} |
|||
|
|||
def batchInsertDetails(details: List[AddressTransactionDetails])(implicit conn: Connection): Option[Unit] = { |
|||
details match { |
|||
case Nil => Some(()) |
|||
case _ => |
|||
val params = details.map { d => |
|||
List( |
|||
'address -> d.address.string: NamedParameter, |
|||
'txid -> d.txid.string: NamedParameter, |
|||
'received -> d.received: NamedParameter, |
|||
'sent -> d.sent: NamedParameter, |
|||
'time -> d.time: NamedParameter) |
|||
} |
|||
|
|||
val batch = BatchSql( |
|||
""" |
|||
|INSERT INTO address_transaction_details |
|||
| (address, txid, received, sent, time) |
|||
|VALUES |
|||
| ({address}, {txid}, {received}, {sent}, {time}) |
|||
""".stripMargin, |
|||
params.head, |
|||
params.tail: _* |
|||
) |
|||
|
|||
val success = batch.execute().forall(_ == 1) |
|||
|
|||
if (success) { |
|||
Some(()) |
|||
} else { |
|||
None |
|||
} |
|||
} |
|||
} |
|||
|
|||
def deleteDetails(txid: TransactionId)(implicit conn: Connection): List[AddressTransactionDetails] = { |
|||
val result = SQL( |
|||
""" |
|||
|DELETE FROM address_transaction_details |
|||
|WHERE txid = {txid} |
|||
|RETURNING address, txid, received, sent, time |
|||
""".stripMargin |
|||
).on( |
|||
'txid -> txid.string |
|||
).as(parseAddressTransactionDetails.*) |
|||
|
|||
result |
|||
} |
|||
} |
|||
@ -0,0 +1,75 @@ |
|||
package com.xsn.explorer.data.anorm.dao |
|||
|
|||
import java.sql.Connection |
|||
|
|||
import anorm._ |
|||
import com.xsn.explorer.data.anorm.parsers.TransactionParsers._ |
|||
import com.xsn.explorer.models.{Address, Transaction, TransactionId} |
|||
|
|||
class TransactionInputPostgresDAO { |
|||
|
|||
def batchInsertInputs( |
|||
inputs: List[(TransactionId, Transaction.Input)])( |
|||
implicit conn: Connection): Option[List[(TransactionId, Transaction.Input)]] = { |
|||
|
|||
inputs match { |
|||
case Nil => Some(inputs) |
|||
|
|||
case _ => |
|||
val params = inputs.map { case (txid, input) => |
|||
List( |
|||
'txid -> txid.string: NamedParameter, |
|||
'index -> input.index: NamedParameter, |
|||
'from_txid -> input.fromTxid.string: NamedParameter, |
|||
'from_output_index -> input.fromOutputIndex: NamedParameter, |
|||
'value -> input.value: NamedParameter, |
|||
'address -> input.address.string: NamedParameter) |
|||
} |
|||
|
|||
val batch = BatchSql( |
|||
""" |
|||
|INSERT INTO transaction_inputs |
|||
| (txid, index, from_txid, from_output_index, value, address) |
|||
|VALUES |
|||
| ({txid}, {index}, {from_txid}, {from_output_index}, {value}, {address}) |
|||
""".stripMargin, |
|||
params.head, |
|||
params.tail: _* |
|||
) |
|||
|
|||
val success = batch.execute().forall(_ == 1) |
|||
|
|||
if (success) { |
|||
Some(inputs) |
|||
} else { |
|||
None |
|||
} |
|||
} |
|||
} |
|||
|
|||
def deleteInputs(txid: TransactionId)(implicit conn: Connection): List[Transaction.Input] = { |
|||
SQL( |
|||
""" |
|||
|DELETE FROM transaction_inputs |
|||
|WHERE txid = {txid} |
|||
|RETURNING txid, index, from_txid, from_output_index, value, address |
|||
""".stripMargin |
|||
).on( |
|||
'txid -> txid.string |
|||
).as(parseTransactionInput.*).flatten |
|||
} |
|||
|
|||
def getInputs(txid: TransactionId, address: Address)(implicit conn: Connection): List[Transaction.Input] = { |
|||
SQL( |
|||
""" |
|||
|SELECT txid, index, from_txid, from_output_index, value, address |
|||
|FROM transaction_inputs |
|||
|WHERE txid = {txid} AND |
|||
| address = {address} |
|||
""".stripMargin |
|||
).on( |
|||
'txid -> txid.string, |
|||
'address -> address.string |
|||
).as(parseTransactionInput.*).flatten |
|||
} |
|||
} |
|||
@ -0,0 +1,128 @@ |
|||
package com.xsn.explorer.data.anorm.dao |
|||
|
|||
import java.sql.Connection |
|||
|
|||
import anorm._ |
|||
import com.xsn.explorer.data.anorm.parsers.TransactionParsers._ |
|||
import com.xsn.explorer.models.{Address, Transaction, TransactionId} |
|||
|
|||
class TransactionOutputPostgresDAO { |
|||
|
|||
def getUnspentOutputs(address: Address)(implicit conn: Connection): List[Transaction.Output] = { |
|||
SQL( |
|||
""" |
|||
|SELECT txid, index, value, address, hex_script, tpos_owner_address, tpos_merchant_address |
|||
|FROM transaction_outputs |
|||
|WHERE address = {address} AND |
|||
| spent_on IS NULL AND |
|||
| value > 0 |
|||
""".stripMargin |
|||
).on( |
|||
'address -> address.string |
|||
).as(parseTransactionOutput.*).flatten |
|||
} |
|||
|
|||
def batchInsertOutputs( |
|||
outputs: List[Transaction.Output])( |
|||
implicit conn: Connection): Option[List[Transaction.Output]] = { |
|||
|
|||
outputs match { |
|||
case Nil => Some(outputs) |
|||
case _ => |
|||
val params = outputs.map { output => |
|||
List( |
|||
'txid -> output.txid.string: NamedParameter, |
|||
'index -> output.index: NamedParameter, |
|||
'value -> output.value: NamedParameter, |
|||
'address -> output.address.string: NamedParameter, |
|||
'hex_script -> output.script.string: NamedParameter, |
|||
'tpos_owner_address -> output.tposOwnerAddress.map(_.string): NamedParameter, |
|||
'tpos_merchant_address -> output.tposMerchantAddress.map(_.string): NamedParameter) |
|||
} |
|||
|
|||
val batch = BatchSql( |
|||
""" |
|||
|INSERT INTO transaction_outputs |
|||
| (txid, index, value, address, hex_script, tpos_owner_address, tpos_merchant_address) |
|||
|VALUES |
|||
| ({txid}, {index}, {value}, {address}, {hex_script}, {tpos_owner_address}, {tpos_merchant_address}) |
|||
""".stripMargin, |
|||
params.head, |
|||
params.tail: _* |
|||
) |
|||
|
|||
val success = batch.execute().forall(_ == 1) |
|||
|
|||
if (success) { |
|||
Some(outputs) |
|||
} else { |
|||
None |
|||
} |
|||
} |
|||
} |
|||
|
|||
def deleteOutputs(txid: TransactionId)(implicit conn: Connection): List[Transaction.Output] = { |
|||
val result = SQL( |
|||
""" |
|||
|DELETE FROM transaction_outputs |
|||
|WHERE txid = {txid} |
|||
|RETURNING txid, index, hex_script, value, address, tpos_owner_address, tpos_merchant_address |
|||
""".stripMargin |
|||
).on( |
|||
'txid -> txid.string |
|||
).as(parseTransactionOutput.*) |
|||
|
|||
result.flatten |
|||
} |
|||
|
|||
def getOutputs(txid: TransactionId, address: Address)(implicit conn: Connection): List[Transaction.Output] = { |
|||
SQL( |
|||
""" |
|||
|SELECT txid, index, hex_script, value, address, tpos_owner_address, tpos_merchant_address |
|||
|FROM transaction_outputs |
|||
|WHERE txid = {txid} AND |
|||
| address = {address} |
|||
""".stripMargin |
|||
).on( |
|||
'txid -> txid.string, |
|||
'address -> address.string |
|||
).as(parseTransactionOutput.*).flatten |
|||
} |
|||
|
|||
def batchSpend(txid: TransactionId, inputs: List[Transaction.Input])(implicit conn: Connection): Option[Unit] = { |
|||
inputs match { |
|||
case Nil => Option(()) |
|||
case _ => |
|||
val txidArray = inputs |
|||
.map { input => s"'${input.fromTxid.string}'" } |
|||
.mkString("[", ",", "]") |
|||
|
|||
val indexArray = inputs.map(_.fromOutputIndex).mkString("[", ",", "]") |
|||
|
|||
// Note: the TransactionId must meet a safe format, this approach speeds up the inserts |
|||
val result = SQL( |
|||
s""" |
|||
|UPDATE transaction_outputs t |
|||
|SET spent_on = tmp.spent_on |
|||
|FROM ( |
|||
| WITH CTE AS ( |
|||
| SELECT '${txid.string}' AS spent_on |
|||
| ) |
|||
| SELECT spent_on, txid, index |
|||
| FROM CTE CROSS JOIN (SELECT |
|||
| UNNEST(array$indexArray) AS index, |
|||
| UNNEST(array$txidArray) AS txid) x |
|||
|) AS tmp |
|||
|WHERE t.txid = tmp.txid AND |
|||
| t.index = tmp.index |
|||
""".stripMargin |
|||
).executeUpdate() |
|||
|
|||
if (result == inputs.size) { |
|||
Option(()) |
|||
} else { |
|||
None |
|||
} |
|||
} |
|||
} |
|||
} |
|||
Loading…
Reference in new issue