Allow Attached rules to be flagged 'light' (PegaSysEng#580)

vinistevam · Jan 24, 2019 · 7f16827 · 7f16827
1 parent 6e66f79
commit 7f16827
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/...c/main/java/tech/pegasys/pantheon/ethereum/mainnet/AttachedBlockHeaderValidationRule.java b/...c/main/java/tech/pegasys/pantheon/ethereum/mainnet/AttachedBlockHeaderValidationRule.java
@@ -26,4 +26,8 @@ public interface AttachedBlockHeaderValidationRule<C> {
    * @return {@code true} if valid; otherwise {@code false}
    */
   boolean validate(BlockHeader header, BlockHeader parent, ProtocolContext<C> protocolContext);
+
+  default boolean includeInLightValidation() {
+    return true;
+  }
 }
diff --git a/ethereum/core/src/main/java/tech/pegasys/pantheon/ethereum/mainnet/BlockHeaderValidator.java b/ethereum/core/src/main/java/tech/pegasys/pantheon/ethereum/mainnet/BlockHeaderValidator.java
@@ -120,7 +120,7 @@ public static class Builder<C> {
     private final List<Rule<C>> rules = new ArrayList<>();
 
     public Builder<C> addRule(final AttachedBlockHeaderValidationRule<C> rule) {
-      this.rules.add(new Rule<>(false, rule, true));
+      this.rules.add(new Rule<>(false, rule, rule.includeInLightValidation()));
       return this;
     }