Aggregate on demand for v0.11

beacon-chain/operations/attestations/BUILD.bazel

@@ -3,7 +3,6 @@ load("@io_bazel_rules_go//go:def.bzl", "go_library", "go_test")
 go_library(
     name = "go_default_library",
     srcs = [
-        "aggregate.go",
         "log.go",
         "metrics.go",
         "pool.go",
@@ -34,7 +33,6 @@ go_library(
 go_test(
     name = "go_default_test",
     srcs = [
-        "aggregate_test.go",
         "pool_test.go",
         "prepare_forkchoice_test.go",
         "prune_expired_test.go",
@@ -50,6 +48,5 @@ go_test(
         "@com_github_gogo_protobuf//proto:go_default_library",
         "@com_github_prysmaticlabs_ethereumapis//eth/v1alpha1:go_default_library",
         "@com_github_prysmaticlabs_go_bitfield//:go_default_library",
-        "@in_gopkg_d4l3k_messagediff_v1//:go_default_library",
     ],
 )

beacon-chain/operations/attestations/kv/unaggregated.go

@@ -52,6 +52,22 @@ func (p *AttCaches) UnaggregatedAttestations() []*ethpb.Attestation {
 	return atts
 }
 
+// UnAggregatedAttestationsBySlotIndex returns the unaggregated attestations in cache,
+// filtered by committee index and slot.
+func (p *AttCaches) UnAggregatedAttestationsBySlotIndex(slot uint64, committeeIndex uint64) []*ethpb.Attestation {
+	atts := make([]*ethpb.Attestation, 0)
+
+	p.unAggregateAttLock.RLock()
+	defer p.unAggregateAttLock.RUnlock()
+	for _, a := range p.unAggregatedAtt {
+		if slot == a.Data.Slot && committeeIndex == a.Data.CommitteeIndex {
+			atts = append(atts, a)
+		}
+	}
+
+	return atts
+}
+
 // DeleteUnaggregatedAttestation deletes the unaggregated attestations in cache.
 func (p *AttCaches) DeleteUnaggregatedAttestation(att *ethpb.Attestation) error {
 	if att == nil {
@@ -73,6 +89,29 @@ func (p *AttCaches) DeleteUnaggregatedAttestation(att *ethpb.Attestation) error
 	return nil
 }
 
+// DeleteUnaggregatedAttestations deletes the unaggregated attestations in cache.
+func (p *AttCaches) DeleteUnaggregatedAttestations(atts []*ethpb.Attestation) error {
+	p.unAggregateAttLock.Lock()
+	defer p.unAggregateAttLock.Unlock()
+	for _, att := range atts {
+		if att == nil {
+			continue
+		}
+		if helpers.IsAggregated(att) {
+			return errors.New("attestation is aggregated")
+		}
+
+		r, err := hashFn(att)
+		if err != nil {
+			return errors.Wrap(err, "could not tree hash attestation")
+		}
+
+		delete(p.unAggregatedAtt, r)
+	}
+
+	return nil
+}
+
 // UnaggregatedAttestationCount returns the number of unaggregated attestations key in the pool.
 func (p *AttCaches) UnaggregatedAttestationCount() int {
 	p.unAggregateAttLock.RLock()

beacon-chain/operations/attestations/kv/unaggregated_test.go

@@ -51,3 +51,51 @@ func TestKV_Unaggregated_CanDelete(t *testing.T) {
 		t.Error("Did not receive correct aggregated atts")
 	}
 }
+
+func TestKV_Unaggregated_FilterBySlotsAndCommitteeIDs(t *testing.T) {
+	cache := NewAttCaches()
+
+	att1 := &ethpb.Attestation{Data: &ethpb.AttestationData{Slot: 1, CommitteeIndex: 1}, AggregationBits: bitfield.Bitlist{0b101}}
+	att2 := &ethpb.Attestation{Data: &ethpb.AttestationData{Slot: 2, CommitteeIndex: 1}, AggregationBits: bitfield.Bitlist{0b110}}
+	att3 := &ethpb.Attestation{Data: &ethpb.AttestationData{Slot: 1, CommitteeIndex: 1}, AggregationBits: bitfield.Bitlist{0b110}}
+	att4 := &ethpb.Attestation{Data: &ethpb.AttestationData{Slot: 1, CommitteeIndex: 2}, AggregationBits: bitfield.Bitlist{0b110}}
+
+	atts := []*ethpb.Attestation{att1, att2, att3, att4}
+
+	for _, att := range atts {
+		if err := cache.SaveUnaggregatedAttestation(att); err != nil {
+			t.Fatal(err)
+		}
+	}
+
+	returned := cache.UnAggregatedAttestationsBySlotIndex(1, 1)
+
+	if !reflect.DeepEqual([]*ethpb.Attestation{att1, att3}, returned) {
+		t.Error("Did not receive correct aggregated atts")
+	}
+}
+
+func TestKV_Unaggregated_BatchDelete(t *testing.T) {
+	cache := NewAttCaches()
+
+	att1 := &ethpb.Attestation{Data: &ethpb.AttestationData{Slot: 1, CommitteeIndex: 1}, AggregationBits: bitfield.Bitlist{0b101}}
+	att2 := &ethpb.Attestation{Data: &ethpb.AttestationData{Slot: 2, CommitteeIndex: 1}, AggregationBits: bitfield.Bitlist{0b110}}
+	att3 := &ethpb.Attestation{Data: &ethpb.AttestationData{Slot: 1, CommitteeIndex: 1}, AggregationBits: bitfield.Bitlist{0b110}}
+	att4 := &ethpb.Attestation{Data: &ethpb.AttestationData{Slot: 1, CommitteeIndex: 2}, AggregationBits: bitfield.Bitlist{0b110}}
+
+	atts := []*ethpb.Attestation{att1, att2, att3, att4}
+
+	for _, att := range atts {
+		if err := cache.SaveUnaggregatedAttestation(att); err != nil {
+			t.Fatal(err)
+		}
+	}
+	if err := cache.DeleteUnaggregatedAttestations(atts); err != nil {
+		t.Fatal(err)
+	}
+
+	returned := cache.UnaggregatedAttestations()
+	if !reflect.DeepEqual([]*ethpb.Attestation{}, returned) {
+		t.Error("Did not receive correct aggregated atts")
+	}
+}

beacon-chain/operations/attestations/pool.go

@@ -22,7 +22,9 @@ type Pool interface {
 	SaveUnaggregatedAttestation(att *ethpb.Attestation) error
 	SaveUnaggregatedAttestations(atts []*ethpb.Attestation) error
 	UnaggregatedAttestations() []*ethpb.Attestation
+	UnAggregatedAttestationsBySlotIndex(slot uint64, committeeIndex uint64) []*ethpb.Attestation
 	DeleteUnaggregatedAttestation(att *ethpb.Attestation) error
+	DeleteUnaggregatedAttestations(atts []*ethpb.Attestation) error
 	UnaggregatedAttestationCount() int
 	// For attestations that were included in the block.
 	SaveBlockAttestation(att *ethpb.Attestation) error

beacon-chain/operations/attestations/service.go

@@ -43,7 +43,6 @@ func NewService(ctx context.Context, cfg *Config) (*Service, error) {
 // Start an attestation pool service's main event loop.
 func (s *Service) Start() {
 	go s.prepareForkChoiceAtts()
-	go s.aggregateRoutine()
 	go s.pruneAttsPool()
 }
 

beacon-chain/rpc/validator/BUILD.bazel

@@ -41,8 +41,6 @@ go_library(
         "//shared/featureconfig:go_default_library",
         "//shared/hashutil:go_default_library",
         "//shared/params:go_default_library",
-        "//shared/roughtime:go_default_library",
-        "//shared/slotutil:go_default_library",
         "//shared/traceutil:go_default_library",
         "//shared/trieutil:go_default_library",
         "@com_github_gogo_protobuf//types:go_default_library",

beacon-chain/rpc/validator/aggregator.go

@@ -55,7 +55,21 @@ func (as *Server) SubmitAggregateSelectionProof(ctx context.Context, req *ethpb.
 	}
 
 	// Retrieve the unaggregated attestation from pool.
+	unaggregatedAtts := as.AttPool.UnAggregatedAttestationsBySlotIndex(req.Slot, req.CommitteeIndex)
+	// In case there's left over aggregated attestations in the pool.
 	aggregatedAtts := as.AttPool.AggregatedAttestationsBySlotIndex(req.Slot, req.CommitteeIndex)
+	aggregatedAtts, err = helpers.AggregateAttestations(append(aggregatedAtts, unaggregatedAtts...))
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, "Could not get aggregate attestations: %v", err)
+	}
+
+	// Save the aggregated attestations to the pool.
+	if err := as.AttPool.SaveAggregatedAttestations(aggregatedAtts); err != nil {
+		return nil, status.Errorf(codes.Internal, "Could not save aggregated attestations: %v", err)
+	}
+	if err := as.AttPool.DeleteUnaggregatedAttestations(unaggregatedAtts); err != nil {
+		return nil, status.Errorf(codes.Internal, "Could not delete unaggregated attestations: %v", err)
+	}
 
 	// Filter out the best aggregated attestation (ie. the one with the most aggregated bits).
 	if len(aggregatedAtts) == 0 {