perf: receive done

bnb-chain · Jun 19, 2024 · 15d8f61 · 15d8f61
1 parent 37e3aa3
commit 15d8f61
Showing 1 changed file with 13 additions and 5 deletions.
diff --git a/modular/receiver/receive_task.go b/modular/receiver/receive_task.go
@@ -114,10 +114,18 @@ func (r *ReceiveModular) HandleDoneReceivePieceTask(ctx context.Context, task ta
 		log.CtxErrorw(ctx, "failed to get checksum from db", "task", task, "error", err)
 		return nil, ErrGfSpDBWithDetail("failed to get checksum from db, error: " + err.Error())
 	}
+	skipInsertIntegrityMeta := false
 	if len(pieceChecksums) != int(segmentCount) {
-		log.CtxErrorw(ctx, "replicate piece unfinished", "task", task)
-		err = ErrUnfinishedTask
-		return nil, ErrUnfinishedTask
+		// If it is successful, return directly
+		integrityMeta, integrityErr := r.baseApp.GfSpDB().GetObjectIntegrity(task.GetObjectInfo().Id.Uint64(), task.GetRedundancyIdx())
+		if integrityMeta != nil && integrityErr == nil {
+			pieceChecksums = integrityMeta.PieceChecksumList
+			skipInsertIntegrityMeta = true
+		} else {
+			log.CtxErrorw(ctx, "replicate piece unfinished", "task", task)
+			err = ErrUnfinishedTask
+			return nil, ErrUnfinishedTask
+		}
 	}
 
 	expectedIntegrityHash := task.GetObjectInfo().GetChecksums()[task.GetRedundancyIdx()+1]
@@ -137,7 +145,7 @@ func (r *ReceiveModular) HandleDoneReceivePieceTask(ctx context.Context, task ta
 	}
 
 	setIntegrityTime := time.Now()
-	if task.GetObjectInfo().GetIsUpdating() {
+	if !skipInsertIntegrityMeta && task.GetObjectInfo().GetIsUpdating() {
 		integrityMeta := &corespdb.ShadowIntegrityMeta{
 			ObjectID:          task.GetObjectInfo().Id.Uint64(),
 			RedundancyIndex:   task.GetRedundancyIdx(),
@@ -147,7 +155,7 @@ func (r *ReceiveModular) HandleDoneReceivePieceTask(ctx context.Context, task ta
 			ObjectSize:        task.GetObjectInfo().GetPayloadSize(),
 		}
 		err = r.baseApp.GfSpDB().SetShadowObjectIntegrity(integrityMeta)
-	} else {
+	} else if !skipInsertIntegrityMeta {
 		integrityMeta := &corespdb.IntegrityMeta{
 			ObjectID:          task.GetObjectInfo().Id.Uint64(),
 			RedundancyIndex:   task.GetRedundancyIdx(),