Remove default parameters from DataMessageHandler (#4918)

* Remove default parameters from DataMessageHandler

* Remove syncPeer default parameter
This commit is contained in:
Chris Stewart 2022-12-12 15:46:34 -06:00 committed by GitHub
parent 424ea33541
commit aab2b274f0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 82 additions and 45 deletions

View file

@ -38,14 +38,18 @@ class DataMessageHandlerTest extends NodeUnitTest with CachedTor {
val senderF = node.peerMsgSenders(0)
for {
chainApi <- node.chainApiFromDb()
dataMessageHandler = DataMessageHandler(chainApi,
None,
node,
HeaderSync,
syncPeer = Some(peer))(
node.executionContext,
node.nodeAppConfig,
node.chainConfig)
dataMessageHandler = DataMessageHandler(
chainApi = chainApi,
walletCreationTimeOpt = None,
node = node,
state = HeaderSync,
initialSyncDone = None,
currentFilterBatch = Vector.empty,
filterHeaderHeightOpt = None,
filterHeightOpt = None,
syncing = true,
syncPeer = Some(peer)
)(node.executionContext, node.nodeAppConfig, node.chainConfig)
// Use signet genesis block header, this should be invalid for regtest
invalidPayload =
@ -86,13 +90,18 @@ class DataMessageHandlerTest extends NodeUnitTest with CachedTor {
_ = node.nodeAppConfig.addCallbacks(nodeCallbacks)
dataMessageHandler =
DataMessageHandler(genesisChainApi,
None,
node,
HeaderSync,
syncPeer = Some(peer))(node.executionContext,
node.nodeAppConfig,
node.chainConfig)
DataMessageHandler(
chainApi = genesisChainApi,
walletCreationTimeOpt = None,
node = node,
state = HeaderSync,
initialSyncDone = None,
currentFilterBatch = Vector.empty,
filterHeaderHeightOpt = None,
filterHeightOpt = None,
syncing = true,
syncPeer = Some(peer)
)(node.executionContext, node.nodeAppConfig, node.chainConfig)
sender <- senderF
_ <- dataMessageHandler.handleDataPayload(payload, sender, peer)
result <- resultP.future
@ -126,13 +135,18 @@ class DataMessageHandlerTest extends NodeUnitTest with CachedTor {
_ = node.nodeAppConfig.addCallbacks(callbacks)
dataMessageHandler =
DataMessageHandler(genesisChainApi,
None,
node,
HeaderSync,
syncPeer = Some(peer))(node.executionContext,
node.nodeAppConfig,
node.chainConfig)
DataMessageHandler(
chainApi = genesisChainApi,
walletCreationTimeOpt = None,
node = node,
state = HeaderSync,
initialSyncDone = None,
currentFilterBatch = Vector.empty,
filterHeaderHeightOpt = None,
filterHeightOpt = None,
syncing = true,
syncPeer = Some(peer)
)(node.executionContext, node.nodeAppConfig, node.chainConfig)
sender <- senderF
_ <- dataMessageHandler.handleDataPayload(payload, sender, peer)
result <- resultP.future
@ -164,13 +178,18 @@ class DataMessageHandlerTest extends NodeUnitTest with CachedTor {
nodeCallbacks = NodeCallbacks.onCompactFilterReceived(callback)
_ = node.nodeAppConfig.addCallbacks(nodeCallbacks)
dataMessageHandler =
DataMessageHandler(genesisChainApi,
None,
node,
HeaderSync,
syncPeer = Some(peer))(node.executionContext,
node.nodeAppConfig,
node.chainConfig)
DataMessageHandler(
chainApi = genesisChainApi,
walletCreationTimeOpt = None,
node = node,
state = HeaderSync,
initialSyncDone = None,
currentFilterBatch = Vector.empty,
filterHeaderHeightOpt = None,
filterHeightOpt = None,
syncing = true,
syncPeer = Some(peer)
)(node.executionContext, node.nodeAppConfig, node.chainConfig)
sender <- senderF
_ <- dataMessageHandler.handleDataPayload(payload, sender, peer)
result <- resultP.future
@ -203,13 +222,18 @@ class DataMessageHandlerTest extends NodeUnitTest with CachedTor {
_ = node.nodeAppConfig.addCallbacks(nodeCallbacks)
dataMessageHandler =
DataMessageHandler(genesisChainApi,
None,
node,
HeaderSync,
syncPeer = Some(peer))(node.executionContext,
node.nodeAppConfig,
node.chainConfig)
DataMessageHandler(
chainApi = genesisChainApi,
walletCreationTimeOpt = None,
node = node,
state = HeaderSync,
initialSyncDone = None,
currentFilterBatch = Vector.empty,
filterHeaderHeightOpt = None,
filterHeightOpt = None,
syncing = true,
syncPeer = Some(peer)
)(node.executionContext, node.nodeAppConfig, node.chainConfig)
sender <- senderF
_ <- dataMessageHandler.handleDataPayload(payload, sender, peer)
result <- resultP.future

View file

@ -1,5 +1,6 @@
package org.bitcoins.node
import akka.Done
import akka.actor.ActorSystem
import org.bitcoins.chain.config.ChainAppConfig
import org.bitcoins.chain.models.BlockHeaderDAO
@ -22,7 +23,7 @@ import org.bitcoins.node.networking.peer.{
}
import java.time.Instant
import scala.concurrent.Future
import scala.concurrent.{Future, Promise}
case class NeutrinoNode(
chainApi: ChainApi,
@ -44,8 +45,20 @@ case class NeutrinoNode(
val controlMessageHandler: ControlMessageHandler = ControlMessageHandler(this)
private var dataMessageHandler: DataMessageHandler =
DataMessageHandler(chainApi, walletCreationTimeOpt, this, HeaderSync)
private var dataMessageHandler: DataMessageHandler = {
DataMessageHandler(
chainApi = chainApi,
walletCreationTimeOpt = walletCreationTimeOpt,
node = this,
state = HeaderSync,
initialSyncDone = Some(Promise[Done]()),
currentFilterBatch = Vector.empty,
filterHeaderHeightOpt = None,
filterHeightOpt = None,
syncing = false,
syncPeer = None
)
}
override def getDataMessageHandler: DataMessageHandler = dataMessageHandler

View file

@ -32,12 +32,12 @@ case class DataMessageHandler(
walletCreationTimeOpt: Option[Instant],
node: Node,
state: DataMessageHandlerState,
initialSyncDone: Option[Promise[Done]] = None,
currentFilterBatch: Vector[CompactFilterMessage] = Vector.empty,
filterHeaderHeightOpt: Option[Int] = None,
filterHeightOpt: Option[Int] = None,
syncing: Boolean = false,
syncPeer: Option[Peer] = None)(implicit
initialSyncDone: Option[Promise[Done]],
currentFilterBatch: Vector[CompactFilterMessage],
filterHeaderHeightOpt: Option[Int],
filterHeightOpt: Option[Int],
syncing: Boolean,
syncPeer: Option[Peer])(implicit
ec: ExecutionContext,
appConfig: NodeAppConfig,
chainConfig: ChainAppConfig)