Browse Source

ingest: make zmq client store raw transaction data

wip_broken_chromebook
George Tankersley 6 years ago
parent
commit
71c0624abe
  1. 42
      cmd/ingest/main.go
  2. 11
      parser/block.go
  3. 20
      parser/transaction.go
  4. 2
      parser/transaction_test.go

42
cmd/ingest/main.go

@ -188,25 +188,25 @@ func handleBlock(db *sql.DB, sequence int, blockData []byte) {
return
}
displayHash := hex.EncodeToString(block.GetEncodableHash())
blockHash := hex.EncodeToString(block.GetEncodableHash())
marshaledBlock, _ := proto.Marshal(block.ToCompact())
err = storage.StoreBlock(
db,
block.GetHeight(),
displayHash,
blockHash,
block.HasSaplingTransactions(),
marshaledBlock,
)
entry := log.WithFields(logrus.Fields{
"seqnum": sequence,
"block_height": block.GetHeight(),
"block_hash": displayHash,
"block_version": block.GetVersion(),
"tx_count": block.GetTxCount(),
"has_sapling_tx": block.HasSaplingTransactions(),
"error": err,
"seqnum": sequence,
"block_height": block.GetHeight(),
"block_hash": blockHash,
"block_version": block.GetVersion(),
"tx_count": block.GetTxCount(),
"has_sapling": block.HasSaplingTransactions(),
"error": err,
})
if err != nil {
@ -214,4 +214,28 @@ func handleBlock(db *sql.DB, sequence int, blockData []byte) {
} else {
entry.Info("received new block")
}
for index, tx := range block.Transactions() {
txHash := hex.EncodeToString(tx.GetEncodableHash())
err = storage.StoreTransaction(
db,
block.GetHeight(),
blockHash,
index,
txHash,
tx.Bytes(),
)
entry = log.WithFields(logrus.Fields{
"block_height": block.GetHeight(),
"block_hash": blockHash,
"tx_index": index,
"has_sapling": tx.HasSaplingTransactions(),
"error": err,
})
if err != nil {
entry.Error("error storing tx")
} else {
entry.Debug("storing tx")
}
}
}

11
parser/block.go

@ -9,8 +9,8 @@ import (
)
type block struct {
hdr *blockHeader
vtx []*transaction
hdr *blockHeader
vtx []*Transaction
}
func NewBlock() *block {
@ -25,6 +25,11 @@ func (b *block) GetTxCount() int {
return len(b.vtx)
}
func (b *block) Transactions() []*Transaction {
// TODO: these should NOT be mutable
return b.vtx
}
// GetDisplayHash returns the block hash in big-endian display order.
func (b *block) GetDisplayHash() []byte {
return b.hdr.GetDisplayHash()
@ -101,7 +106,7 @@ func (b *block) ParseFromSlice(data []byte) (rest []byte, err error) {
}
data = []byte(s)
vtx := make([]*transaction, 0, txCount)
vtx := make([]*Transaction, 0, txCount)
for i := 0; len(data) > 0; i++ {
tx := NewTransaction()
data, err = tx.ParseFromSlice(data)

20
parser/transaction.go

@ -263,14 +263,14 @@ func (p *joinSplit) ParseFromSlice(data []byte) ([]byte, error) {
return []byte(s), nil
}
type transaction struct {
type Transaction struct {
*rawTransaction
rawBytes []byte
txId []byte
}
// GetDisplayHash returns the transaction hash in big-endian display order.
func (tx *transaction) GetDisplayHash() []byte {
func (tx *Transaction) GetDisplayHash() []byte {
if tx.txId != nil {
return tx.txId
}
@ -290,17 +290,21 @@ func (tx *transaction) GetDisplayHash() []byte {
}
// GetEncodableHash returns the transaction hash in little-endian wire format order.
func (tx *transaction) GetEncodableHash() []byte {
func (tx *Transaction) GetEncodableHash() []byte {
digest := sha256.Sum256(tx.rawBytes)
digest = sha256.Sum256(digest[:])
return digest[:]
}
func (tx *transaction) HasSaplingTransactions() bool {
func (tx *Transaction) Bytes() []byte {
return tx.rawBytes
}
func (tx *Transaction) HasSaplingTransactions() bool {
return tx.version >= 4 && (len(tx.shieldedSpends)+len(tx.shieldedOutputs)) > 0
}
func (tx *transaction) ToCompact(index int) *rpc.CompactTx {
func (tx *Transaction) ToCompact(index int) *rpc.CompactTx {
ctx := &rpc.CompactTx{
Index: uint64(index), // index is contextual
Hash: tx.GetEncodableHash(),
@ -317,7 +321,7 @@ func (tx *transaction) ToCompact(index int) *rpc.CompactTx {
return ctx
}
func (tx *transaction) ParseFromSlice(data []byte) ([]byte, error) {
func (tx *Transaction) ParseFromSlice(data []byte) ([]byte, error) {
s := bytestring.String(data)
// declare here to prevent shadowing problems in cryptobyte assignments
@ -465,8 +469,8 @@ func (tx *transaction) ParseFromSlice(data []byte) ([]byte, error) {
return []byte(s), nil
}
func NewTransaction() *transaction {
return &transaction{
func NewTransaction() *Transaction {
return &Transaction{
rawTransaction: new(rawTransaction),
}
}

2
parser/transaction_test.go

@ -225,7 +225,7 @@ func TestSproutTransactionParser(t *testing.T) {
}
}
func subTestCommonBlockMeta(tt *txTestVector, tx *transaction, t *testing.T, caseNum int) bool {
func subTestCommonBlockMeta(tt *txTestVector, tx *Transaction, t *testing.T, caseNum int) bool {
headerBytes, _ := hex.DecodeString(tt.header)
header := binary.LittleEndian.Uint32(headerBytes)
if (header >> 31) == 1 != tx.fOverwintered {

Loading…
Cancel
Save