专栏首页友弟技术工作室​​区块链持久化--golang实现区块链系列之三

​​区块链持久化--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 条评论
登录 后参与评论

相关文章

  • Docker镜像的基本操作

    若与
  • docker容器基本操作

    若与
  • Mac开发环境配置之HomeBrew

    如果你使用惯了linux,你会对软件包管理、命令行不能自拔。由于mac的底层是类unix系统,也可以配置向linux一样好用。 ? 安装HomeBrew /us...

    若与
  • docker安装Hadoop

    https://docs.docker.com/toolbox/toolbox_install_mac/ 安装好Toolbox后,在启动台多出Kitemati...

    birdskyws
  • 分析师:惠普分拆后是为了更专注用户,还是为了应对联想和戴尔

    惠普终于确定将公司分为 HP Inc .与 HP Enterprise ,前者主力产品为 PC 与打印机,后者为专攻企业市场的服务器、储存设备、软件与服务,有分...

    人称T客
  • nodejs使用superagent爬取网站内容中文乱码的解决方法

    使用superagent爬取网站内容,当网页编码不是utf-8编码时,中文就会返回乱码,原因是superagent只支持utf-8的网页编码,我们可以使用其扩展...

    飞奔去旅行
  • 删除docker的运行目录出错

    1.1.        rm: cannot remove '/var/lib/docker/aufs': Device or resource busy

    一夕如环
  • 编写你的第一个 Flutter App

    Flutter 是 Google 用以帮助开发者在 iOS 和 Andorid 两个平台开发高质量 UI 的移动 SDK,免费且开源。为了体验 Flutter ...

    icepy
  • Kubernetes中的emptyDir存储卷和节点存储卷

    Kubernetes支持存储卷类型中,emptyDir存储卷的生命周期与其所属的Pod对象相同,它无法脱离Pod对象的生命周期提供数据存储功能,因此emptyD...

    小小科
  • Django之模板系统

      当模版引擎遇到一个变量,它将计算这个变量,然后用结果替换掉它本身。 变量的命名包括任何字母数字以及下划线 ("_")的组合。 变量名称中不能有空格或标点符号...

    py3study

扫码关注云+社区

领取腾讯云代金券