目录
flag基本使用
os.Args基本使用
flag与os.Args组合使用
通过命令行添加/查询区块
测试代码与测试结果
一. flag基本使用
通常我们在写命令行程序（工具、server）时，对命令参数进行解析是常见的需求。各种语言一般都会提供解析命令行参数的方法或库，以方便程序员使用。在 go 标准库中提供了一个包：flag，方便进行命令行解析。
1.导入flag包
import ( "flag")
2.使用示例
//定义一个字符串flag,flag名为printchain,默认值为:hello BTC world,参数说明: 输出所有的区块信息flagString := flag.String("printchain", "hello BTC world", "输出所有的区块信息")//定义一个整型flag,flag名为:number ,默认值为:6 ,参数说明:输入一个整数flagInt := flag.Int("number", 6, "输出一个整数...")//定义一个布尔类型的flag,flag名为:open,默认值:false,参数说明:判断真假flagBool := flag.Bool("open", false, "判断真假...")//解析flagflag.Parse()//输入参数后的值fmt.Printf("%s\n", *flagString)fmt.Printf("%d\n", *flagInt)fmt.Printf("%t\n", *flagBool)执行结果
//编译main.gogo build main.go
2.1 默认执行
./main //执行./main，输出所有flag的默认值hello BTC world6false
2.2 .输入一个未定义的flag
./main -xx //xx并未定义，输出错误信息以及所有flag使用方法(通过flag.Parse()生效)flag provided but not defined: -xx Usage of ./main: -number int 输出一个整数... (default 6) -open 判断真假... -printchain string 输出所有的区块信息 (default "hello BTC world")
2.3 输入已定义的flag
./main -printchain //仅仅传入flag,提示flag需要一个参数值flag needs an argument: -printchainUsage of ./main: -number int 输出一个整数... (default 6) -open 判断真假... -printchain string 输出所有的区块信息 (default "hello BTC world")
2.4 输入已定义的flag以及对应类型的值
./main -printchain "hello bruce" //传入flag=printchain,value="hello bruce"hello bruce6false./main -number 88 //传入flag=number,value=88hello BTC world88false./main -open true //传入flag=open,value=true (默认为false,传入-open即为true)hello BTC world6true
二.os.Args基本使用
os包提供了一些与操作系统交互的函数和变量，并且go对其做了一些封装。程序的命令行参数可以从os包的Args变量获取；os包外部使用os.Args访问该变量。
Go言里也采用左闭右开形式, 即，区间包括第一个索引元素，不包括最后一个, 因为这样可以简化逻辑。os.Args的第一个元素，os.Args[0], 是命令本身的名字；其它的元素则是程序启动时传给它的参数。
1.导入os包
import os
使用示例
//实例化os.Args
args := os.Args
//打印args切片所有内容
fmt.Printf("%v\n",args)
//打印args切片的第二个参数
fmt.Printf("%v\n",args[1])
//打印args切片的第三个参数
fmt.Printf("%v\n",args[2])
传入参数
./main arg1 arg2 //执行main程序，传入两个参数arg[./main arg1 arg2] //argsarg1 //args[1]arg2 //args[2]
三.flag与os.Args组合使用
1.创建flag对象
//通过flag.NewFlagSet实例化flag对象addBlockCmd := flag.NewFlagSet("addBlock", flag.ExitOnError)printChainCmd := flag.NewFlagSet("printChain", flag.ExitOnError)2.拼接flag
flag名称：data
默认值:bruce
使用说明:交易数据
使用方法： ./main addBlock -data
flagAddBlockData := addBlockCmd.String("data", "bruce", "交易数据")3.判断flag
//判断os.Args中第二个flag的值switch os.Args[1] {case "addBlock": err := addBlockCmd.Parse(os.Args[2:]) if err != nil { log.Panic(err) }case "printChain": err := printChainCmd.Parse(os.Args[2:]) if err != nil { log.Panic(err) }default: printUsage() os.Exit(1)}4.解析flag输出信息
if addBlockCmd.Parsed() { if *flagAddBlockData == "" { printUsage() os.Exit(1) } fmt.Println(*flagAddBlockData)}if printChainCmd.Parsed() { fmt.Println("测试输出区块信息...")}5.相关函数
5.1 打印使用说明
func printUsage() { fmt.Println("Usage:") fmt.Println("\taddblock -data DATA -- 交易数据") fmt.Println("\tprintChain --输出区块信息")}5.2 判断是否输入flag
func isValid() { //如果args切片长度小于2(即没有跟参数，则打印使用方法并退出程序) if len(os.Args) < 2 { printUsage() os.Exit(1) }}6.测试命令行
6.1 无参数
./mainUsage:addBlock -data DATA -- 交易数据:printChain --输出区块信息
6.2 无效参数
./main xxxUsage:addBlock -data DATA -- 交易数据:printChain --输出区块信息
6.3 参数默认值
./main addBlock //输出flag=data的默认值brucebruce ./main printChain //输出解析到printChain后打印的信息测试输出区块信息...
6.4 指定参数
./main addBlock -data "brucefeng" //输出flag=data的输入值brucefengbrucefeng
四.通过命令行添加/查询区块
1.实现目标
Usage: createblockchain -data DATA --交易数据 addBlock -data DATA -- 交易数据 printChain --输出区块信息
2.定义命令行属性与方法 CLI.go
2.1 导入相关包
import ( "fmt" "os" "flag" "log")
2.2 定义结构体
type CLI struct{}2.3 打印帮助提示
func printUsage() { fmt.Println("Usage:") fmt.Println("\tcreateblockchain -data DATA --交易数据") fmt.Println("\taddBlock -data DATA -- 交易数据") fmt.Println("\tprintChain --输出区块信息")}2.4 判断参数输入是否合法
func isValid() { if len(os.Args) < 2 { printUsage() os.Exit(1) }}2.5 定义创世区块的方法
func (cli *CLI) createGenenisBlockChain(data string) { CreateBlockChainWithGenesisBlock(data)}2.6 定义普通区块的方法
func (cli *CLI) addBlock(data string) {//判断数据库是否存在 if !DBExists() { fmt.Println("当前不存在区块链，请先创建创世区块") os.Exit(1) }//通过BlockChainObject()函数获取一个block blockchain := BlockChainObject() defer blockchain.DB.Close() //通过blockchain的AddBlockChain方法添加区块 blockchain.AddBlockChain(data)}2.7 定义遍历区块的方法
func (cli *CLI) printChain() {//判断数据库是否存在 if !DBExists() { fmt.Println("当前不存在区块链") os.Exit(1) } blockchain := BlockChainObject() defer blockchain.DB.Close() blockchain.PrintChain()}2.8 定义命令行参数方法集合
func (cli *CLI) Run() {//1.判断命令行参数是否合法 isValid()//2.通过flag.NewFlagSet实例化三个flag对象 //创建创世区块 createBlockCmd := flag.NewFlagSet("createblockchain", flag.ExitOnError) //创建普通区块 addBlockCmd := flag.NewFlagSet("addBlock", flag.ExitOnError) //遍历打印区块信息 printChainCmd := flag.NewFlagSet("printChain", flag.ExitOnError)//3.定义命令行对象的相关属性 flagCreateBlockChainWithData := createBlockCmd.String("data", "", "交易数据") flagAddBlockData := addBlockCmd.String("data", "", "交易数据")//4.根据参数值决定进行解析的信息 switch os.Args[1] { case "createblockchain": err := createBlockCmd.Parse(os.Args[2:]) if err != nil { log.Panic(err) } case "addBlock": err := addBlockCmd.Parse(os.Args[2:]) if err != nil { log.Panic(err) } case "printChain": err := printChainCmd.Parse(os.Args[2:]) if err != nil { log.Panic(err) } default: printUsage() os.Exit(1) }//5. 判断是否解析成功 Parsed reports whether f.Parse has been called. //5.1 如果createBlockCmd解析成功，输出的DATA的值作为创建创世区块的传入参数 if createBlockCmd.Parsed() { if *flagCreateBlockChainWithData == "" { fmt.Println("创世区块交易数据不能为空") printUsage() os.Exit(1) } //调用createGenenisBlockChain()方法创建创世区块 cli.createGenenisBlockChain(*flatCreateBlockChainWithData) } //5.2 如果addBlockCmd解析成功，输出的DATA的值作为创建新区块的传入参数 if addBlockCmd.Parsed() { if *flagAddBlockData == "" { fmt.Println("创建区块交易数据不能为空") printUsage() os.Exit(1) } //调用addBlock方法创建区块 cli.addBlock(*flagAddBlockData) } //5.3 如果printChainCmd解析成功，调用遍历区块链的方法 if printChainCmd.Parsed() { cli.printChain() }}2.9 代码整合
package BLCimport ( "fmt" "os" "flag" "log")type CLI struct{}func printUsage() { fmt.Println("Usage:") fmt.Println("\tcreateblockchain -data DATA --交易数据") fmt.Println("\taddBlock -data DATA -- 交易数据") fmt.Println("\tprintChain --输出区块信息")}func isValid() { if len(os.Args) < 2 { printUsage() os.Exit(1) }}func (cli *CLI) addBlock(data string) { if !DBExists() { fmt.Println("数据库不存在") os.Exit(1) } blockchain := BlockChainObject() defer blockchain.DB.Close() blockchain.AddBlockChain(data)}func (cli *CLI) printChain() { if !DBExists() { fmt.Println("数据库不存在") os.Exit(1) } blockchain := BlockChainObject() defer blockchain.DB.Close() blockchain.PrintChain()}func (cli *CLI) createGenenisBlockChain(data string) { CreateBlockChainWithGenesisBlock(data)}func (cli *CLI) Run() { isValid() addBlockCmd := flag.NewFlagSet("addBlock", flag.ExitOnError) printChainCmd := flag.NewFlagSet("printChain", flag.ExitOnError) createBlockCmd := flag.NewFlagSet("createblockchain", flag.ExitOnError) flagAddBlockData := addBlockCmd.String("data", "", "交易数据") flagCreateBlockChainWithData := createBlockCmd.String("data", "", "交易数据") switch os.Args[1] { case "addBlock": err := addBlockCmd.Parse(os.Args[2:]) if err != nil { log.Panic(err) } case "printChain": err := printChainCmd.Parse(os.Args[2:]) if err != nil { log.Panic(err) } case "createblockchain": err := createBlockCmd.Parse(os.Args[2:]) if err != nil { log.Panic(err) } default: printUsage() os.Exit(1) } if addBlockCmd.Parsed() { if *flagAddBlockData == "" { printUsage() os.Exit(1) } cli.addBlock(*flagAddBlockData) } if printChainCmd.Parsed() { cli.printChain() } if createBlockCmd.Parsed() { if *flagCreateBlockChainWithData == "" { fmt.Println("交易数据不能为空") printUsage() os.Exit(1) } cli.createGenenisBlockChain(*flagCreateBlockChainWithData) }}3.生成区块链方法改造 BlockChain.go
3.1 定义判断数据库是否存在的方法
func DBExists() bool { if _, err := os.Stat(dbName); os.IsNotExist(err) { return false } return true}3.2 定义返回BlockChain对象
func BlockChainObject() *BlockChain {//定义tip用于存储从数据库中获取到的最新区块的Hash值 var tip []byte //打开数据库 db, err := bolt.Open(dbName, 0600, nil) if err != nil { log.Fatal(err) } //通过key="l"获取value值(最新区块的Hash值) err = db.View(func(tx *bolt.Tx) error { //获取表对象 b := tx.Bucket([]byte(blockTableName)) if b != nil { //获取最新区块的Hash值 tip = b.Get([]byte("l")) } return nil }) //返回保存最新区块Hash信息的BlockChain对象 return &BlockChain{tip, db}}3.3 创建带有创世区块的区块链
func CreateBlockChainWithGenesisBlock(data string) { //判断数据库是否存在 if DBExists() { fmt.Println("创世区块已经存在") os.Exit(1) } //打开数据库 db, err := bolt.Open(dbName, 0600, nil) if err != nil { log.Fatal(err) } err = db.Update(func(tx *bolt.Tx) error { //创建数据库表 b, err := tx.CreateBucket([]byte(blockTableName)) if err != nil { log.Panic(err) } if b != nil { //创建创世区块 genesisBlock := CreateGenesisBlock(data) //将创世区块存储至表中 err := b.Put(genesisBlock.Hash, genesisBlock.Serialize()) if err != nil { log.Panic(err) } //存储最新的区块链的hash err = b.Put([]byte("l"), genesisBlock.Hash) if err != nil { log.Panic(err) } } return nil }) if err != nil { log.Fatal(err) }}3.4 创建添加区块的方法
func (blc *BlockChain) AddBlockChain(data string) { err := blc.DB.Update(func(tx *bolt.Tx) error { //1.获取表 b := tx.Bucket([]byte(blockTableName)) //2.创建新区块 if b != nil { //获取最新区块 byteBytes := b.Get(blc.Tip) //反序列化 block := DeserializeBlock(byteBytes) //3. 将区块序列化并且存储到数据库中 newBlock := NewBlock(data, block.Height+1, block.Hash) err := b.Put(newBlock.Hash, newBlock.Serialize()) if err != nil { log.Panic(err) } //4.更新数据库中"l"对应的Hash err = b.Put([]byte("l"), newBlock.Hash) if err != nil { log.Panic(err) } //5. 更新blockchain的Tip blc.Tip = newBlock.Hash } return nil }) if err != nil { log.Panic(err) }}3.5 遍历区块链的方法
func (blc *BlockChain) PrintChain() { blockchainIterator := blc.Iterator() for { block := blockchainIterator.Next() fmt.Printf("Height:%d\n", block.Height) fmt.Printf("PreBlockHash:%x\n", block.PreBlockHash) fmt.Printf("Data:%s\n", block.Data) fmt.Printf("TimeStamp:%s\n", time.Unix(block.TimeStamp, 0).Format("2006-01-02 03:04:05 PM")) fmt.Printf("Hash:%x\n", block.Hash) fmt.Printf("Nonce:%d\n", block.Nonce) var hashInt big.Int hashInt.SetBytes(block.PreBlockHash) if big.NewInt(0).Cmp(&hashInt) == 0 { break } }}3.6 代码整合
package BLCimport ( "github.com/boltdb/bolt" "log" "math/big" "fmt" "time" "os")const dbName = "blockchain.db" //数据库名const blockTableName = "blocks" //表名type BlockChain struct { Tip []byte //区块链里面最后一个区块的Hash DB *bolt.DB //数据库}//迭代器func (blockchain *BlockChain) Iterator() *BlockChainIterator { return &BlockChainIterator{blockchain.Tip, blockchain.DB}}//判断数据库是否存在func DBExists() bool { if _, err := os.Stat(dbName); os.IsNotExist(err) { return false } return true}func (blc *BlockChain) PrintChain() { blockchainIterator := blc.Iterator() for { block := blockchainIterator.Next() fmt.Printf("Height:%d\n", block.Height) fmt.Printf("PreBlockHash:%x\n", block.PreBlockHash) fmt.Printf("Data:%s\n", block.Data) fmt.Printf("TimeStamp:%s\n", time.Unix(block.TimeStamp, 0).Format("2006-01-02 03:04:05 PM")) fmt.Printf("Hash:%x\n", block.Hash) fmt.Printf("Nonce:%d\n", block.Nonce) var hashInt big.Int hashInt.SetBytes(block.PreBlockHash) if big.NewInt(0).Cmp(&hashInt) == 0 { break } }}//func (blc *BlockChain) AddBlockChain(data string) { err := blc.DB.Update(func(tx *bolt.Tx) error { //1.获取表 b := tx.Bucket([]byte(blockTableName)) //2.创建新区块 if b != nil { //获取最新区块 byteBytes := b.Get(blc.Tip) //反序列化 block := DeserializeBlock(byteBytes) //3. 将区块序列化并且存储到数据库中 newBlock := NewBlock(data, block.Height+1, block.Hash) err := b.Put(newBlock.Hash, newBlock.Serialize()) if err != nil { log.Panic(err) } //4.更新数据库中"l"对应的Hash err = b.Put([]byte("l"), newBlock.Hash) if err != nil { log.Panic(err) } //5. 更新blockchain的Tip blc.Tip = newBlock.Hash } return nil }) if err != nil { log.Panic(err) }}//1.创建带有创世区块的区块链func CreateBlockChainWithGenesisBlock(data string) { //判断数据库是否存在 if DBExists() { fmt.Println("创世区块已经存在") os.Exit(1) } //打开数据库 db, err := bolt.Open(dbName, 0600, nil) if err != nil { log.Fatal(err) } err = db.Update(func(tx *bolt.Tx) error { //创建数据库表 b, err := tx.CreateBucket([]byte(blockTableName)) if err != nil { log.Panic(err) } if b != nil { //创建创世区块 genesisBlock := CreateGenesisBlock(data) //将创世区块存储至表中 err := b.Put(genesisBlock.Hash, genesisBlock.Serialize()) if err != nil { log.Panic(err) } //存储最新的区块链的hash err = b.Put([]byte("l"), genesisBlock.Hash) if err != nil { log.Panic(err) } } return nil }) if err != nil { log.Fatal(err) }}//返回BlockChain对象func BlockChainObject() *BlockChain { var tip []byte //打开数据库 db, err := bolt.Open(dbName, 0600, nil) if err != nil { log.Fatal(err) } err = db.View(func(tx *bolt.Tx) error { b := tx.Bucket([]byte(blockTableName)) if b != nil { //读取最新区块的Hash tip = b.Get([]byte("l")) } return nil }) return &BlockChain{tip, db}}未改造代码块
Block.go
BlockChainItertor.go
help.go
ProofOfWork.go
以上未改造代码可以参考:【Golang区块链开发003】区块序列化存储
https://mp.weixin.qq.com/s?__biz=MzA4Mzg5NTEyOA==&mid=2651781600&idx=1&sn=ce72dd45079759dd004be83ed3f9f90c&chksm=84152fd7b362a6c18d1b752b8ffc14d07a057dd4f263c14c3058a822a7b74f7ae7bcb9eb1cd2#rd
五.测试代码与测试结果
测试代码main.go
func main() {
//初始化CLI对象
cli := BLC.CLI{}
fmt.Println("====返回结果======")
//执行Run命令调用对应参数对应的方法
cli.Run()
}
2.测试结果
2.1 无参数执行
./main //直接执行====返回结果======Usage: createblockchain -data DATA --交易数据 addBlock -data DATA -- 交易数据 printChain --输出区块信息
2.2 首次遍历区块链
./main printChain====返回结果======当前不存在区块链
2.3 首次添加普通区块
./main addBlock 或者 ./main addBlock -data "bruce"====返回结果======当前不存在区块链，请先创建创世区块
2.4 创建创世区块
./main createblockchain --data "Create Genenis Block 20180708"====返回结果======第1个区块，挖矿成功:0000d5b050b448b6db0d273b9320036f93b7cb2e8f8005c1e6192dc91b4a33812018-07-08 22:03:23.939771259 +0800 CST m=+0.087935424
2.5 遍历区块链
./main printChain====返回结果======Height:1PreBlockHash:0000000000000000000000000000000000000000000000000000000000000000Data:Create Genenis Block 20180708TimeStamp:2018-07-08 10:03:23 PMHash:0000d5b050b448b6db0d273b9320036f93b7cb2e8f8005c1e6192dc91b4a3381Nonce:61212
2.6 添加区块
./main addBlock -data "bruce"====返回结果======第2个区块，挖矿成功:000081872fe39a35be79e30f915c9716b9bbbae74f665fdd53f4ab57ae4b379d2018-07-08 22:06:27.642330532 +0800 CST m=+0.055636890
2.7 遍历区块链
./main printChain====返回结果======Height:2PreBlockHash:0000d5b050b448b6db0d273b9320036f93b7cb2e8f8005c1e6192dc91b4a3381Data:bruceTimeStamp:2018-07-08 10:06:27 PMHash:000081872fe39a35be79e30f915c9716b9bbbae74f665fdd53f4ab57ae4b379dNonce:39906Height:1PreBlockHash:0000000000000000000000000000000000000000000000000000000000000000Data:Create Genenis Block 20180708TimeStamp:2018-07-08 10:03:23 PMHash:0000d5b050b448b6db0d273b9320036f93b7cb2e8f8005c1e6192dc91b4a3381Nonce:61212
2.8 添加多个区块
./main addBlock -data "bruce"./main addBlock -data "send 100 BTC TO bruce"......第3个区块，挖矿成功:0000565d60b6a26b8fc238bfeffb2cc1de20d28ca2a312bef9601997bb914fc32018-07-08 22:09:52.644353193 +0800 CST m=+0.032078853第4个区块，挖矿成功:0000dd1354cc868b96aeb18360b320b7a30ea0c00b27b79c32c0d9269ff2dbb32018-07-08 22:10:22.805477297 +0800 CST m=+0.210806453
2.9 遍历整个区块链
./main printChain====返回结果======Height:6PreBlockHash:000019ba65a19b26f81fae22025e963e68a6cf0983fb211aae208c428368252dData:bruce send 30 BTC TO jackmaTimeStamp:2018-07-08 10:12:57 PMHash:0000e3ba0e34ace19cba574deefdf5b75315b2ce10e488332952544e5a056571Nonce:3502Height:5PreBlockHash:0000dd1354cc868b96aeb18360b320b7a30ea0c00b27b79c32c0d9269ff2dbb3Data:bruce send 50 BTC TO ponymaTimeStamp:2018-07-08 10:12:44 PMHash:000019ba65a19b26f81fae22025e963e68a6cf0983fb211aae208c428368252dNonce:80833Height:4PreBlockHash:0000565d60b6a26b8fc238bfeffb2cc1de20d28ca2a312bef9601997bb914fc3Data:send 100 BTC TO bruceTimeStamp:2018-07-08 10:10:22 PMHash:0000dd1354cc868b96aeb18360b320b7a30ea0c00b27b79c32c0d9269ff2dbb3Nonce:182779Height:3PreBlockHash:000081872fe39a35be79e30f915c9716b9bbbae74f665fdd53f4ab57ae4b379dData:bruceTimeStamp:2018-07-08 10:09:52 PMHash:0000565d60b6a26b8fc238bfeffb2cc1de20d28ca2a312bef9601997bb914fc3Nonce:21138Height:2PreBlockHash:0000d5b050b448b6db0d273b9320036f93b7cb2e8f8005c1e6192dc91b4a3381Data:bruceTimeStamp:2018-07-08 10:06:27 PMHash:000081872fe39a35be79e30f915c9716b9bbbae74f665fdd53f4ab57ae4b379dNonce:39906Height:1PreBlockHash:0000000000000000000000000000000000000000000000000000000000000000Data:Create Genenis Block 20180708TimeStamp:2018-07-08 10:03:23 PMHash:0000d5b050b448b6db0d273b9320036f93b7cb2e8f8005c1e6192dc91b4a3381Nonce:61212
2.10 测试重新创建区块结构
./main createblockchain -data "第二次创建创世区块"====返回结果======创世区块已经存在
随时随地看视频
