fix(filer): remove cancellation guard from RollbackTransaction and clean up #8909 (#8916)

* fix(filer): remove cancellation guard from RollbackTransaction and clean up #8909

RollbackTransaction is a cleanup operation that must succeed even when
the context is cancelled — guarding it causes the exact orphaned state
that #8909 was trying to prevent.

Also:
- Use single-evaluation `if err := ctx.Err(); err != nil` pattern
  instead of double-calling ctx.Err()
- Remove spurious blank lines before guards
- Add context.DeadlineExceeded test coverage
- Simplify tests from ~230 lines to ~130 lines

* fix(filer): call cancel() in expiredCtx and test rollback with expired context

- Call cancel() instead of suppressing it to avoid leaking timer resources
- Test RollbackTransaction with both cancelled and expired contexts

weed/filer/filerstore_wrapper.go

@@ -128,10 +128,8 @@ func (fsw *FilerStoreWrapper) Initialize(configuration util.Configuration, prefi
 }
 
 func (fsw *FilerStoreWrapper) InsertEntry(ctx context.Context, entry *Entry) error {
-	// Check if context was already cancelled before ignoring cancellation
+	if err := ctx.Err(); err != nil {
-	// This prevents writing metadata for operations that have already failed
+		return err
-	if ctx.Err() != nil {
-		return ctx.Err()
 	}
 	ctx = context.WithoutCancel(ctx)
 	actualStore := fsw.getActualStore(entry.FullPath)
@@ -160,9 +158,8 @@ func (fsw *FilerStoreWrapper) InsertEntry(ctx context.Context, entry *Entry) err
 // InsertEntryKnownAbsent skips the pre-insert FindEntry path when the caller has
 // already established that the target path does not exist.
 func (fsw *FilerStoreWrapper) InsertEntryKnownAbsent(ctx context.Context, entry *Entry) error {
-
+	if err := ctx.Err(); err != nil {
-	if ctx.Err() != nil {
+		return err
-		return ctx.Err()
 	}
 	ctx = context.WithoutCancel(ctx)
 	actualStore := fsw.getActualStore(entry.FullPath)
@@ -187,9 +184,8 @@ func (fsw *FilerStoreWrapper) InsertEntryKnownAbsent(ctx context.Context, entry
 }
 
 func (fsw *FilerStoreWrapper) UpdateEntry(ctx context.Context, entry *Entry) error {
-
+	if err := ctx.Err(); err != nil {
-	if ctx.Err() != nil {
+		return err
-		return ctx.Err()
 	}
 	ctx = context.WithoutCancel(ctx)
 	actualStore := fsw.getActualStore(entry.FullPath)
@@ -249,9 +245,8 @@ func (fsw *FilerStoreWrapper) FindEntry(ctx context.Context, fp util.FullPath) (
 }
 
 func (fsw *FilerStoreWrapper) DeleteEntry(ctx context.Context, fp util.FullPath) (err error) {
-
+	if err := ctx.Err(); err != nil {
-	if ctx.Err() != nil {
+		return err
-		return ctx.Err()
 	}
 	ctx = context.WithoutCancel(ctx)
 	actualStore := fsw.getActualStore(fp)
@@ -281,9 +276,8 @@ func (fsw *FilerStoreWrapper) DeleteEntry(ctx context.Context, fp util.FullPath)
 }
 
 func (fsw *FilerStoreWrapper) DeleteOneEntry(ctx context.Context, existingEntry *Entry) (err error) {
-
+	if err := ctx.Err(); err != nil {
-	if ctx.Err() != nil {
+		return err
-		return ctx.Err()
 	}
 	ctx = context.WithoutCancel(ctx)
 	actualStore := fsw.getActualStore(existingEntry.FullPath)
@@ -309,9 +303,8 @@ func (fsw *FilerStoreWrapper) DeleteOneEntry(ctx context.Context, existingEntry
 }
 
 func (fsw *FilerStoreWrapper) DeleteFolderChildren(ctx context.Context, fp util.FullPath) (err error) {
-
+	if err := ctx.Err(); err != nil {
-	if ctx.Err() != nil {
+		return err
-		return ctx.Err()
 	}
 	ctx = context.WithoutCancel(ctx)
 	actualStore := fsw.getActualStore(fp + "/")
@@ -419,28 +412,22 @@ func (fsw *FilerStoreWrapper) prefixFilterEntries(ctx context.Context, dirPath u
 }
 
 func (fsw *FilerStoreWrapper) BeginTransaction(ctx context.Context) (context.Context, error) {
-
+	if err := ctx.Err(); err != nil {
-	if ctx.Err() != nil {
+		return nil, err
-		return nil, ctx.Err()
 	}
 	ctx = context.WithoutCancel(ctx)
 	return fsw.getDefaultStore().BeginTransaction(ctx)
 }
 
 func (fsw *FilerStoreWrapper) CommitTransaction(ctx context.Context) error {
-
+	if err := ctx.Err(); err != nil {
-	if ctx.Err() != nil {
+		return err
-		return ctx.Err()
 	}
 	ctx = context.WithoutCancel(ctx)
 	return fsw.getDefaultStore().CommitTransaction(ctx)
 }
 
 func (fsw *FilerStoreWrapper) RollbackTransaction(ctx context.Context) error {
-
-	if ctx.Err() != nil {
-		return ctx.Err()
-	}
 	ctx = context.WithoutCancel(ctx)
 	return fsw.getDefaultStore().RollbackTransaction(ctx)
 }
@@ -450,9 +437,8 @@ func (fsw *FilerStoreWrapper) Shutdown() {
 }
 
 func (fsw *FilerStoreWrapper) KvPut(ctx context.Context, key []byte, value []byte) (err error) {
-
+	if err := ctx.Err(); err != nil {
-	if ctx.Err() != nil {
+		return err
-		return ctx.Err()
 	}
 	ctx = context.WithoutCancel(ctx)
 	return fsw.getDefaultStore().KvPut(ctx, key, value)
@@ -462,9 +448,8 @@ func (fsw *FilerStoreWrapper) KvGet(ctx context.Context, key []byte) (value []by
 	return fsw.getDefaultStore().KvGet(ctx, key)
 }
 func (fsw *FilerStoreWrapper) KvDelete(ctx context.Context, key []byte) (err error) {
-
+	if err := ctx.Err(); err != nil {
-	if ctx.Err() != nil {
+		return err
-		return ctx.Err()
 	}
 	ctx = context.WithoutCancel(ctx)
 	return fsw.getDefaultStore().KvDelete(ctx, key)

weed/filer/filerstore_wrapper_test.go

@@ -5,6 +5,7 @@ import (
 	"errors"
 	"os"
 	"testing"
+	"time"
 
 	"github.com/seaweedfs/seaweedfs/weed/util"
 	"github.com/stretchr/testify/assert"
@@ -71,238 +72,136 @@ func TestFilerStoreWrapperMimeNormalization(t *testing.T) {
 	}
 }
 
-// TestFilerStoreWrapperContextCancellation verifies that write operations
+// cancelledCtx returns a context that is already cancelled.
-// respect context cancellation to prevent orphaned metadata
+func cancelledCtx() context.Context {
-func TestFilerStoreWrapperContextCancellation(t *testing.T) {
+	ctx, cancel := context.WithCancel(context.Background())
-	tests := []struct {
+	cancel()
-		name          string
+	return ctx
-		cancelContext bool
+}
-		expectError   bool
+
-		isWriteOp     bool
+// expiredCtx returns a context whose deadline has already passed.
-		run           func(*FilerStoreWrapper, context.Context, *Entry) error
+func expiredCtx() context.Context {
-	}{
+	ctx, cancel := context.WithDeadline(context.Background(), time.Now().Add(-time.Second))
-		{
+	cancel() // release resources immediately as it's already expired
-			name:          "InsertEntry with cancelled context fails",
+	return ctx
-			cancelContext: true,
+}
-			expectError:   true,
+
-			isWriteOp:     true,
+func TestFilerStoreWrapperWriteOpsRejectCancelledContext(t *testing.T) {
-			run: func(fsw *FilerStoreWrapper, ctx context.Context, entry *Entry) error {
+	newEntry := func(path string) *Entry {
-				return fsw.InsertEntry(ctx, entry)
+		return &Entry{
-			},
+			FullPath: util.FullPath(path),
-		},
+			Attr:     Attr{Mode: 0o660, Mime: "application/octet-stream"},
-		{
+		}
-			name:          "InsertEntry with active context succeeds",
-			cancelContext: false,
-			expectError:   false,
-			isWriteOp:     true,
-			run: func(fsw *FilerStoreWrapper, ctx context.Context, entry *Entry) error {
-				return fsw.InsertEntry(ctx, entry)
-			},
-		},
-		{
-			name:          "InsertEntryKnownAbsent with cancelled context fails",
-			cancelContext: true,
-			expectError:   true,
-			isWriteOp:     true,
-			run: func(fsw *FilerStoreWrapper, ctx context.Context, entry *Entry) error {
-				return fsw.InsertEntryKnownAbsent(ctx, entry)
-			},
-		},
-		{
-			name:          "UpdateEntry with cancelled context fails",
-			cancelContext: true,
-			expectError:   true,
-			isWriteOp:     true,
-			run: func(fsw *FilerStoreWrapper, ctx context.Context, entry *Entry) error {
-				_ = fsw.InsertEntry(context.Background(), entry)
-				entry.Attr.Mime = "updated"
-				return fsw.UpdateEntry(ctx, entry)
-			},
-		},
-		{
-			name:          "DeleteEntry with cancelled context fails",
-			cancelContext: true,
-			expectError:   true,
-			isWriteOp:     true,
-			run: func(fsw *FilerStoreWrapper, ctx context.Context, entry *Entry) error {
-				_ = fsw.InsertEntry(context.Background(), entry)
-				return fsw.DeleteEntry(ctx, entry.FullPath)
-			},
-		},
-		{
-			name:          "DeleteOneEntry with cancelled context fails",
-			cancelContext: true,
-			expectError:   true,
-			isWriteOp:     true,
-			run: func(fsw *FilerStoreWrapper, ctx context.Context, entry *Entry) error {
-				_ = fsw.InsertEntry(context.Background(), entry)
-				return fsw.DeleteOneEntry(ctx, entry)
-			},
-		},
-		{
-			name:          "DeleteFolderChildren with cancelled context fails",
-			cancelContext: true,
-			expectError:   true,
-			isWriteOp:     true,
-			run: func(fsw *FilerStoreWrapper, ctx context.Context, entry *Entry) error {
-				childEntry := &Entry{
-					FullPath: util.FullPath("/test/folder/child"),
-					Attr:     Attr{Mode: 0o660},
-				}
-				_ = fsw.InsertEntry(context.Background(), childEntry)
-				return fsw.DeleteFolderChildren(ctx, util.FullPath("/test/folder"))
-			},
-		},
-		{
-			name:          "KvPut with cancelled context fails",
-			cancelContext: true,
-			expectError:   true,
-			isWriteOp:     true,
-			run: func(fsw *FilerStoreWrapper, ctx context.Context, entry *Entry) error {
-				return fsw.KvPut(ctx, []byte("test-key"), []byte("test-value"))
-			},
-		},
-		{
-			name:          "KvDelete with cancelled context fails",
-			cancelContext: true,
-			expectError:   true,
-			isWriteOp:     true,
-			run: func(fsw *FilerStoreWrapper, ctx context.Context, entry *Entry) error {
-				_ = fsw.KvPut(context.Background(), []byte("test-key"), []byte("test-value"))
-				return fsw.KvDelete(ctx, []byte("test-key"))
-			},
-		},
-		{
-			name:          "FindEntry with cancelled context succeeds (read operation)",
-			cancelContext: true,
-			expectError:   false,
-			isWriteOp:     false,
-			run: func(fsw *FilerStoreWrapper, ctx context.Context, entry *Entry) error {
-				_ = fsw.InsertEntry(context.Background(), entry)
-				_, err := fsw.FindEntry(ctx, entry.FullPath)
-				return err
-			},
-		},
-		{
-			name:          "KvGet with cancelled context succeeds (read operation)",
-			cancelContext: true,
-			expectError:   false,
-			isWriteOp:     false,
-			run: func(fsw *FilerStoreWrapper, ctx context.Context, entry *Entry) error {
-				_ = fsw.KvPut(context.Background(), []byte("test-key"), []byte("test-value"))
-				_, err := fsw.KvGet(ctx, []byte("test-key"))
-				return err
-			},
-		},
 	}
 
-	for _, tt := range tests {
+	// Each write operation that should be guarded.
-		t.Run(tt.name, func(t *testing.T) {
+	writeOps := []struct {
-			store := newStubFilerStore()
+		name string
-			wrapper := NewFilerStoreWrapper(store)
+		run  func(*FilerStoreWrapper, context.Context) error
-			entry := &Entry{
+	}{
-				FullPath: util.FullPath("/test/object"),
+		{"InsertEntry", func(fsw *FilerStoreWrapper, ctx context.Context) error {
-				Attr: Attr{
+			return fsw.InsertEntry(ctx, newEntry("/test/a"))
-					Mode: 0o660,
+		}},
-					Mime: "application/octet-stream",
+		{"InsertEntryKnownAbsent", func(fsw *FilerStoreWrapper, ctx context.Context) error {
-				},
+			return fsw.InsertEntryKnownAbsent(ctx, newEntry("/test/b"))
-			}
+		}},
+		{"UpdateEntry", func(fsw *FilerStoreWrapper, ctx context.Context) error {
+			_ = fsw.InsertEntry(context.Background(), newEntry("/test/c"))
+			return fsw.UpdateEntry(ctx, newEntry("/test/c"))
+		}},
+		{"DeleteEntry", func(fsw *FilerStoreWrapper, ctx context.Context) error {
+			_ = fsw.InsertEntry(context.Background(), newEntry("/test/d"))
+			return fsw.DeleteEntry(ctx, "/test/d")
+		}},
+		{"DeleteOneEntry", func(fsw *FilerStoreWrapper, ctx context.Context) error {
+			e := newEntry("/test/e")
+			_ = fsw.InsertEntry(context.Background(), e)
+			return fsw.DeleteOneEntry(ctx, e)
+		}},
+		{"DeleteFolderChildren", func(fsw *FilerStoreWrapper, ctx context.Context) error {
+			_ = fsw.InsertEntry(context.Background(), newEntry("/test/folder/child"))
+			return fsw.DeleteFolderChildren(ctx, "/test/folder")
+		}},
+		{"BeginTransaction", func(fsw *FilerStoreWrapper, ctx context.Context) error {
+			_, err := fsw.BeginTransaction(ctx)
+			return err
+		}},
+		{"CommitTransaction", func(fsw *FilerStoreWrapper, ctx context.Context) error {
+			return fsw.CommitTransaction(ctx)
+		}},
+		{"KvPut", func(fsw *FilerStoreWrapper, ctx context.Context) error {
+			return fsw.KvPut(ctx, []byte("k"), []byte("v"))
+		}},
+		{"KvDelete", func(fsw *FilerStoreWrapper, ctx context.Context) error {
+			_ = fsw.KvPut(context.Background(), []byte("k"), []byte("v"))
+			return fsw.KvDelete(ctx, []byte("k"))
+		}},
+	}
 
-			ctx, cancel := context.WithCancel(context.Background())
+	badContexts := []struct {
-			if tt.cancelContext {
+		name      string
-				cancel()
+		ctx       context.Context
-			} else {
+		wantError error
-				defer cancel()
+	}{
-			}
+		{"cancelled", cancelledCtx(), context.Canceled},
+		{"deadline exceeded", expiredCtx(), context.DeadlineExceeded},
+	}
 
-			err := tt.run(wrapper, ctx, entry)
+	for _, op := range writeOps {
-
+		for _, bc := range badContexts {
-			if tt.expectError {
+			t.Run(op.name+"/"+bc.name, func(t *testing.T) {
+				wrapper := NewFilerStoreWrapper(newStubFilerStore())
+				err := op.run(wrapper, bc.ctx)
 				require.Error(t, err)
-				if tt.isWriteOp {
+				assert.True(t, errors.Is(err, bc.wantError), "got %v, want %v", err, bc.wantError)
-					assert.True(t, errors.Is(err, context.Canceled))
+			})
-				}
+		}
-			} else {
-				require.NoError(t, err)
-			}
-		})
 	}
 }
 
-// TestFilerStoreWrapperTransactionContextCancellation verifies that transaction
+func TestFilerStoreWrapperWriteOpsSucceedWithActiveContext(t *testing.T) {
-// operations respect context cancellation.
+	wrapper := NewFilerStoreWrapper(newStubFilerStore())
-func TestFilerStoreWrapperTransactionContextCancellation(t *testing.T) {
+	ctx := context.Background()
-	tests := []struct {
+	entry := &Entry{
-		name          string
+		FullPath: util.FullPath("/test/obj"),
-		cancelContext bool
+		Attr:     Attr{Mode: 0o660},
-		run           func(*FilerStoreWrapper, context.Context) error
-	}{
-		{
-			name:          "BeginTransaction with cancelled context fails",
-			cancelContext: true,
-			run: func(fsw *FilerStoreWrapper, ctx context.Context) error {
-				_, err := fsw.BeginTransaction(ctx)
-				return err
-			},
-		},
-		{
-			name:          "BeginTransaction with active context succeeds",
-			cancelContext: false,
-			run: func(fsw *FilerStoreWrapper, ctx context.Context) error {
-				_, err := fsw.BeginTransaction(ctx)
-				return err
-			},
-		},
-		{
-			name:          "CommitTransaction with cancelled context fails",
-			cancelContext: true,
-			run: func(fsw *FilerStoreWrapper, ctx context.Context) error {
-				return fsw.CommitTransaction(ctx)
-			},
-		},
-		{
-			name:          "CommitTransaction with active context succeeds",
-			cancelContext: false,
-			run: func(fsw *FilerStoreWrapper, ctx context.Context) error {
-				return fsw.CommitTransaction(ctx)
-			},
-		},
-		{
-			name:          "RollbackTransaction with cancelled context fails",
-			cancelContext: true,
-			run: func(fsw *FilerStoreWrapper, ctx context.Context) error {
-				return fsw.RollbackTransaction(ctx)
-			},
-		},
-		{
-			name:          "RollbackTransaction with active context succeeds",
-			cancelContext: false,
-			run: func(fsw *FilerStoreWrapper, ctx context.Context) error {
-				return fsw.RollbackTransaction(ctx)
-			},
-		},
 	}
 
-	for _, tt := range tests {
+	require.NoError(t, wrapper.InsertEntry(ctx, entry))
-		t.Run(tt.name, func(t *testing.T) {
+	require.NoError(t, wrapper.UpdateEntry(ctx, entry))
-			store := newStubFilerStore()
+	require.NoError(t, wrapper.DeleteOneEntry(ctx, entry))
-			wrapper := NewFilerStoreWrapper(store)
+	require.NoError(t, wrapper.InsertEntryKnownAbsent(ctx, entry))
+	require.NoError(t, wrapper.DeleteEntry(ctx, entry.FullPath))
+	require.NoError(t, wrapper.KvPut(ctx, []byte("k"), []byte("v")))
+	require.NoError(t, wrapper.KvDelete(ctx, []byte("k")))
 
-			ctx, cancel := context.WithCancel(context.Background())
+	txCtx, err := wrapper.BeginTransaction(ctx)
-			if tt.cancelContext {
+	require.NoError(t, err)
-				cancel()
+	require.NoError(t, wrapper.CommitTransaction(txCtx))
-			} else {
+}
-				defer cancel()
+
-			}
+func TestFilerStoreWrapperReadOpsSucceedWithCancelledContext(t *testing.T) {
-
+	wrapper := NewFilerStoreWrapper(newStubFilerStore())
-			err := tt.run(wrapper, ctx)
+	entry := &Entry{
-
+		FullPath: util.FullPath("/test/readable"),
-			if tt.cancelContext {
+		Attr:     Attr{Mode: 0o660},
-				require.Error(t, err)
+	}
-				assert.True(t, errors.Is(err, context.Canceled))
+	require.NoError(t, wrapper.InsertEntry(context.Background(), entry))
-			} else {
+	require.NoError(t, wrapper.KvPut(context.Background(), []byte("rk"), []byte("rv")))
-				require.NoError(t, err)
+
-			}
+	ctx := cancelledCtx()
-		})
+
-	}
+	_, err := wrapper.FindEntry(ctx, entry.FullPath)
+	assert.NoError(t, err)
+
+	_, err = wrapper.KvGet(ctx, []byte("rk"))
+	assert.NoError(t, err)
+}
+
+// RollbackTransaction must succeed even when the context is cancelled or
+// expired, because it is a cleanup operation called after failures.
+func TestFilerStoreWrapperRollbackSucceedsWithCancelledContext(t *testing.T) {
+	wrapper := NewFilerStoreWrapper(newStubFilerStore())
+	assert.NoError(t, wrapper.RollbackTransaction(cancelledCtx()))
+	assert.NoError(t, wrapper.RollbackTransaction(expiredCtx()))
 }