only add to taskResponseMap if doesn't already exist

Layr-Labs · May 25, 2024 · 63b6a39 · 63b6a39
1 parent 86a5aa1
commit 63b6a39
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/services/bls_aggregation/blsagg.go b/services/bls_aggregation/blsagg.go
@@ -186,7 +186,6 @@ func (a *BlsAggregatorService) ProcessNewSignature(
 	ctx context.Context,
 	taskIndex types.TaskIndex,
 	taskResponse types.TaskResponse,
-	//taskResponseDigest types.TaskResponseDigest,
 	blsSignature *bls.Signature,
 	operatorId types.OperatorId,
 ) error {
@@ -199,8 +198,12 @@ func (a *BlsAggregatorService) ProcessNewSignature(
 	// compute the taskResponseDigest, note that this is now enforcing a specific encoding for the taskResponse
 	taskResponseDigest := types.TaskResponseDigest(sha256.Sum256(taskResponse))
 
-	// Store the TaskResponse in our mapping
-	a.taskResponseMap[taskResponseDigest] = taskResponse
+	// check if the taskResponseDigest is already in the map
+	_, taskResponseExists := a.taskResponseMap[taskResponseDigest]
+	if !taskResponseExists {
+		// Store the TaskResponse in our mapping
+		a.taskResponseMap[taskResponseDigest] = taskResponse
+	}
 
 	signatureVerificationErrorC := make(chan error)
 	// send the task to the goroutine processing this task