add GetBalance to chain

RichardWeiYang · Oct 22, 2017 · 56c9e3f · 56c9e3f
1 parent 9b2f73b
commit 56c9e3f
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/blockchain.go b/blockchain.go
@@ -376,3 +376,16 @@ func (bc *Blockchain) PrintHTML() string {
 	return strings.Join(lines, "\n")
 
 }
+
+func (bc *Blockchain) GetBalance(address string) int {
+	UTXOSet := UTXOSet{bc}
+	balance := 0
+	pubKeyHash := Base58Decode([]byte(address))
+	pubKeyHash = pubKeyHash[1 : len(pubKeyHash)-4]
+	UTXOs := UTXOSet.FindUTXO(pubKeyHash)
+
+	for _, out := range UTXOs {
+		balance += out.Value
+	}
+	return balance
+}