梅克爾樹的結(jié)構(gòu)比較簡(jiǎn)單瓤湘,其設(shè)計(jì)思想比較巧妙瓢颅,它是SPV實(shí)現(xiàn)的關(guān)鍵。
梅克爾樹是二叉樹弛说,節(jié)點(diǎn)存儲(chǔ)哈希指針挽懦,葉子節(jié)點(diǎn)保存數(shù)據(jù)纺阔。
- Merkle Tree節(jié)點(diǎn)結(jié)構(gòu)體
type MerkleNode struct {
Left *MerkleNode
Right *MerkleNode
Data []byte
}
- 創(chuàng)建Merkle Tree節(jié)點(diǎn)
func NewMerkleNode(left,right *MerkleNode, data []byte) *MerkleNode {
mNode := MerkleNode{}
if left == nil && right == nil {
hash := sha256.Sum256(data)
mNode.Data = hash[:]
}else {
prevHashes := append(left.Data,right.Data...)
hash := sha256.Sum256(prevHashes)
mNode.Data = hash[:]
}
mNode.Left = left
mNode.Right = right
return &mNode
}
- 構(gòu)建Merkle Tree
func NewMerkleTree(data [][]byte) *MerkleTree {
var nodes []MerkleNode
if len(data) % 2 != 0 {
data = append(data, data[len(data) - 1])
}
for _, dataitem := range data {
node := NewMerkleNode(nil, nil, dataitem)
nodes = append(nodes, *node)
}
for i := 0; i<len(data)/2; i++ {
var newNodes []MerkleNode
for j := 0; j < len(nodes); j += 2 {
node := NewMerkleNode(&nodes[j], &nodes[j+1], nil)
newNodes = append(newNodes , *node)
}
nodes = newNodes
}
mTree := MerkleTree{&nodes[0]}
return &mTree
}