interesting
/
blockchain_go
mirror of https://github.com/Jeiwan/blockchain_go.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Store a DB connection in Blockchain

This commit is contained in:
Ivan Kuznetsov 2017-08-28 16:28:23 +07:00
parent e6cf54a124
commit a93e1e96c9
3 changed files with 23 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
blockchain.db
View File

Binary file not shown.

40
blockchain.go
View File

@ -12,33 +12,33 @@ const blocksBucket = "blocks"
// Blockchain keeps a sequence of Blocks // Blockchain keeps a sequence of Blocks
type Blockchain struct { type Blockchain struct {
tip []byte tip []byte
db *bolt.DB
} }
// BlockchainIterator is used to iterate over blockchain blocks // BlockchainIterator is used to iterate over blockchain blocks
type BlockchainIterator struct { type BlockchainIterator struct {
currentHash []byte currentHash []byte
db *bolt.DB
} }
// AddBlock saves provided data as a block in the blockchain // AddBlock saves provided data as a block in the blockchain
func (bc *Blockchain) AddBlock(data string) { func (bc *Blockchain) AddBlock(data string) {
var lastHash []byte var lastHash []byte
db, err := bolt.Open(dbFile, 0600, nil) err := bc.db.View(func(tx *bolt.Tx) error {
if err != nil {
log.Panic(err)
}
defer db.Close()
err = db.View(func(tx *bolt.Tx) error {
b := tx.Bucket([]byte(blocksBucket)) b := tx.Bucket([]byte(blocksBucket))
lastHash = b.Get([]byte("l")) lastHash = b.Get([]byte("l"))
return nil return nil
}) })
if err != nil {
log.Panic(err)
}
newBlock := NewBlock(data, lastHash) newBlock := NewBlock(data, lastHash)
err = db.Update(func(tx *bolt.Tx) error { err = bc.db.Update(func(tx *bolt.Tx) error {
b := tx.Bucket([]byte(blocksBucket)) b := tx.Bucket([]byte(blocksBucket))
err := b.Put(newBlock.Hash, newBlock.Serialize()) err := b.Put(newBlock.Hash, newBlock.Serialize())
if err != nil { if err != nil {
@ -58,7 +58,7 @@ func (bc *Blockchain) AddBlock(data string) {
// Iterator ... // Iterator ...
func (bc *Blockchain) Iterator() *BlockchainIterator { func (bc *Blockchain) Iterator() *BlockchainIterator {
bci := &BlockchainIterator{bc.tip} bci := &BlockchainIterator{bc.tip, bc.db}
return bci return bci
} }
@ -66,13 +66,8 @@ func (bc *Blockchain) Iterator() *BlockchainIterator {
// Next returns next block starting from the tip // Next returns next block starting from the tip
func (i *BlockchainIterator) Next() *Block { func (i *BlockchainIterator) Next() *Block {
var block *Block var block *Block
db, err := bolt.Open(dbFile, 0600, nil)
if err != nil {
log.Panic(err)
}
defer db.Close()
err = db.View(func(tx *bolt.Tx) error { err := i.db.View(func(tx *bolt.Tx) error {
b := tx.Bucket([]byte(blocksBucket)) b := tx.Bucket([]byte(blocksBucket))
encodedBlock := b.Get(i.currentHash) encodedBlock := b.Get(i.currentHash)
block = DeserializeBlock(encodedBlock) block = DeserializeBlock(encodedBlock)
@ -80,6 +75,10 @@ func (i *BlockchainIterator) Next() *Block {
return nil return nil
}) })
if err != nil {
log.Panic(err)
}
i.currentHash = block.PrevBlockHash i.currentHash = block.PrevBlockHash
return block return block
@ -87,12 +86,11 @@ func (i *BlockchainIterator) Next() *Block {
// NewBlockchain creates a new Blockchain with genesis Block // NewBlockchain creates a new Blockchain with genesis Block
func NewBlockchain() *Blockchain { func NewBlockchain() *Blockchain {
bc := Blockchain{} var tip []byte
db, err := bolt.Open(dbFile, 0600, nil) db, err := bolt.Open(dbFile, 0600, nil)
if err != nil { if err != nil {
log.Panic(err) log.Panic(err)
} }
defer db.Close()
err = db.Update(func(tx *bolt.Tx) error { err = db.Update(func(tx *bolt.Tx) error {
b := tx.Bucket([]byte(blocksBucket)) b := tx.Bucket([]byte(blocksBucket))
@ -114,17 +112,19 @@ func NewBlockchain() *Blockchain {
if err != nil { if err != nil {
log.Panic(err) log.Panic(err)
} }
bc.tip = genesis.Hash tip = genesis.Hash
} else { } else {
tip := b.Get([]byte("l")) tip = b.Get([]byte("l"))
bc.tip = tip
} }
return nil return nil
}) })
if err != nil { if err != nil {
log.Panic(err) log.Panic(err)
} }
bc := Blockchain{tip, db}
return &bc return &bc
} }

5
main.go
View File

@ -7,9 +7,10 @@ import (
func main() { func main() {
bc := NewBlockchain() bc := NewBlockchain()
defer bc.db.Close()
bc.AddBlock("Send 1 BTC to Ivan") // bc.AddBlock("Send 1 BTC to Ivan")
bc.AddBlock("Send 2 more BTC to Ivan") // bc.AddBlock("Send 2 more BTC to Ivan")
bci := bc.Iterator() bci := bc.Iterator()