Function TestHintFileIntegration_CrashRecovery

hintfile_integration_test.go:387–443

TestHintFileIntegration_CrashRecovery tests database recovery after simulated crashes

(t *testing.T)

// TestHintFileIntegration_CrashRecovery tests database recovery after simulated crashes
func TestHintFileIntegration_CrashRecovery(t *testing.T) {
	if runtime.GOOS == "windows" {
		t.Skip()
	}
	runForMergeModes(t, func(t *testing.T, mode mergeTestMode) {
		bucket := "bucket"
		opts := DefaultOptions
		opts.SegmentSize = 64 * KB
		opts.Dir = filepath.Join(t.TempDir(), fmt.Sprintf("test-hintfile-crash-%s", mode.name))
		opts.EnableHintFile = true
		opts.EnableMergeV2 = mode.enableMergeV2

		removeDir(opts.Dir)

		// Create database and write data
		db, err := Open(opts)
		require.NoError(t, err)
		txCreateBucket(t, db, DataStructureBTree, bucket, nil)

		n := 2000
		for i := 0; i < n; i++ {
			txPut(t, db, bucket, testutils.GetTestBytes(i), testutils.GetTestBytes(i), Persistent, nil, nil)
		}

		// Perform merge to create hint files
		require.NoError(t, db.Merge())

		// Simulate crash by not properly closing the database
		// (just release resources without calling Close())
		if err := db.ActiveFile.rwManager.Release(); err != nil {
			t.Errorf("Failed to release rwManager: %v", err)
		}
		if err := db.fm.Close(); err != nil {
			t.Errorf("Failed to close file manager: %v", err)
		}
		if err := db.flock.Unlock(); err != nil {
			t.Errorf("Failed to unlock file lock: %v", err)
		}

		// Restart database (should handle incomplete shutdown gracefully)
		db, err = Open(opts)
		require.NoError(t, err)

		// Verify all data is correctly recovered
		for i := 0; i < n; i++ {
			txGet(t, db, bucket, testutils.GetTestBytes(i), testutils.GetTestBytes(i), nil)
		}

		// Verify record count
		dbCnt, err := db.getRecordCount()
		require.NoError(t, err)
		require.Equal(t, int64(n), dbCnt)

		require.NoError(t, db.Close())
		removeDir(opts.Dir)
	})
}