Skip to content

Commit

Permalink
feat: handle exited/exiting validators during top up (#6880)
Browse files Browse the repository at this point in the history
* Handle exiting validataor

* lint

* Add todo
  • Loading branch information
ensi321 authored and g11tech committed Jul 1, 2024
1 parent 2a98d3f commit 76559dd
Showing 1 changed file with 33 additions and 5 deletions.
Original file line number Diff line number Diff line change
@@ -1,26 +1,49 @@
import {FAR_FUTURE_EPOCH} from "@lodestar/params";
import {CachedBeaconStateElectra} from "../types.js";
import {increaseBalance} from "../util/balance.js";
import {getActivationExitChurnLimit} from "../util/validator.js";
import {getCurrentEpoch} from "../util/epoch.js";

/**
* Starting from Electra:
* Process pending balance deposits from state subject to churn limit and depsoitBalanceToConsume.
* For each eligible `deposit`, call `increaseBalance()`.
* Remove the processed deposits from `state.pendingBalanceDeposits`.
* Update `state.depositBalanceToConsume` for the next epoch
*
* TODO Electra: Update ssz library to support batch push to `pendingBalanceDeposits`
*/
export function processPendingBalanceDeposits(state: CachedBeaconStateElectra): void {
const availableForProcessing = state.depositBalanceToConsume + BigInt(getActivationExitChurnLimit(state));
const currentEpoch = getCurrentEpoch(state);
let processedAmount = 0n;
let nextDepositIndex = 0;
const depositsToPostpone = [];

for (const deposit of state.pendingBalanceDeposits.getAllReadonly()) {
const {amount} = deposit;
if (processedAmount + amount > availableForProcessing) {
break;
const {amount, index: depositIndex} = deposit;
const validator = state.validators.get(depositIndex);

// Validator is exiting, postpone the deposit until after withdrawable epoch
if (validator.exitEpoch < FAR_FUTURE_EPOCH) {
if (currentEpoch <= validator.withdrawableEpoch) {
depositsToPostpone.push(deposit);
} else {
// Deposited balance will never become active. Increase balance but do not consume churn
increaseBalance(state, deposit.index, Number(amount));
}
} else {
// Validator is not exiting, attempt to process deposit
if (processedAmount + amount > availableForProcessing) {
// Deposit does not fit in the churn, no more deposit processing in this epoch.
break;
} else {
// Deposit fits in the churn, process it. Increase balance and consume churn.
increaseBalance(state, deposit.index, Number(amount));
processedAmount = processedAmount + amount;
}
}
increaseBalance(state, deposit.index, Number(amount));
processedAmount = processedAmount + amount;
// Regardless of how the deposit was handled, we move on in the queue.
nextDepositIndex++;
}

Expand All @@ -32,4 +55,9 @@ export function processPendingBalanceDeposits(state: CachedBeaconStateElectra):
} else {
state.depositBalanceToConsume = availableForProcessing - processedAmount;
}

// TODO Electra: add a function in ListCompositeTreeView to support batch push operation
for (const deposit of depositsToPostpone) {
state.pendingBalanceDeposits.push(deposit);
}
}

0 comments on commit 76559dd

Please sign in to comment.