mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2026-07-22 07:53:38 +00:00
634 lines
21 KiB
Go
634 lines
21 KiB
Go
package erasure_coding_test
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
erasure_coding "github.com/seaweedfs/seaweedfs/weed/storage/erasure_coding"
|
|
"github.com/seaweedfs/seaweedfs/weed/storage/needle"
|
|
"github.com/seaweedfs/seaweedfs/weed/storage/types"
|
|
)
|
|
|
|
func TestHasLiveNeedles_AllDeletedIsFalse(t *testing.T) {
|
|
dir := t.TempDir()
|
|
|
|
collection := "foo"
|
|
base := filepath.Join(dir, collection+"_1")
|
|
|
|
// Build an ecx file with only deleted entries.
|
|
// ecx file entries are the same format as .idx entries.
|
|
ecx := makeNeedleMapEntry(types.NeedleId(1), types.Offset{}, types.TombstoneFileSize)
|
|
if err := os.WriteFile(base+".ecx", ecx, 0644); err != nil {
|
|
t.Fatalf("write ecx: %v", err)
|
|
}
|
|
|
|
hasLive, err := erasure_coding.HasLiveNeedles(base)
|
|
if err != nil {
|
|
t.Fatalf("HasLiveNeedles: %v", err)
|
|
}
|
|
if hasLive {
|
|
t.Fatalf("expected no live entries")
|
|
}
|
|
}
|
|
|
|
func TestHasLiveNeedles_WithLiveEntryIsTrue(t *testing.T) {
|
|
dir := t.TempDir()
|
|
|
|
collection := "foo"
|
|
base := filepath.Join(dir, collection+"_1")
|
|
|
|
// Build an ecx file containing at least one live entry.
|
|
// ecx file entries are the same format as .idx entries.
|
|
live := makeNeedleMapEntry(types.NeedleId(1), types.Offset{}, types.Size(1))
|
|
if err := os.WriteFile(base+".ecx", live, 0644); err != nil {
|
|
t.Fatalf("write ecx: %v", err)
|
|
}
|
|
|
|
hasLive, err := erasure_coding.HasLiveNeedles(base)
|
|
if err != nil {
|
|
t.Fatalf("HasLiveNeedles: %v", err)
|
|
}
|
|
if !hasLive {
|
|
t.Fatalf("expected live entries")
|
|
}
|
|
}
|
|
|
|
func TestHasLiveNeedles_EmptyFileIsFalse(t *testing.T) {
|
|
dir := t.TempDir()
|
|
|
|
base := filepath.Join(dir, "foo_1")
|
|
|
|
// Create an empty ecx file.
|
|
if err := os.WriteFile(base+".ecx", []byte{}, 0644); err != nil {
|
|
t.Fatalf("write ecx: %v", err)
|
|
}
|
|
|
|
hasLive, err := erasure_coding.HasLiveNeedles(base)
|
|
if err != nil {
|
|
t.Fatalf("HasLiveNeedles: %v", err)
|
|
}
|
|
if hasLive {
|
|
t.Fatalf("expected no live entries for empty file")
|
|
}
|
|
}
|
|
|
|
func TestHasLiveNeedles_TruncatedEcxErrors(t *testing.T) {
|
|
dir := t.TempDir()
|
|
|
|
base := filepath.Join(dir, "foo_1")
|
|
|
|
entry := makeNeedleMapEntry(types.NeedleId(1), types.Offset{}, types.TombstoneFileSize)
|
|
truncated := append(entry, entry[:types.NeedleIdSize]...)
|
|
if err := os.WriteFile(base+".ecx", truncated, 0644); err != nil {
|
|
t.Fatalf("write ecx: %v", err)
|
|
}
|
|
|
|
if _, err := erasure_coding.HasLiveNeedles(base); err == nil {
|
|
t.Fatalf("expected error for truncated ecx")
|
|
}
|
|
}
|
|
|
|
func makeNeedleMapEntry(key types.NeedleId, offset types.Offset, size types.Size) []byte {
|
|
b := make([]byte, types.NeedleIdSize+types.OffsetSize+types.SizeSize)
|
|
types.NeedleIdToBytes(b[0:types.NeedleIdSize], key)
|
|
types.OffsetToBytes(b[types.NeedleIdSize:types.NeedleIdSize+types.OffsetSize], offset)
|
|
types.SizeToBytes(b[types.NeedleIdSize+types.OffsetSize:types.NeedleIdSize+types.OffsetSize+types.SizeSize], size)
|
|
return b
|
|
}
|
|
|
|
// TestWriteIdxFileFromEcIndex_PreservesDeletedNeedles verifies that WriteIdxFileFromEcIndex
|
|
// correctly marks deleted needles in the generated .idx file.
|
|
// This tests the fix for issue #7751 where deleted files in encoded volumes
|
|
// were not properly marked as deleted when decoded.
|
|
func TestWriteIdxFileFromEcIndex_PreservesDeletedNeedles(t *testing.T) {
|
|
dir := t.TempDir()
|
|
base := filepath.Join(dir, "foo_1")
|
|
|
|
// Create an .ecx file with one live needle and one deleted needle
|
|
needle1 := makeNeedleMapEntry(types.NeedleId(1), types.ToOffset(64), types.Size(100))
|
|
needle2 := makeNeedleMapEntry(types.NeedleId(2), types.ToOffset(128), types.TombstoneFileSize) // deleted
|
|
|
|
ecxData := append(needle1, needle2...)
|
|
if err := os.WriteFile(base+".ecx", ecxData, 0644); err != nil {
|
|
t.Fatalf("write ecx: %v", err)
|
|
}
|
|
|
|
// Generate .idx from .ecx
|
|
if err := erasure_coding.WriteIdxFileFromEcIndex(base); err != nil {
|
|
t.Fatalf("WriteIdxFileFromEcIndex: %v", err)
|
|
}
|
|
|
|
// Verify .idx file has the same content
|
|
idxData, err := os.ReadFile(base + ".idx")
|
|
if err != nil {
|
|
t.Fatalf("read idx: %v", err)
|
|
}
|
|
|
|
if len(idxData) != len(ecxData) {
|
|
t.Fatalf("idx file size mismatch: got %d, want %d", len(idxData), len(ecxData))
|
|
}
|
|
|
|
// Verify the second needle is still marked as deleted
|
|
entrySize := types.NeedleIdSize + types.OffsetSize + types.SizeSize
|
|
entry2 := idxData[entrySize : entrySize*2]
|
|
size2 := types.BytesToSize(entry2[types.NeedleIdSize+types.OffsetSize:])
|
|
if !size2.IsDeleted() {
|
|
t.Fatalf("expected needle 2 to be marked as deleted, got size: %d", size2)
|
|
}
|
|
}
|
|
|
|
// TestWriteIdxFileFromEcIndex_ProcessesEcjJournal verifies that WriteIdxFileFromEcIndex
|
|
// correctly processes deletions from the .ecj journal file.
|
|
func TestWriteIdxFileFromEcIndex_ProcessesEcjJournal(t *testing.T) {
|
|
dir := t.TempDir()
|
|
base := filepath.Join(dir, "foo_1")
|
|
|
|
// Create an .ecx file with two live needles
|
|
needle1 := makeNeedleMapEntry(types.NeedleId(1), types.ToOffset(64), types.Size(100))
|
|
needle2 := makeNeedleMapEntry(types.NeedleId(2), types.ToOffset(128), types.Size(200))
|
|
|
|
ecxData := append(needle1, needle2...)
|
|
if err := os.WriteFile(base+".ecx", ecxData, 0644); err != nil {
|
|
t.Fatalf("write ecx: %v", err)
|
|
}
|
|
|
|
// Create an .ecj file that records needle 2 as deleted
|
|
ecjData := make([]byte, types.NeedleIdSize)
|
|
types.NeedleIdToBytes(ecjData, types.NeedleId(2))
|
|
if err := os.WriteFile(base+".ecj", ecjData, 0644); err != nil {
|
|
t.Fatalf("write ecj: %v", err)
|
|
}
|
|
|
|
// Generate .idx from .ecx and .ecj
|
|
if err := erasure_coding.WriteIdxFileFromEcIndex(base); err != nil {
|
|
t.Fatalf("WriteIdxFileFromEcIndex: %v", err)
|
|
}
|
|
|
|
// Verify .idx file has 3 entries: 2 from .ecx + 1 deletion from .ecj
|
|
idxData, err := os.ReadFile(base + ".idx")
|
|
if err != nil {
|
|
t.Fatalf("read idx: %v", err)
|
|
}
|
|
|
|
entrySize := types.NeedleIdSize + types.OffsetSize + types.SizeSize
|
|
expectedSize := entrySize * 3 // 2 from ecx + 1 deletion append from ecj
|
|
if len(idxData) != expectedSize {
|
|
t.Fatalf("idx file size mismatch: got %d, want %d", len(idxData), expectedSize)
|
|
}
|
|
|
|
// The third entry should be the deletion record for needle 2
|
|
entry3 := idxData[entrySize*2 : entrySize*3]
|
|
key3 := types.BytesToNeedleId(entry3[0:types.NeedleIdSize])
|
|
size3 := types.BytesToSize(entry3[types.NeedleIdSize+types.OffsetSize:])
|
|
|
|
if key3 != types.NeedleId(2) {
|
|
t.Fatalf("expected needle id 2 in deletion record, got: %d", key3)
|
|
}
|
|
if !size3.IsDeleted() {
|
|
t.Fatalf("expected deletion record to have tombstone size, got: %d", size3)
|
|
}
|
|
}
|
|
|
|
// TestDecodeAtomicPublish verifies the decoded .idx/.dat are published via a
|
|
// temp file plus rename: a successful write leaves the final file with no
|
|
// leftover .tmp, and a failed write leaves neither the final file nor a
|
|
// partial .tmp beside the source shards.
|
|
func TestDecodeAtomicPublish(t *testing.T) {
|
|
dir := t.TempDir()
|
|
base := filepath.Join(dir, "foo_1")
|
|
|
|
// .ecx with one live needle so WriteIdxFileFromEcIndex succeeds.
|
|
ecxData := makeNeedleMapEntry(types.NeedleId(1), types.ToOffset(64), types.Size(100))
|
|
if err := os.WriteFile(base+".ecx", ecxData, 0644); err != nil {
|
|
t.Fatalf("write ecx: %v", err)
|
|
}
|
|
if err := erasure_coding.WriteIdxFileFromEcIndex(base); err != nil {
|
|
t.Fatalf("WriteIdxFileFromEcIndex: %v", err)
|
|
}
|
|
if _, err := os.Stat(base + ".idx"); err != nil {
|
|
t.Fatalf("decoded .idx missing: %v", err)
|
|
}
|
|
if _, err := os.Stat(base + ".idx.tmp"); !os.IsNotExist(err) {
|
|
t.Fatalf("expected no leftover .idx.tmp, stat err=%v", err)
|
|
}
|
|
|
|
// A WriteDatFile pointed at a missing shard must fail and leave neither the
|
|
// final .dat nor a partial .dat.tmp behind.
|
|
datBase := filepath.Join(dir, "bar_2")
|
|
missingShards := []string{filepath.Join(dir, "does_not_exist.ec00")}
|
|
if err := erasure_coding.WriteDatFile(datBase, 100, missingShards); err == nil {
|
|
t.Fatalf("expected WriteDatFile to fail on missing shard")
|
|
}
|
|
if _, err := os.Stat(datBase + ".dat"); !os.IsNotExist(err) {
|
|
t.Fatalf("failed decode must not leave a .dat, stat err=%v", err)
|
|
}
|
|
if _, err := os.Stat(datBase + ".dat.tmp"); !os.IsNotExist(err) {
|
|
t.Fatalf("failed decode must not leave a .dat.tmp, stat err=%v", err)
|
|
}
|
|
}
|
|
|
|
// TestDecodeWithNonEmptyEcj_AllDeleted verifies the full decode pre-processing
|
|
// when .ecj contains deletions for ALL live entries in .ecx.
|
|
// After RebuildEcxFile merges .ecj into .ecx, HasLiveNeedles must return false
|
|
// and WriteIdxFileFromEcIndex must produce an .idx where every entry is tombstoned.
|
|
func TestDecodeWithNonEmptyEcj_AllDeleted(t *testing.T) {
|
|
dir := t.TempDir()
|
|
base := filepath.Join(dir, "test_1")
|
|
|
|
// .ecx: two live entries
|
|
needle1 := makeNeedleMapEntry(types.NeedleId(1), types.ToOffset(64), types.Size(100))
|
|
needle2 := makeNeedleMapEntry(types.NeedleId(2), types.ToOffset(128), types.Size(200))
|
|
ecxData := append(needle1, needle2...)
|
|
if err := os.WriteFile(base+".ecx", ecxData, 0644); err != nil {
|
|
t.Fatalf("write ecx: %v", err)
|
|
}
|
|
|
|
// .ecj: both needles deleted
|
|
ecjData := make([]byte, 2*types.NeedleIdSize)
|
|
types.NeedleIdToBytes(ecjData[0:types.NeedleIdSize], types.NeedleId(1))
|
|
types.NeedleIdToBytes(ecjData[types.NeedleIdSize:], types.NeedleId(2))
|
|
if err := os.WriteFile(base+".ecj", ecjData, 0644); err != nil {
|
|
t.Fatalf("write ecj: %v", err)
|
|
}
|
|
|
|
// Before rebuild, ecx entries look live
|
|
hasLive, err := erasure_coding.HasLiveNeedles(base)
|
|
if err != nil {
|
|
t.Fatalf("HasLiveNeedles before rebuild: %v", err)
|
|
}
|
|
if !hasLive {
|
|
t.Fatal("expected live entries before rebuild")
|
|
}
|
|
|
|
// Simulate what VolumeEcShardsToVolume now does: merge .ecj into .ecx
|
|
if err := erasure_coding.RebuildEcxFile(base); err != nil {
|
|
t.Fatalf("RebuildEcxFile: %v", err)
|
|
}
|
|
|
|
// .ecj should be removed after rebuild
|
|
if _, err := os.Stat(base + ".ecj"); !os.IsNotExist(err) {
|
|
t.Fatal("expected .ecj to be removed after RebuildEcxFile")
|
|
}
|
|
|
|
// After rebuild, HasLiveNeedles must return false
|
|
hasLive, err = erasure_coding.HasLiveNeedles(base)
|
|
if err != nil {
|
|
t.Fatalf("HasLiveNeedles after rebuild: %v", err)
|
|
}
|
|
if hasLive {
|
|
t.Fatal("expected no live entries after rebuild merged all deletions")
|
|
}
|
|
|
|
// WriteIdxFileFromEcIndex should still work (no .ecj to process)
|
|
if err := erasure_coding.WriteIdxFileFromEcIndex(base); err != nil {
|
|
t.Fatalf("WriteIdxFileFromEcIndex: %v", err)
|
|
}
|
|
|
|
idxData, err := os.ReadFile(base + ".idx")
|
|
if err != nil {
|
|
t.Fatalf("read idx: %v", err)
|
|
}
|
|
|
|
// .idx should have exactly 2 entries (copied from .ecx, both now tombstoned)
|
|
entrySize := types.NeedleIdSize + types.OffsetSize + types.SizeSize
|
|
if len(idxData) != 2*entrySize {
|
|
t.Fatalf("idx file size: got %d, want %d", len(idxData), 2*entrySize)
|
|
}
|
|
|
|
// Both entries must be tombstoned
|
|
for i := 0; i < 2; i++ {
|
|
entry := idxData[i*entrySize : (i+1)*entrySize]
|
|
size := types.BytesToSize(entry[types.NeedleIdSize+types.OffsetSize:])
|
|
if !size.IsDeleted() {
|
|
t.Fatalf("entry %d: expected tombstone, got size %d", i+1, size)
|
|
}
|
|
}
|
|
}
|
|
|
|
// TestDecodeWithNonEmptyEcj_PartiallyDeleted verifies decode pre-processing
|
|
// when .ecj deletes only some entries. After RebuildEcxFile, HasLiveNeedles
|
|
// must still return true for the surviving entries, and WriteIdxFileFromEcIndex
|
|
// must produce an .idx that correctly distinguishes live from deleted needles.
|
|
func TestDecodeWithNonEmptyEcj_PartiallyDeleted(t *testing.T) {
|
|
dir := t.TempDir()
|
|
base := filepath.Join(dir, "test_1")
|
|
|
|
// .ecx: three live entries
|
|
needle1 := makeNeedleMapEntry(types.NeedleId(1), types.ToOffset(64), types.Size(100))
|
|
needle2 := makeNeedleMapEntry(types.NeedleId(2), types.ToOffset(128), types.Size(200))
|
|
needle3 := makeNeedleMapEntry(types.NeedleId(3), types.ToOffset(256), types.Size(300))
|
|
ecxData := append(append(needle1, needle2...), needle3...)
|
|
if err := os.WriteFile(base+".ecx", ecxData, 0644); err != nil {
|
|
t.Fatalf("write ecx: %v", err)
|
|
}
|
|
|
|
// .ecj: only needle 2 is deleted
|
|
ecjData := make([]byte, types.NeedleIdSize)
|
|
types.NeedleIdToBytes(ecjData, types.NeedleId(2))
|
|
if err := os.WriteFile(base+".ecj", ecjData, 0644); err != nil {
|
|
t.Fatalf("write ecj: %v", err)
|
|
}
|
|
|
|
// Merge .ecj into .ecx
|
|
if err := erasure_coding.RebuildEcxFile(base); err != nil {
|
|
t.Fatalf("RebuildEcxFile: %v", err)
|
|
}
|
|
|
|
// HasLiveNeedles must still return true (needles 1 and 3 survive)
|
|
hasLive, err := erasure_coding.HasLiveNeedles(base)
|
|
if err != nil {
|
|
t.Fatalf("HasLiveNeedles: %v", err)
|
|
}
|
|
if !hasLive {
|
|
t.Fatal("expected live entries after partial deletion")
|
|
}
|
|
|
|
// WriteIdxFileFromEcIndex
|
|
if err := erasure_coding.WriteIdxFileFromEcIndex(base); err != nil {
|
|
t.Fatalf("WriteIdxFileFromEcIndex: %v", err)
|
|
}
|
|
|
|
idxData, err := os.ReadFile(base + ".idx")
|
|
if err != nil {
|
|
t.Fatalf("read idx: %v", err)
|
|
}
|
|
|
|
entrySize := types.NeedleIdSize + types.OffsetSize + types.SizeSize
|
|
if len(idxData) != 3*entrySize {
|
|
t.Fatalf("idx file size: got %d, want %d", len(idxData), 3*entrySize)
|
|
}
|
|
|
|
// Verify each entry
|
|
for i := 0; i < 3; i++ {
|
|
entry := idxData[i*entrySize : (i+1)*entrySize]
|
|
key := types.BytesToNeedleId(entry[0:types.NeedleIdSize])
|
|
size := types.BytesToSize(entry[types.NeedleIdSize+types.OffsetSize:])
|
|
|
|
switch key {
|
|
case types.NeedleId(1), types.NeedleId(3):
|
|
if size.IsDeleted() {
|
|
t.Fatalf("needle %d: should be live, got tombstone", key)
|
|
}
|
|
case types.NeedleId(2):
|
|
if !size.IsDeleted() {
|
|
t.Fatalf("needle %d: should be tombstoned, got size %d", key, size)
|
|
}
|
|
default:
|
|
t.Fatalf("unexpected needle id %d", key)
|
|
}
|
|
}
|
|
}
|
|
|
|
// TestDecodeWithEmptyEcj verifies that the decode flow is a no-op when
|
|
// .ecj exists but is empty (no deletions recorded).
|
|
func TestDecodeWithEmptyEcj(t *testing.T) {
|
|
dir := t.TempDir()
|
|
base := filepath.Join(dir, "test_1")
|
|
|
|
// .ecx: one live entry
|
|
needle1 := makeNeedleMapEntry(types.NeedleId(1), types.ToOffset(64), types.Size(100))
|
|
if err := os.WriteFile(base+".ecx", needle1, 0644); err != nil {
|
|
t.Fatalf("write ecx: %v", err)
|
|
}
|
|
|
|
// .ecj: empty
|
|
if err := os.WriteFile(base+".ecj", []byte{}, 0644); err != nil {
|
|
t.Fatalf("write ecj: %v", err)
|
|
}
|
|
|
|
// RebuildEcxFile with empty .ecj should not change anything
|
|
if err := erasure_coding.RebuildEcxFile(base); err != nil {
|
|
t.Fatalf("RebuildEcxFile: %v", err)
|
|
}
|
|
|
|
// HasLiveNeedles must still return true
|
|
hasLive, err := erasure_coding.HasLiveNeedles(base)
|
|
if err != nil {
|
|
t.Fatalf("HasLiveNeedles: %v", err)
|
|
}
|
|
if !hasLive {
|
|
t.Fatal("expected live entries with empty .ecj")
|
|
}
|
|
}
|
|
|
|
// TestDecodeWithNoEcjFile verifies that the decode flow works when no .ecj
|
|
// file exists at all.
|
|
func TestDecodeWithNoEcjFile(t *testing.T) {
|
|
dir := t.TempDir()
|
|
base := filepath.Join(dir, "test_1")
|
|
|
|
// .ecx: one live entry
|
|
needle1 := makeNeedleMapEntry(types.NeedleId(1), types.ToOffset(64), types.Size(100))
|
|
if err := os.WriteFile(base+".ecx", needle1, 0644); err != nil {
|
|
t.Fatalf("write ecx: %v", err)
|
|
}
|
|
|
|
// No .ecj file
|
|
|
|
// RebuildEcxFile should be a no-op
|
|
if err := erasure_coding.RebuildEcxFile(base); err != nil {
|
|
t.Fatalf("RebuildEcxFile: %v", err)
|
|
}
|
|
|
|
hasLive, err := erasure_coding.HasLiveNeedles(base)
|
|
if err != nil {
|
|
t.Fatalf("HasLiveNeedles: %v", err)
|
|
}
|
|
if !hasLive {
|
|
t.Fatal("expected live entries without .ecj file")
|
|
}
|
|
}
|
|
|
|
// TestEcxFileDeletionVisibleAfterSync verifies that deletions made to .ecx
|
|
// via MarkNeedleDeleted are visible to other readers after Sync().
|
|
// This is a regression test for issue #7751.
|
|
func TestEcxFileDeletionVisibleAfterSync(t *testing.T) {
|
|
dir := t.TempDir()
|
|
base := filepath.Join(dir, "foo_1")
|
|
|
|
// Create an .ecx file with one live needle
|
|
needle1 := makeNeedleMapEntry(types.NeedleId(1), types.ToOffset(64), types.Size(100))
|
|
if err := os.WriteFile(base+".ecx", needle1, 0644); err != nil {
|
|
t.Fatalf("write ecx: %v", err)
|
|
}
|
|
|
|
// Open the file for writing (simulating what EcVolume does)
|
|
ecxFile, err := os.OpenFile(base+".ecx", os.O_RDWR, 0644)
|
|
if err != nil {
|
|
t.Fatalf("open ecx: %v", err)
|
|
}
|
|
|
|
// Mark needle as deleted using MarkNeedleDeleted
|
|
err = erasure_coding.MarkNeedleDeleted(ecxFile, 0)
|
|
if err != nil {
|
|
ecxFile.Close()
|
|
t.Fatalf("MarkNeedleDeleted: %v", err)
|
|
}
|
|
|
|
// Sync the file to ensure changes are visible to other readers
|
|
if err := ecxFile.Sync(); err != nil {
|
|
ecxFile.Close()
|
|
t.Fatalf("Sync: %v", err)
|
|
}
|
|
ecxFile.Close()
|
|
|
|
// Now open with a new file handle and verify deletion is visible
|
|
data, err := os.ReadFile(base + ".ecx")
|
|
if err != nil {
|
|
t.Fatalf("read ecx: %v", err)
|
|
}
|
|
|
|
size := types.BytesToSize(data[types.NeedleIdSize+types.OffsetSize:])
|
|
if !size.IsDeleted() {
|
|
t.Fatalf("expected needle to be marked as deleted after sync, got size: %d", size)
|
|
}
|
|
}
|
|
|
|
// TestEcxFileDeletionNotVisibleWithoutSync verifies that without Sync(),
|
|
// deletions may not be visible to other readers (demonstrating the bug).
|
|
// Note: This test may be flaky depending on OS caching behavior, but it
|
|
// documents the expected behavior.
|
|
func TestEcxFileDeletionWithSeparateHandles(t *testing.T) {
|
|
dir := t.TempDir()
|
|
base := filepath.Join(dir, "foo_1")
|
|
|
|
// Create an .ecx file with one live needle
|
|
needle1 := makeNeedleMapEntry(types.NeedleId(1), types.ToOffset(64), types.Size(100))
|
|
if err := os.WriteFile(base+".ecx", needle1, 0644); err != nil {
|
|
t.Fatalf("write ecx: %v", err)
|
|
}
|
|
|
|
// Open the file for writing (writer handle)
|
|
writerFile, err := os.OpenFile(base+".ecx", os.O_RDWR, 0644)
|
|
if err != nil {
|
|
t.Fatalf("open ecx for write: %v", err)
|
|
}
|
|
defer writerFile.Close()
|
|
|
|
// Open the file for reading (reader handle - simulating CopyFile behavior)
|
|
readerFile, err := os.OpenFile(base+".ecx", os.O_RDONLY, 0644)
|
|
if err != nil {
|
|
t.Fatalf("open ecx for read: %v", err)
|
|
}
|
|
defer readerFile.Close()
|
|
|
|
// Mark needle as deleted via writer handle
|
|
err = erasure_coding.MarkNeedleDeleted(writerFile, 0)
|
|
if err != nil {
|
|
t.Fatalf("MarkNeedleDeleted: %v", err)
|
|
}
|
|
|
|
// Sync the writer to flush changes
|
|
if err := writerFile.Sync(); err != nil {
|
|
t.Fatalf("Sync: %v", err)
|
|
}
|
|
|
|
// Read via reader handle - after sync, changes should be visible
|
|
data := make([]byte, types.NeedleIdSize+types.OffsetSize+types.SizeSize)
|
|
if _, err := readerFile.ReadAt(data, 0); err != nil {
|
|
t.Fatalf("ReadAt: %v", err)
|
|
}
|
|
|
|
size := types.BytesToSize(data[types.NeedleIdSize+types.OffsetSize:])
|
|
if !size.IsDeleted() {
|
|
t.Fatalf("expected deletion to be visible after Sync(), got size: %d", size)
|
|
}
|
|
}
|
|
|
|
// TestEcVolumeDeleteDurableToJournal tracks issue #7751: a runtime needle
|
|
// delete must be observable by the ec.decode CopyFile path. Under the
|
|
// current design .ecx is an immutable sealed index at runtime — deletes
|
|
// are journaled to .ecj and tracked in an in-memory set — so the
|
|
// durability chain decode relies on is:
|
|
//
|
|
// 1. DeleteNeedleFromEcx appends the needle id to .ecj and fsyncs it.
|
|
// 2. Runtime reads via FindNeedleFromEcx consult the in-memory set and
|
|
// return TombstoneFileSize even though the sealed .ecx record on
|
|
// disk still shows the original size.
|
|
// 3. ec.decode later closes the EcVolume and calls RebuildEcxFile on
|
|
// the now-quiescent files, which walks .ecj and writes tombstones
|
|
// into .ecx. CopyFile then reads the rebuilt .ecx.
|
|
//
|
|
// This test exercises the full chain on a tempdir fixture.
|
|
func TestEcVolumeDeleteDurableToJournal(t *testing.T) {
|
|
dir := t.TempDir()
|
|
|
|
collection := "test"
|
|
vid := 1
|
|
base := filepath.Join(dir, collection+"_1")
|
|
|
|
// Seed .ecx with two live needles.
|
|
needle1 := makeNeedleMapEntry(types.NeedleId(1), types.ToOffset(64), types.Size(100))
|
|
needle2 := makeNeedleMapEntry(types.NeedleId(2), types.ToOffset(128), types.Size(200))
|
|
ecxData := append(needle1, needle2...)
|
|
if err := os.WriteFile(base+".ecx", ecxData, 0644); err != nil {
|
|
t.Fatalf("write ecx: %v", err)
|
|
}
|
|
if err := os.WriteFile(base+".ecj", []byte{}, 0644); err != nil {
|
|
t.Fatalf("write ecj: %v", err)
|
|
}
|
|
if err := os.WriteFile(base+".ec00", make([]byte, 8), 0644); err != nil {
|
|
t.Fatalf("write ec00: %v", err)
|
|
}
|
|
if err := os.WriteFile(base+".vif", []byte{}, 0644); err != nil {
|
|
t.Fatalf("write vif: %v", err)
|
|
}
|
|
|
|
ecVolume, err := erasure_coding.NewEcVolume("hdd", dir, dir, collection, needle.VolumeId(vid))
|
|
if err != nil {
|
|
t.Fatalf("NewEcVolume: %v", err)
|
|
}
|
|
|
|
// Runtime delete must not mutate .ecx.
|
|
if err := ecVolume.DeleteNeedleFromEcx(types.NeedleId(2)); err != nil {
|
|
t.Fatalf("DeleteNeedleFromEcx: %v", err)
|
|
}
|
|
|
|
// FindNeedleFromEcx masks the id via the in-memory set.
|
|
_, size, err := ecVolume.FindNeedleFromEcx(types.NeedleId(2))
|
|
if err != nil {
|
|
t.Fatalf("FindNeedleFromEcx(2): %v", err)
|
|
}
|
|
if !size.IsDeleted() {
|
|
t.Fatalf("expected FindNeedleFromEcx to return tombstone for deleted needle, got size=%d", size)
|
|
}
|
|
|
|
// Direct .ecx reader should still see the original size — .ecx is
|
|
// immutable at runtime.
|
|
entrySize := int64(types.NeedleIdSize + types.OffsetSize + types.SizeSize)
|
|
rawBuf := make([]byte, entrySize)
|
|
rawReader, err := os.Open(base + ".ecx")
|
|
if err != nil {
|
|
t.Fatalf("open ecx raw: %v", err)
|
|
}
|
|
if _, err := rawReader.ReadAt(rawBuf, entrySize); err != nil {
|
|
t.Fatalf("read raw ecx entry: %v", err)
|
|
}
|
|
rawReader.Close()
|
|
rawSize := types.BytesToSize(rawBuf[types.NeedleIdSize+types.OffsetSize:])
|
|
if rawSize.IsDeleted() {
|
|
t.Fatalf("runtime delete must not mutate .ecx on disk; got tombstone in raw entry")
|
|
}
|
|
|
|
// Close the volume so RebuildEcxFile can operate on the files, then
|
|
// fold .ecj into .ecx as the decode path does and verify the rebuilt
|
|
// index has the tombstone visible to external readers.
|
|
ecVolume.Close()
|
|
if err := erasure_coding.RebuildEcxFile(base); err != nil {
|
|
t.Fatalf("RebuildEcxFile: %v", err)
|
|
}
|
|
rebuilt, err := os.Open(base + ".ecx")
|
|
if err != nil {
|
|
t.Fatalf("open rebuilt ecx: %v", err)
|
|
}
|
|
defer rebuilt.Close()
|
|
if _, err := rebuilt.ReadAt(rawBuf, entrySize); err != nil {
|
|
t.Fatalf("read rebuilt ecx entry: %v", err)
|
|
}
|
|
rebuiltSize := types.BytesToSize(rawBuf[types.NeedleIdSize+types.OffsetSize:])
|
|
if !rebuiltSize.IsDeleted() {
|
|
t.Fatalf("expected needle 2 to be tombstoned in rebuilt .ecx, got size=%d", rebuiltSize)
|
|
}
|
|
}
|