keep encoded blob result until terminal

Layr-Labs · Sep 10, 2024 · d51fa10 · d51fa10
1 parent a0191b8
commit d51fa10
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 8 deletions.
diff --git a/disperser/batcher/batcher.go b/disperser/batcher/batcher.go
@@ -321,6 +321,7 @@ func (b *Batcher) updateConfirmationInfo(
 		} else {
 			updateConfirmationInfoErr = fmt.Errorf("HandleSingleBatch: trying to update confirmation info for blob in status other than confirmed or insufficient signatures: %s", status.String())
 		}
+		b.EncodingStreamer.RemoveEncodedBlob(metadata)
 		if updateConfirmationInfoErr != nil {
 			b.logger.Error("HandleSingleBatch: error updating blob confirmed metadata", "err", updateConfirmationInfoErr)
 			blobsToRetry = append(blobsToRetry, batchData.blobs[blobIndex])

diff --git a/disperser/batcher/batcher_test.go b/disperser/batcher/batcher_test.go
@@ -448,8 +448,8 @@ func TestBlobRetry(t *testing.T) {
 	assert.NoError(t, err)
 	assert.Equal(t, disperser.Dispersing, meta.BlobStatus)
 	encodedResult, err = components.encodingStreamer.EncodedBlobstore.GetEncodingResult(blobKey, 0)
-	assert.ErrorContains(t, err, "no such key")
-	assert.Nil(t, encodedResult)
+	assert.NoError(t, err)
+	assert.NotNil(t, encodedResult)
 
 	err = components.encodingStreamer.RequestEncoding(ctx, out)
 	assert.NoError(t, err)

diff --git a/disperser/batcher/encoding_streamer.go b/disperser/batcher/encoding_streamer.go
@@ -769,8 +769,6 @@ func (e *EncodingStreamer) transitionBlobToDispersing(ctx context.Context, metad
 		e.logger.Error("error marking blob as dispersing", "err", err, "blobKey", blobKey.String())
 		return err
 	}
-	// remove encoded blob from storage so we don't disperse it again
-	e.RemoveEncodedBlob(metadata)
 	return nil
 }
 

diff --git a/disperser/batcher/encoding_streamer_test.go b/disperser/batcher/encoding_streamer_test.go
@@ -655,11 +655,11 @@ func TestGetBatch(t *testing.T) {
 	assert.Equal(t, disperser.Dispersing, metadata2.BlobStatus)
 	assert.Nil(t, err)
 	res, err := encodingStreamer.EncodedBlobstore.GetEncodingResult(metadataKey1, core.QuorumID(0))
-	assert.Nil(t, res)
-	assert.ErrorContains(t, err, "GetEncodedBlob: no such key")
+	assert.NoError(t, err)
+	assert.NotNil(t, res)
 	res, err = encodingStreamer.EncodedBlobstore.GetEncodingResult(metadataKey1, core.QuorumID(1))
-	assert.Nil(t, res)
-	assert.ErrorContains(t, err, "GetEncodedBlob: no such key")
+	assert.NoError(t, err)
+	assert.NotNil(t, res)
 	res, err = encodingStreamer.EncodedBlobstore.GetEncodingResult(metadataKey2, core.QuorumID(0))
 	assert.Nil(t, res)
 	assert.ErrorContains(t, err, "GetEncodedBlob: no such key")