diff --git a/rskj-core/src/main/java/co/rsk/peg/federation/FederationContext.java b/rskj-core/src/main/java/co/rsk/peg/federation/FederationContext.java index 31d855abdb..f7c0091f87 100644 --- a/rskj-core/src/main/java/co/rsk/peg/federation/FederationContext.java +++ b/rskj-core/src/main/java/co/rsk/peg/federation/FederationContext.java @@ -8,11 +8,13 @@ public class FederationContext { private final Federation activeFederation; - private Federation retiringFederation; - private Script lastRetiredFederationP2SHScript; + private final Federation retiringFederation; + private final Script lastRetiredFederationP2SHScript; - public FederationContext(Federation activeFederation) { - this.activeFederation = activeFederation; + private FederationContext(FederationContextBuilder federationContextBuilder) { + this.activeFederation = federationContextBuilder.activeFederation; + this.retiringFederation = federationContextBuilder.retiringFederation; + this.lastRetiredFederationP2SHScript = federationContextBuilder.lastRetiredFederationP2SHScript; } public Federation getActiveFederation() { @@ -23,21 +25,43 @@ public Optional getRetiringFederation() { return Optional.ofNullable(retiringFederation); } - public void setRetiringFederation(Federation retiringFederation) { - this.retiringFederation = retiringFederation; - } - public Optional