update.go

package blockcreator

import (
	"fmt"

	"github.com/threefoldtech/rivine/modules"
	"github.com/threefoldtech/rivine/pkg/encoding/siabin"
	"github.com/threefoldtech/rivine/types"
)

// ProcessConsensusChange will update the blockcreator's most recent block.
func (b *BlockCreator) ProcessConsensusChange(cc modules.ConsensusChange) {
	b.mu.Lock()
	defer b.mu.Unlock()

	// Update the block creator's understanding of the block height.
	for _, block := range cc.RevertedBlocks {
		// Only doing the block check if the height is above zero saves hashing
		// and saves a nontrivial amount of time during IBD.
		if b.persist.Height > 0 || block.ID() != b.genesisID {
			b.persist.Height--
		} else if b.persist.Height != 0 {
			// Sanity check - if the current block is the genesis block, the
			// blockcreator height should be set to zero.
			b.log.Critical("BlockCreator has detected a genesis block, but the height of the block creator is set to ", b.persist.Height)
			b.persist.Height = 0
		}
	}
	for _, block := range cc.AppliedBlocks {
		// Only doing the block check if the height is above zero saves hashing
		// and saves a nontrivial amount of time during IBD.
		if b.persist.Height > 0 || block.ID() != b.genesisID {
			b.persist.Height++
		} else if b.persist.Height != 0 {
			// Sanity check - if the current block is the genesis block, the
			// block creator height should be set to zero.
			b.log.Critical("BlockCreator has detected a genesis block, but the height of the block creator is set to ", b.persist.Height)
			b.persist.Height = 0
		}
	}

	// Update the unsolved block.
	b.unsolvedBlock.ParentID = cc.AppliedBlocks[len(cc.AppliedBlocks)-1].ID()

	b.persist.RecentChange = cc.ID
	b.persist.ParentID = b.unsolvedBlock.ParentID
	err := b.save()
	if err != nil {
		b.log.Println(err)
	}

}

// ReceiveUpdatedUnconfirmedTransactions will replace the current unconfirmed
// set of transactions with the input transactions.
func (b *BlockCreator) ReceiveUpdatedUnconfirmedTransactions(unconfirmedTransactions []types.Transaction, _ modules.ConsensusChange) error {
	b.mu.Lock()
	defer b.mu.Unlock()
	// Edge case - if there are no transactions, set the block's transactions
	// to nil and return.
	if len(unconfirmedTransactions) == 0 {
		b.unsolvedBlock.Transactions = nil
		return nil
	}

	// Add transactions to the block until the block size limit is reached.
	// Transactions are assumed to be in a sensible order.
	var i int
	remainingSize := int(b.chainCts.BlockSizeLimit - 5e3) //check this 5k for the first extra
	var (
		err     error
		txBytes []byte
	)
	for i = range unconfirmedTransactions {
		txBytes, err = siabin.Marshal(unconfirmedTransactions[i])
		if err != nil {
			return fmt.Errorf("failed to (siabin) marshal tx %d: %v", i, err)
		}
		remainingSize -= len(txBytes)
		if remainingSize < 0 {
			break
		}
	}
	b.unsolvedBlock.Transactions = unconfirmedTransactions[:i+1]
	return nil
}