Skip to content

Commit 700257a

Browse files
authored
getBits method names clearer (#9453)
1 parent 0d7f50c commit 700257a

File tree

8 files changed

+93
-80
lines changed

8 files changed

+93
-80
lines changed

ethereum/statetransition/src/main/java/tech/pegasys/teku/statetransition/attestation/PooledAttestationWithData.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,9 +21,9 @@ public record PooledAttestationWithData(AttestationData data, PooledAttestation
2121

2222
public Attestation toAttestation(final AttestationSchema<Attestation> attestationSchema) {
2323
return attestationSchema.create(
24-
pooledAttestation.bits().getAggregationBits(),
24+
pooledAttestation.bits().getAggregationSszBits(),
2525
data,
2626
pooledAttestation.aggregatedSignature(),
27-
pooledAttestation.bits()::getCommitteeBits);
27+
pooledAttestation.bits()::getCommitteeSszBits);
2828
}
2929
}

ethereum/statetransition/src/main/java/tech/pegasys/teku/statetransition/attestation/utils/AttestationBits.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -69,9 +69,9 @@ static AttestationBits of(
6969

7070
boolean isSuperSetOf(AttestationBits other);
7171

72-
SszBitlist getAggregationBits();
72+
SszBitlist getAggregationSszBits();
7373

74-
SszBitvector getCommitteeBits();
74+
SszBitvector getCommitteeSszBits();
7575

7676
Int2IntMap getCommitteesSize();
7777

ethereum/statetransition/src/main/java/tech/pegasys/teku/statetransition/attestation/utils/AttestationBitsElectra.java

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -39,8 +39,8 @@ class AttestationBitsElectra implements AttestationBits {
3939
private Int2ObjectMap<BitSet> committeeAggregationBitsMap;
4040
private BitSet committeeBits;
4141

42-
private SszBitlist cachedAggregationBits = null;
43-
private SszBitvector cachedCommitteeBits = null;
42+
private SszBitlist cachedAggregationSszBits = null;
43+
private SszBitvector cachedCommitteeSszBits = null;
4444

4545
AttestationBitsElectra(
4646
final SszBitlist initialAggregationBits,
@@ -249,9 +249,9 @@ private boolean isSuperSetOf(
249249
}
250250

251251
@Override
252-
public SszBitlist getAggregationBits() {
253-
if (cachedAggregationBits != null) {
254-
return cachedAggregationBits;
252+
public SszBitlist getAggregationSszBits() {
253+
if (cachedAggregationSszBits != null) {
254+
return cachedAggregationSszBits;
255255
}
256256
final List<Integer> committeeIndicesInOrder = new ArrayList<>();
257257
for (int i = committeeBits.nextSetBit(0); i >= 0; i = committeeBits.nextSetBit(i + 1)) {
@@ -279,24 +279,24 @@ public SszBitlist getAggregationBits() {
279279
}
280280
currentOffset += committeeSize;
281281
}
282-
cachedAggregationBits =
282+
cachedAggregationSszBits =
283283
aggregationBitsSchema.wrapBitSet(totalBitlistSize, combinedAggregationBits);
284-
return cachedAggregationBits;
284+
return cachedAggregationSszBits;
285285
}
286286

287287
@Override
288-
public SszBitvector getCommitteeBits() {
289-
if (cachedCommitteeBits == null) {
290-
cachedCommitteeBits =
288+
public SszBitvector getCommitteeSszBits() {
289+
if (cachedCommitteeSszBits == null) {
290+
cachedCommitteeSszBits =
291291
committeeBitsSchema.wrapBitSet(
292292
committeeBitsSchema.getLength(), (BitSet) this.committeeBits.clone());
293293
}
294-
return cachedCommitteeBits;
294+
return cachedCommitteeSszBits;
295295
}
296296

297297
private void invalidateCache() {
298-
this.cachedAggregationBits = null;
299-
this.cachedCommitteeBits = null;
298+
this.cachedAggregationSszBits = null;
299+
this.cachedCommitteeSszBits = null;
300300
}
301301

302302
@Override
@@ -359,7 +359,7 @@ public String toString() {
359359
.add("committeeBits", committeeBits.cardinality())
360360
.add("committeeAggregationBitsMap", totalSetBits)
361361
.add("committeesSize", committeesSize.size())
362-
.add("cached", cachedAggregationBits != null || cachedCommitteeBits != null)
362+
.add("sszBitsCached", cachedAggregationSszBits != null || cachedCommitteeSszBits != null)
363363
.toString();
364364
}
365365

ethereum/statetransition/src/main/java/tech/pegasys/teku/statetransition/attestation/utils/AttestationBitsPhase0.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -66,12 +66,12 @@ public boolean isSuperSetOf(final AttestationBits other) {
6666
}
6767

6868
@Override
69-
public SszBitlist getAggregationBits() {
69+
public SszBitlist getAggregationSszBits() {
7070
return aggregationBits;
7171
}
7272

7373
@Override
74-
public SszBitvector getCommitteeBits() {
74+
public SszBitvector getCommitteeSszBits() {
7575
throw new IllegalStateException("Committee bits not available in phase0");
7676
}
7777

ethereum/statetransition/src/test/java/tech/pegasys/teku/statetransition/attestation/MatchingDataAttestationGroupTest.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -444,10 +444,10 @@ private PooledAttestation createPooledAttestation(
444444

445445
private Attestation toAttestation(final PooledAttestation pooledAttestation) {
446446
return attestationSchema.create(
447-
pooledAttestation.bits().getAggregationBits(),
447+
pooledAttestation.bits().getAggregationSszBits(),
448448
attestationData,
449449
pooledAttestation.aggregatedSignature(),
450-
pooledAttestation.bits()::getCommitteeBits);
450+
pooledAttestation.bits()::getCommitteeSszBits);
451451
}
452452

453453
private PooledAttestationWithData toPooledAttestationWithData(

ethereum/statetransition/src/test/java/tech/pegasys/teku/statetransition/attestation/MatchingDataAttestationGroupV2Test.java

Lines changed: 11 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -770,12 +770,13 @@ private List<UInt64> validatorBitToValidatorIndex(final int... validatorBits) {
770770

771771
private List<UInt64> validatorBitToValidatorIndex(final AttestationBits bits) {
772772
if (!bits.requiresCommitteeBits()) {
773-
return validatorBitToValidatorIndex(bits.getAggregationBits().getAllSetBits().toIntArray());
773+
return validatorBitToValidatorIndex(
774+
bits.getAggregationSszBits().getAllSetBits().toIntArray());
774775
}
775776

776777
// only 2 committees are supported
777778
assertThat(committeeSizes.keySet()).isEqualTo(Set.of(0, 1));
778-
final IntList committeeBits = bits.getCommitteeBits().getAllSetBits();
779+
final IntList committeeBits = bits.getCommitteeSszBits().getAllSetBits();
779780
assertThat(committeeBits.size()).isLessThanOrEqualTo(2);
780781

781782
final List<UInt64> result = new ArrayList<>();
@@ -784,14 +785,18 @@ private List<UInt64> validatorBitToValidatorIndex(final AttestationBits bits) {
784785
result.addAll(
785786
validatorBitToValidatorIndex(
786787
Optional.of(0),
787-
bits.getAggregationBits().getAsBitSet(0, committeeSizes.get(0)).stream().toArray()));
788+
bits.getAggregationSszBits().getAsBitSet(0, committeeSizes.get(0)).stream()
789+
.toArray()));
788790
offset = committeeSizes.get(0);
789791
}
790792
if (committeeBits.contains(1)) {
791793
result.addAll(
792794
validatorBitToValidatorIndex(
793795
Optional.of(1),
794-
bits.getAggregationBits().getAsBitSet(offset, committeeSizes.get(1) + offset).stream()
796+
bits
797+
.getAggregationSszBits()
798+
.getAsBitSet(offset, committeeSizes.get(1) + offset)
799+
.stream()
795800
.toArray()));
796801
}
797802

@@ -800,10 +805,10 @@ private List<UInt64> validatorBitToValidatorIndex(final AttestationBits bits) {
800805

801806
private Attestation toAttestation(final PooledAttestation pooledAttestation) {
802807
return attestationSchema.create(
803-
pooledAttestation.bits().getAggregationBits(),
808+
pooledAttestation.bits().getAggregationSszBits(),
804809
attestationData,
805810
pooledAttestation.aggregatedSignature(),
806-
pooledAttestation.bits()::getCommitteeBits); // Supplier for committee bits
811+
pooledAttestation.bits()::getCommitteeSszBits); // Supplier for committee bits
807812
}
808813

809814
private PooledAttestationWithData toPooledAttestationWithData(

0 commit comments

Comments
 (0)