Send consensus valid/invalid messages

This commit is contained in:
Paul Hauner
2021-09-27 12:34:01 +10:00
parent 3c816a3533
commit 82d491c88d
3 changed files with 27 additions and 8 deletions

View File

@@ -5,18 +5,21 @@ use types::Hash256;
pub struct ExecutePayloadHandle {
pub(crate) block_hash: Hash256,
pub(crate) execution_layer: ExecutionLayer,
pub(crate) status: Option<ConsensusStatus>,
}
impl ExecutePayloadHandle {
pub fn publish_consensus_valid(self) {
pub fn publish_consensus_valid(mut self) {
self.publish(ConsensusStatus::Valid)
}
pub fn publish_consensus_invalid(self) {
pub fn publish_consensus_invalid(mut self) {
self.publish(ConsensusStatus::Invalid)
}
fn publish(&self, status: ConsensusStatus) {
fn publish(&mut self, status: ConsensusStatus) {
self.status = Some(status);
if let Err(e) = self.execution_layer.block_on(|execution_layer| {
execution_layer.consensus_validated(self.block_hash, status)
}) {
@@ -34,6 +37,8 @@ impl ExecutePayloadHandle {
impl Drop for ExecutePayloadHandle {
fn drop(&mut self) {
self.publish(ConsensusStatus::Invalid)
if self.status.is_none() {
self.publish(ConsensusStatus::Invalid)
}
}
}

View File

@@ -152,6 +152,7 @@ impl ExecutionLayer {
let execute_payload_handle = ExecutePayloadHandle {
block_hash: execution_payload.block_hash,
execution_layer: self.clone(),
status: None,
};
Ok((execute_payload_response, execute_payload_handle))