com.wavesplatform.history.History.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of node Show documentation
node
The newest version!
package com.wavesplatform.history

import com.wavesplatform.block.{Block, MicroBlock}
import com.wavesplatform.common.state.ByteStr
import com.wavesplatform.database
import com.wavesplatform.database.RDB
import com.wavesplatform.protobuf.PBSnapshots
import com.wavesplatform.protobuf.snapshot.TransactionStateSnapshot
import com.wavesplatform.state.{Blockchain, Height, StateSnapshot}

trait History {
  def loadBlockBytes(id: ByteStr): Option[(Byte, Array[Byte])]
  def loadMicroBlock(id: ByteStr): Option[MicroBlock]
  def blockIdsAfter(candidates: Seq[ByteStr], count: Int): Seq[ByteStr]
  def loadBlockSnapshots(id: ByteStr): Option[Seq[TransactionStateSnapshot]]
  def loadMicroBlockSnapshots(id: ByteStr): Option[Seq[TransactionStateSnapshot]]
}

object History {
  private def versionedBytes(block: Block): (Byte, Array[Byte]) = block.header.version -> block.bytes()

  def apply(
      blockchain: Blockchain,
      liquidBlock: ByteStr => Option[Block],
      microBlock: ByteStr => Option[MicroBlock],
      liquidBlockSnapshot: ByteStr => Option[StateSnapshot],
      microBlockSnapshot: ByteStr => Option[StateSnapshot],
      rdb: RDB
  ): History =
    new History {
      override def loadBlockBytes(id: ByteStr): Option[(Byte, Array[Byte])] =
        liquidBlock(id)
          .orElse(blockchain.heightOf(id).flatMap { h =>
            database.loadBlock(Height(h), rdb)
          })
          .map(versionedBytes)

      override def loadMicroBlock(id: ByteStr): Option[MicroBlock] = microBlock(id)

      override def blockIdsAfter(candidates: Seq[ByteStr], count: Int): Seq[ByteStr] =
        candidates.view.flatMap(blockchain.heightOf).headOption.fold[Seq[ByteStr]](Seq.empty) { firstCommonHeight =>
          (firstCommonHeight to firstCommonHeight + count).flatMap(blockchain.blockId)
        }

      override def loadBlockSnapshots(id: ByteStr): Option[Seq[TransactionStateSnapshot]] =
        liquidBlockSnapshot(id)
          .map(_.transactions.values.toSeq.map(txInfo => PBSnapshots.toProtobuf(txInfo.snapshot, txInfo.status)))
          .orElse(blockchain.heightOf(id).map { h =>
            database.loadTxStateSnapshots(Height(h), rdb)
          })

      override def loadMicroBlockSnapshots(id: ByteStr): Option[Seq[TransactionStateSnapshot]] =
        microBlockSnapshot(id)
          .map(_.transactions.values.toSeq.map { txInfo =>
            PBSnapshots.toProtobuf(txInfo.snapshot, txInfo.status)
          })
    }
}