@aztec/slasher 0.0.1-commit.f5d02921e → 0.0.1-commit.f650c0a5c

package/src/stores/payloads_store.ts

@@ -1,149 +0,0 @@
-import { EthAddress } from '@aztec/foundation/eth-address';
-import type { AztecAsyncKVStore, AztecAsyncMap } from '@aztec/kv-store';
-import {
-  type SlashPayload,
-  type SlashPayloadRound,
-  deserializeSlashPayload,
-  serializeSlashPayload,
-} from '@aztec/stdlib/slashing';
-
-export class SlasherPayloadsStore {
-  /** Map from payload address to payload data */
-  private payloads: AztecAsyncMap<string, Buffer>;
-
-  /** Map from `round:payload` to votes */
-  private roundPayloadVotes: AztecAsyncMap<string, bigint>;
-
-  constructor(
-    private kvStore: AztecAsyncKVStore,
-    private settings?: {
-      slashingPayloadLifetimeInRounds?: number;
-    },
-  ) {
-    this.payloads = kvStore.openMap('slash-payloads');
-    this.roundPayloadVotes = kvStore.openMap('round-payload-votes');
-  }
-
-  public async getPayloadsForRound(round: bigint): Promise<SlashPayloadRound[]> {
-    const payloads: SlashPayloadRound[] = [];
-    const votes = await this.getVotesForRound(round);
-    for (const [address, votesCount] of votes) {
-      const payload = await this.getPayload(address);
-      if (payload) {
-        payloads.push({ ...payload, votes: votesCount, round });
-      }
-    }
-    return payloads;
-  }
-
-  public async getPayloadAtRound(payloadAddress: EthAddress, round: bigint): Promise<SlashPayloadRound | undefined> {
-    const address = payloadAddress.toString();
-    const buffer = await this.payloads.getAsync(address);
-    if (!buffer) {
-      return undefined;
-    }
-
-    const data = deserializeSlashPayload(buffer);
-    const votes = (await this.roundPayloadVotes.getAsync(this.getPayloadVotesKey(round, address))) ?? 0n;
-
-    return { ...data, votes, round };
-  }
-
-  private async getVotesForRound(round: bigint): Promise<[string, bigint][]> {
-    const votes: [string, bigint][] = [];
-    for await (const [fullKey, roundVotes] of this.roundPayloadVotes.entriesAsync(
-      this.getPayloadVotesKeyRangeForRound(round),
-    )) {
-      // Extract just the address part from the key (remove "round:" prefix)
-      const address = fullKey.split(':')[1];
-      votes.push([address, roundVotes]);
-    }
-    return votes;
-  }
-
-  private getRoundKey(round: bigint): string {
-    return round.toString().padStart(16, '0');
-  }
-
-  private getPayloadVotesKey(round: bigint, payloadAddress: EthAddress | string): string {
-    return `${this.getRoundKey(round)}:${payloadAddress.toString()}`;
-  }
-
-  private getPayloadVotesKeyRangeForRound(round: bigint): { start: string; end: string } {
-    const start = `${this.getRoundKey(round)}:`;
-    const end = `${this.getRoundKey(round)}:Z`; // 'Z' sorts after any hex address, 0x-prefixed or not
-    return { start, end };
-  }
-
-  /**
-   * Purge vote payload data for expired rounds. Does not delete actual payload data.
-   */
-  public async clearExpiredPayloads(currentRound: bigint): Promise<void> {
-    const lifetimeInRounds = this.settings?.slashingPayloadLifetimeInRounds ?? 0;
-    if (lifetimeInRounds <= 0) {
-      return; // No lifetime configured
-    }
-
-    const expiredBefore = currentRound - BigInt(lifetimeInRounds);
-    if (expiredBefore < 0) {
-      return; // Not enough rounds have passed to expire anything
-    }
-
-    // Collect expired payload votes by scanning round-payload keys
-    const expiredPayloads: string[] = [];
-    const expiredVoteKeys: string[] = [];
-
-    for await (const key of this.roundPayloadVotes.keysAsync({
-      end: `${this.getRoundKey(expiredBefore)}:Z`,
-    })) {
-      const [roundStr, payloadAddress] = key.split(':');
-      if (BigInt(roundStr) <= expiredBefore) {
-        expiredVoteKeys.push(key);
-        expiredPayloads.push(payloadAddress);
-      }
-    }
-
-    if (expiredVoteKeys.length === 0) {
-      return; // No expired payloads to clean up
-    }
-
-    // Remove expired payload vote records
-    // Note that we do not delete payload data since these could be repurposed in future votes
-    await this.kvStore.transactionAsync(async () => {
-      for (const key of expiredVoteKeys) {
-        await this.roundPayloadVotes.delete(key);
-      }
-    });
-  }
-
-  public async incrementPayloadVotes(payloadAddress: EthAddress, round: bigint): Promise<bigint> {
-    const key = this.getPayloadVotesKey(round, payloadAddress);
-    let newVotes: bigint;
-    await this.kvStore.transactionAsync(async () => {
-      const currentVotes = (await this.roundPayloadVotes.getAsync(key)) || 0n;
-      newVotes = currentVotes + 1n;
-      await this.roundPayloadVotes.set(key, newVotes);
-    });
-    return newVotes!;
-  }
-
-  public async addPayload(payload: SlashPayloadRound): Promise<void> {
-    const address = payload.address.toString();
-
-    await this.kvStore.transactionAsync(async () => {
-      await this.payloads.set(address, serializeSlashPayload(payload));
-      await this.roundPayloadVotes.set(this.getPayloadVotesKey(payload.round, address), payload.votes);
-    });
-  }
-
-  public async getPayload(payloadAddress: EthAddress | string): Promise<SlashPayload | undefined> {
-    const address = payloadAddress.toString();
-    const buffer = await this.payloads.getAsync(address);
-    return buffer ? deserializeSlashPayload(buffer) : undefined;
-  }
-
-  public async hasPayload(payload: EthAddress): Promise<boolean> {
-    const address = payload.toString();
-    return (await this.payloads.getAsync(address)) !== undefined;
-  }
-}