​​区块链持久化--golang实现区块链系列之三

本篇使用boltdb的简易数据库将区块链的数据持久化到一个文件中。boltdb是kv形式保存的。 下面代码:

package main

import (
    "github.com/boltdb/bolt"
    "os"
)

const dbfile  = "blockchain.db"
const blockBucket  = "block_demo"
const lastHashKey  = "genesis"

type BlockChian struct {
    //blocks []*Block
    db *bolt.DB
    lastHash []byte
}

func NewBlockChain() *BlockChian {
    //return &Blo ckChian{[]*Block{NewGenesisBlock()}}
    //func Open(path string, mode os.FileMode, options *Options) (*DB, error) {
    //  var db = &DB{opened: true}

    db, err := bolt.Open(dbfile, 7777, nil)
    CheckErr(err)
    var lasthash []byte
    //db.View()
    db.Update(func(tx *bolt.Tx) error {
        bucket := tx.Bucket([]byte(blockBucket))

        if bucket != nil {
            //读取lasthash
            lasthash = bucket.Get([]byte(lastHashKey))
        } else {
            // 1.创建bucket
            // 2. 创世区块
            genesis := NewGenesisBlock()
            bucket, err := tx.CreateBucket([]byte(blockBucket))
            CheckErr(err)

            err = bucket.Put(genesis.Hash, genesis.Serialize())
            CheckErr(err)
            lasthash = genesis.Hash
            err = bucket.Put([]byte(lastHashKey), genesis.Hash)
            CheckErr(err)
        }
        return nil
    })

    return &BlockChian{db:db, lastHash:lasthash}

}


func (bc *BlockChian)AddBlock(data string) {
    //var prevBlockHash  []byte
    /*err := bc.db.View(func(tx *bolt.Tx) error {
        bucket := tx.Bucket([]byte(blockBucket))
        lasthash := bucket.Get([]byte(lastHashKey))
        prevBlockHash = lasthash
        return nil
    })*/


    block := NewBlock(data, bc.lastHash)

    err := bc.db.Update(func(tx *bolt.Tx) error {
        bucket := tx.Bucket([]byte(blockBucket))
        err := bucket.Put(block.Hash, block.Serialize())
        CheckErr(err)

        err = bucket.Put([]byte(lastHashKey), block.Hash)
        CheckErr(err)
        return nil
    })

    CheckErr(err)
}

type BlockChainIterator struct {
    db *bolt.DB
    currentHash []byte
}


func (bc *BlockChian)Iterator() *BlockChainIterator {
    return &BlockChainIterator{bc.db, bc.lastHash}
}

func (it *BlockChainIterator)Next() *Block  {
    var block *Block


    err := it.db.View(func(tx *bolt.Tx) error {
        bucket := tx.Bucket([]byte(blockBucket))
        if bucket == nil {
            os.Exit(1)
        }

        blockTmp := bucket.Get(it.currentHash)
        block = Deserialize(blockTmp)

        it.currentHash = block.PrevBlockHash
        return nil

    })

    CheckErr(err)
    return block
}

boltdb是key-value形式的数据库, 所以,区块链保存hash当做key, 区块链的内容当做value进行保存。

上面有一个特殊不变的key保存最后一个区块的hash。

先获取最后区块的hash,就可以获取到区块的数据,获取数据就可以获取前一个区块的hash,以此递推,可以获取所有区块的数据。

命令行工具:

package main

import (
    "flag"
    "fmt"
    "os"
)

const Usage  = `
    addBlock --data DATA  "add a block to block chain"
    printChain            "print all blocks"
`

type CLI struct {
    bc *BlockChian
}

func (cli *CLI)Run()  {
    if len(os.Args) < 2 {
        fmt.Println("too few parameters!\n", Usage)
        os.Exit(1)
    }
    
    addBlockCmd := flag.NewFlagSet("addBlock", flag.ExitOnError)
    printChainCmd := flag.NewFlagSet("printChain", flag.ExitOnError)
    
    addBlockCmdPara := addBlockCmd.String("data", "", "block info")

    switch os.Args[1] {
    case "addBlock":
        err := addBlockCmd.Parse(os.Args[2:])
        CheckErr(err)
        if addBlockCmd.Parsed() {
            if *addBlockCmdPara == "" {
                fmt.Println("data is empty")
                os.Exit(1)
            }
            cli.AddBlock(*addBlockCmdPara)
        }
    case "printChain":
        err := printChainCmd.Parse(os.Args[2:])
        CheckErr(err)
        if printChainCmd.Parsed() {
            cli.PrintChain()
        }
    default:
        fmt.Println("invalid cmd\n", Usage)
        os.Exit(1)

    
    }
    
}

package main

import "fmt"

func (cli *CLI)AddBlock(data string)  {
    cli.bc.AddBlock(data)
}


func (cli *CLI)PrintChain()  {
    bc := cli.bc
    it := bc.Iterator()
    for {
        // 取回当前hash指定的block,并且将当前的hash指向上一个区块的hash
        block := it.Next()

        fmt.Println("data:", string(block.Data))
        fmt.Println("Version:", block.Version)
        fmt.Printf("Hash:%x\n", block.Hash)
        fmt.Printf("TimeStamp:%d\n", block.TimeStamp)
        fmt.Printf("MerkeRoot:%x\n", block.MerkeRoot)
        fmt.Printf("Nonce:%d\n", block.Nonce)
        fmt.Printf("preblock Hash:%x\n", block.PrevBlockHash)
        println("  ")

        pow := NewProofOfWork(block)
        fmt.Println("Vaild:", pow.IsVaild())

        if len(block.PrevBlockHash) == 0 {
            break
        }


    }
}

终端执行效果:

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏全沾开发(huā)

总结CSS3新特性(媒体查询篇)

总结CSS3新特性(媒体查询篇) CSS3的媒体查询是对CSS2媒体类型的扩展,完善; CSS2的媒体类型仅仅定义了一些设备的关键字,...

460100
来自专栏偏前端工程师的驿站

WebComponent魔法堂:深究Custom Element 之 标准构建

前言  通过《WebComponent魔法堂:深究Custom Element 之 面向痛点编程》,我们明白到其实Custom Element并不是什么新东西,...

208100
来自专栏mySoul

jQuery(一)

jquery为一种库,属于最基础的一个库,伴随着h5的到来,很多jquery的规则直接融入到了规则本身了,直接使用原生的js也能达到相同的目的。虽然如此,不过依...

27540
来自专栏keyWords

4、React组件之性能优化

优化并不嫌早。因为,对性能影响最关键的部分,往往涉及解决方案核心,决定整体的架构, 将来要改变的时候牵扯更大。

13010
来自专栏AhDung

【C#】使用IExtenderProvider为控件添加扩展属性,像ToolTip那样

- 文中的“控件”一词是广义上的说法,泛指包括ToolStripItem、MenuItem在内单个界面元素,并不特指继承自Control类的狭义控件

10720
来自专栏影子

jQuery中的常用内容总结(一)

34490
来自专栏编程

jQuery第八课

jQuery Callback 方法 Callback 函数在当前动画 100% 完成之后执行。 jQuery 动画的问题 许多 jQuery 函数涉及动画。这...

21060
来自专栏葡萄城控件技术团队

Mobile First! Wijmo 5 之 架构

本文就开发者关心的话题之一架构,展开叙述。 ? Wijmo 5是一组JavaScript控件,但是不要与Widgets混淆。在此前开发Wijmo的时候,我们能够...

247100
来自专栏JadePeng的技术博客

Angular快速学习笔记(3) -- 组件与模板

19930
来自专栏简书专栏

基于Excel2013的数据转换和清洗

数字可以被设成的格式有12种:常规、数值、货币、会计专用、日期、时间、百分比、分数、科学记数、文本、特殊、自定义

17920

扫码关注云+社区

领取腾讯云代金券