Implement Blockchain.AddBlock

blockchain.go

@@ -102,6 +102,24 @@ func (bc *Blockchain) Reset() {
 
 }
 
+// AddBlock saves the block into the blockchain
+func (bc *Blockchain) AddBlock(block *Block) {
+	err := bc.db.Update(func(tx *bolt.Tx) error {
+		b := tx.Bucket([]byte(blocksBucket))
+		blockInDb := b.Get(block.Hash)
+
+		if blockInDb != nil {
+			blockData := block.Serialize()
+			b.Put(block.Hash, blockData)
+		}
+
+		return nil
+	})
+	if err != nil {
+		log.Panic(err)
+	}
+}
+
 // FindTransaction finds a transaction by its ID
 func (bc *Blockchain) FindTransaction(ID []byte) (Transaction, error) {
 	bci := bc.Iterator()

server.go

@@ -183,7 +183,7 @@ func handleBlock(request []byte, bc *Blockchain) {
 	block := DeserializeBlock(blockData)
 
 	fmt.Println("Recevied a new block!")
-	fmt.Println(block)
+	bc.AddBlock(block)
 }
 
 func handleInv(request []byte, bc *Blockchain) {
@@ -201,8 +201,8 @@ func handleInv(request []byte, bc *Blockchain) {
 	blocks := bc.GetBlockHashes()
 
 	if len(blocks) < len(payload.Items) {
-		for _, blockHash := range invResponse.Items {
-			sendGetData(sourceNode, "block", blockHash)
+		for _, blockHash := range payload.Items {
+			sendGetData(payload.AddrFrom, "block", blockHash)
 		}
 	}
 }