-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #2 from SiaFoundation/add-txn
Store transactions
- Loading branch information
Showing
12 changed files
with
521 additions
and
289 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,80 @@ | ||
| package sqlite | ||
|
|
||
| import ( | ||
| "fmt" | ||
|
|
||
| "go.sia.tech/core/types" | ||
| ) | ||
|
|
||
| // BlockByID implements explorer.Store. | ||
| func (s *Store) BlockByID(id types.BlockID) (result types.Block, err error) { | ||
| err = s.transaction(func(tx txn) error { | ||
| err = tx.QueryRow(`SELECT parent_id, nonce, timestamp FROM blocks WHERE id=?`, dbEncode(id)).Scan(dbDecode(&result.ParentID), dbDecode(&result.Nonce), dbDecode(&result.Timestamp)) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| result.MinerPayouts, err = blockMinerPayouts(tx, id) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to get miner payouts: %v", err) | ||
| } | ||
|
|
||
| // get block transaction IDs | ||
| transactionIDs, err := blockTransactionIDs(tx, id) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to get block transaction IDs: %v", err) | ||
| } | ||
|
|
||
| // get arbitrary data for each transaction | ||
| txnArbitraryData, err := transactionArbitraryData(tx, transactionIDs) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to get arbitrary data: %v", err) | ||
| } | ||
|
|
||
| for _, id := range transactionIDs { | ||
| txn := types.Transaction{ | ||
| ArbitraryData: txnArbitraryData[id], | ||
| } | ||
| result.Transactions = append(result.Transactions, txn) | ||
| } | ||
| return nil | ||
| }) | ||
| return | ||
| } | ||
|
|
||
| // BlockByHeight implements explorer.Store. | ||
| func (s *Store) BlockByHeight(height uint64) (result types.Block, err error) { | ||
| err = s.transaction(func(tx txn) error { | ||
| var blockID types.BlockID | ||
| err = tx.QueryRow(`SELECT id, parent_id, nonce, timestamp FROM blocks WHERE height=?`, height).Scan(dbDecode(&blockID), dbDecode(&result.ParentID), dbDecode(&result.Nonce), dbDecode(&result.Timestamp)) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| result.MinerPayouts, err = blockMinerPayouts(tx, blockID) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to get miner payouts: %v", err) | ||
| } | ||
|
|
||
| // get block transaction IDs | ||
| transactionIDs, err := blockTransactionIDs(tx, blockID) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to get block transaction IDs: %v", err) | ||
| } | ||
|
|
||
| // get arbitrary data for each transaction | ||
| txnArbitraryData, err := transactionArbitraryData(tx, transactionIDs) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to get arbitrary data: %v", err) | ||
| } | ||
|
|
||
| for _, id := range transactionIDs { | ||
| txn := types.Transaction{ | ||
| ArbitraryData: txnArbitraryData[id], | ||
| } | ||
| result.Transactions = append(result.Transactions, txn) | ||
| } | ||
| return nil | ||
| }) | ||
| return | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,144 @@ | ||
| package sqlite | ||
|
|
||
| import ( | ||
| "database/sql" | ||
| "errors" | ||
| "fmt" | ||
|
|
||
| "go.sia.tech/core/chain" | ||
| "go.sia.tech/core/types" | ||
| ) | ||
|
|
||
| func (s *Store) addBlock(dbTxn txn, b types.Block, height uint64) error { | ||
| // nonce is encoded because database/sql doesn't support uint64 with high bit set | ||
| _, err := dbTxn.Exec("INSERT INTO blocks(id, height, parent_id, nonce, timestamp) VALUES (?, ?, ?, ?, ?);", dbEncode(b.ID()), height, dbEncode(b.ParentID), dbEncode(b.Nonce), dbEncode(b.Timestamp)) | ||
| return err | ||
| } | ||
|
|
||
| func (s *Store) addMinerPayouts(dbTxn txn, bid types.BlockID, scos []types.SiacoinOutput) error { | ||
| stmt, err := dbTxn.Prepare(`INSERT INTO miner_payouts(block_id, block_order, address, value) VALUES (?, ?, ?, ?);`) | ||
| if err != nil { | ||
| return fmt.Errorf("addMinerPayouts: failed to prepare statement: %v", err) | ||
| } | ||
| defer stmt.Close() | ||
|
|
||
| for i, sco := range scos { | ||
| if _, err := stmt.Exec(dbEncode(bid), i, dbEncode(sco.Address), dbEncode(sco.Value)); err != nil { | ||
| return fmt.Errorf("addMinerPayouts: failed to execute statement: %v", err) | ||
| } | ||
| } | ||
| return nil | ||
| } | ||
|
|
||
| func (s *Store) addArbitraryData(dbTxn txn, id int64, txn types.Transaction) error { | ||
| stmt, err := dbTxn.Prepare(`INSERT INTO arbitrary_data(transaction_id, transaction_order, data) VALUES (?, ?, ?)`) | ||
| if err != nil { | ||
| return fmt.Errorf("addArbitraryData: failed to prepare statement: %v", err) | ||
| } | ||
| defer stmt.Close() | ||
|
|
||
| for i, arbitraryData := range txn.ArbitraryData { | ||
| if _, err := stmt.Exec(id, i, arbitraryData); err != nil { | ||
| return fmt.Errorf("addArbitraryData: failed to execute statement: %v", err) | ||
| } | ||
| } | ||
| return nil | ||
| } | ||
|
|
||
| func (s *Store) addTransactions(dbTxn txn, bid types.BlockID, txns []types.Transaction) error { | ||
| transactionsStmt, err := dbTxn.Prepare(`INSERT INTO transactions(transaction_id) VALUES (?);`) | ||
| if err != nil { | ||
| return fmt.Errorf("addTransactions: failed to prepare transactions statement: %v", err) | ||
| } | ||
| defer transactionsStmt.Close() | ||
|
|
||
| blockTransactionsStmt, err := dbTxn.Prepare(`INSERT INTO block_transactions(block_id, transaction_id, block_order) VALUES (?, ?, ?);`) | ||
| if err != nil { | ||
| return fmt.Errorf("addTransactions: failed to prepare block_transactions statement: %v", err) | ||
| } | ||
| defer blockTransactionsStmt.Close() | ||
|
|
||
| for i, txn := range txns { | ||
| result, err := transactionsStmt.Exec(dbEncode(txn.ID())) | ||
| if err != nil { | ||
| return fmt.Errorf("addTransactions: failed to insert into transactions: %v", err) | ||
| } | ||
| txnID, err := result.LastInsertId() | ||
| if err != nil { | ||
| return fmt.Errorf("addTransactions: failed to get insert result ID: %v", err) | ||
| } | ||
|
|
||
| if _, err := blockTransactionsStmt.Exec(dbEncode(bid), txnID, i); err != nil { | ||
| return fmt.Errorf("addTransactions: failed to insert into block_transactions: %v", err) | ||
| } else if err := s.addArbitraryData(dbTxn, txnID, txn); err != nil { | ||
| return fmt.Errorf("addTransactions: failed to add arbitrary data: %v", err) | ||
| } | ||
| } | ||
| return nil | ||
| } | ||
|
|
||
| func (s *Store) deleteBlock(dbTxn txn, bid types.BlockID) error { | ||
| _, err := dbTxn.Exec("DELETE FROM blocks WHERE id = ?", dbEncode(bid)) | ||
| return err | ||
| } | ||
|
|
||
| func (s *Store) applyUpdates() error { | ||
| return s.transaction(func(dbTxn txn) error { | ||
| for _, update := range s.pendingUpdates { | ||
| if err := s.addBlock(dbTxn, update.Block, update.State.Index.Height); err != nil { | ||
| return fmt.Errorf("applyUpdates: failed to add block: %v", err) | ||
| } else if err := s.addMinerPayouts(dbTxn, update.Block.ID(), update.Block.MinerPayouts); err != nil { | ||
| return fmt.Errorf("applyUpdates: failed to add miner payouts: %v", err) | ||
| } else if err := s.addTransactions(dbTxn, update.Block.ID(), update.Block.Transactions); err != nil { | ||
| return fmt.Errorf("applyUpdates: failed to add transactions: %v", err) | ||
| } | ||
| } | ||
| s.pendingUpdates = s.pendingUpdates[:0] | ||
| return nil | ||
| }) | ||
| } | ||
|
|
||
| func (s *Store) revertUpdate(cru *chain.RevertUpdate) error { | ||
| return s.transaction(func(dbTxn txn) error { | ||
| if err := s.deleteBlock(dbTxn, cru.Block.ID()); err != nil { | ||
| return fmt.Errorf("revertUpdate: failed to delete block: %v", err) | ||
| } | ||
| return nil | ||
| }) | ||
| } | ||
|
|
||
| // ProcessChainApplyUpdate implements chain.Subscriber. | ||
| func (s *Store) ProcessChainApplyUpdate(cau *chain.ApplyUpdate, mayCommit bool) error { | ||
| s.mu.Lock() | ||
| defer s.mu.Unlock() | ||
|
|
||
| s.pendingUpdates = append(s.pendingUpdates, cau) | ||
| if mayCommit { | ||
| return s.applyUpdates() | ||
| } | ||
| return nil | ||
| } | ||
|
|
||
| // ProcessChainRevertUpdate implements chain.Subscriber. | ||
| func (s *Store) ProcessChainRevertUpdate(cru *chain.RevertUpdate) error { | ||
| s.mu.Lock() | ||
| defer s.mu.Unlock() | ||
|
|
||
| if len(s.pendingUpdates) > 0 && s.pendingUpdates[len(s.pendingUpdates)-1].Block.ID() == cru.Block.ID() { | ||
| s.pendingUpdates = s.pendingUpdates[:len(s.pendingUpdates)-1] | ||
| return nil | ||
| } | ||
| return s.revertUpdate(cru) | ||
| } | ||
|
|
||
| // Tip implements explorer.Store. | ||
| func (s *Store) Tip() (result types.ChainIndex, err error) { | ||
| const query = `SELECT id, height FROM blocks ORDER BY height DESC LIMIT 1` | ||
| err = s.transaction(func(dbTx txn) error { | ||
| return dbTx.QueryRow(query).Scan(dbDecode(&result.ID), &result.Height) | ||
| }) | ||
| if errors.Is(err, sql.ErrNoRows) { | ||
| return types.ChainIndex{}, ErrNoTip | ||
| } | ||
| return | ||
| } |
Oops, something went wrong.