diff --git a/command.go b/command.go
index 5674e7d4d839f6a372af8631c75a0416392b24a5..d7be14e5ad356fcb3cb13373b326fbe3c274a130 100644
--- a/command.go
+++ b/command.go
@@ -40,7 +40,7 @@ type ICommand interface {
 
 // ICommandMain Command 的业务逻辑，对应 Build + Act 方法
 type ICommandMain interface {
-	Main(ctx context.Context, repo Repository) (err error)
+	Main(ctx context.Context, repo *Repository) (err error)
 }
 
 // ICommandInit Command 的初始化方法，会在锁和事务之前执行，可进行数据校验，前置准备工作
diff --git a/command_test.go b/command_test.go
index 873ac794f9b6e17fa7bc1b2f2e557babf3bb6b02..0518922424275f98b0881ff90960b3056546d2f3 100644
--- a/command_test.go
+++ b/command_test.go
@@ -81,14 +81,13 @@ func TestCommandMain(t *testing.T) {
 
 	var id string
 
-	res := engine.NewStage().Main(func(ctx context.Context, repo Repository) error {
+	res := engine.NewStage().Main(func(ctx context.Context, repo *Repository) error {
 		// 创建
 		o := &order{
 			Title: "testCreate",
 		}
-		if err := repo.Create(o); err != nil {
-			return err
-		}
+		repo.Add(o)
+
 		if err := repo.Save(ctx); err != nil {
 			return err
 		}
@@ -100,7 +99,7 @@ func TestCommandMain(t *testing.T) {
 	assert.NoError(t, res.Error)
 	assert.Contains(t, db.Data, id)
 
-	res = engine.NewStage().Main(func(ctx context.Context, repo Repository) error {
+	res = engine.NewStage().Main(func(ctx context.Context, repo *Repository) error {
 		o := &order{BaseEntity: NewBase(id)}
 		if err := repo.Get(ctx, o); err != nil {
 			return err
@@ -126,9 +125,10 @@ func TestCommandMain(t *testing.T) {
 	po := db.Data[id]
 	assert.Equal(t, "update_2", po.Name)
 
-	res = engine.NewStage().Main(func(ctx context.Context, repo Repository) error {
+	res = engine.NewStage().Main(func(ctx context.Context, repo *Repository) error {
 		o := &order{BaseEntity: NewBase(id)}
-		return repo.Delete(o)
+		repo.Remove(o)
+		return nil
 	}).Save(ctx)
 
 	assert.NoError(t, res.Error)
diff --git a/engine.go b/engine.go
index aa320875782dda2ba8621ef09aa0ba7b1b7b7a6a..b221dc168288108424c2d9e092caf5c21440d700 100644
--- a/engine.go
+++ b/engine.go
@@ -182,6 +182,15 @@ func (h *RootContainer) Remove(root IEntity) {
 // Repository 聚合根实体仓库
 type Repository struct {
 	stage *Stage
+	errs  []error
+}
+
+func (r *Repository) appendError(e error) {
+	r.errs = append(r.errs, e)
+}
+
+func (r *Repository) getError() error {
+	return errors.Join(r.errs...)
 }
 
 // Get 查询并构建聚合根
@@ -202,8 +211,8 @@ func (r *Repository) Get(ctx context.Context, root IEntity, children ...interfac
 	return r.stage.updateSnapshot()
 }
 
-// GetManual 自定义函数获取聚合根实体，并添加到快照
-func (r *Repository) GetManual(ctx context.Context, getter func(ctx context.Context, root ...IEntity), roots ...IEntity) error {
+// CustomGet 自定义函数获取聚合根实体，并添加到快照
+func (r *Repository) CustomGet(ctx context.Context, getter func(ctx context.Context, root ...IEntity), roots ...IEntity) error {
 	getter(ctx, roots...)
 
 	for _, root := range roots {
@@ -218,21 +227,22 @@ func (r *Repository) GetManual(ctx context.Context, getter func(ctx context.Cont
 	return r.stage.updateSnapshot()
 }
 
-// Create 创建聚合根
-func (r *Repository) Create(roots ...IEntity) error {
+// Add 添加聚合根到仓库
+func (r *Repository) Add(roots ...IEntity) {
 	for _, root := range roots {
 		if r.stage.hasSnapshot(root) {
-			return fmt.Errorf("root must be a new entity")
+			r.appendError(fmt.Errorf("root must be a new entity"))
+			return
 		}
 		if err := r.stage.meta.Add(root); err != nil {
-			return err
+			r.appendError(err)
+			return
 		}
 	}
-	return nil
 }
 
-// Delete 删除聚合根，root.GetID 不能为空
-func (r *Repository) Delete(roots ...IEntity) error {
+// Remove 删除聚合根，root.GetID 不能为空
+func (r *Repository) Remove(roots ...IEntity) {
 	toCreate := make([]IEntity, 0)
 	for _, root := range roots {
 		if !r.stage.meta.Has(root) {
@@ -240,30 +250,32 @@ func (r *Repository) Delete(roots ...IEntity) error {
 		}
 	}
 	if len(toCreate) > 0 {
-		if err := r.Create(toCreate...); err != nil {
-			return err
-		}
+		r.Add(toCreate...)
 		if err := r.stage.updateSnapshot(); err != nil {
-			return err
+			r.appendError(err)
+			return
 		}
 	}
 
 	for _, root := range roots {
 		if err := r.stage.meta.Remove(root); err != nil {
-			return err
+			r.appendError(err)
+			return
 		}
 	}
-	return nil
 }
 
 // Save 执行一次保存，并刷新快照
 func (r *Repository) Save(ctx context.Context) error {
+	if err := r.getError(); err != nil {
+		return err
+	}
 	return r.stage.commit(ctx)
 }
 
 type BuildFunc func(ctx context.Context, h DomainBuilder) (roots []IEntity, err error)
 type ActFunc func(ctx context.Context, container RootContainer, roots ...IEntity) error
-type MainFunc func(ctx context.Context, repo Repository) error
+type MainFunc func(ctx context.Context, repo *Repository) error
 type PostSaveFunc func(ctx context.Context, res *Result)
 
 // EventHandlerConstruct EventHandler 的构造函数，带一个入参和一个返回值，入参是与事件类型匹配的事件数据指针类型，返回值是 ICommand
@@ -427,14 +439,16 @@ func (e *Engine) NewStage() *Stage {
 }
 
 func (e *Engine) Create(ctx context.Context, roots ...IEntity) *Result {
-	return e.NewStage().Main(func(ctx context.Context, repo Repository) error {
-		return repo.Create(roots...)
+	return e.NewStage().Main(func(ctx context.Context, repo *Repository) error {
+		repo.Add(roots...)
+		return nil
 	}).Save(ctx)
 }
 
 func (e *Engine) Delete(ctx context.Context, roots ...IEntity) *Result {
-	return e.NewStage().Main(func(ctx context.Context, repo Repository) error {
-		return repo.Delete(roots...)
+	return e.NewStage().Main(func(ctx context.Context, repo *Repository) error {
+		repo.Remove(roots...)
+		return nil
 	}).Save(ctx)
 }
 
@@ -703,7 +717,7 @@ func (e *Stage) runCommand(ctx context.Context, c ICommand) *Result {
 	if err != nil {
 		return ResultErrOrBreak(err)
 	}
-	return e.WithOption(PostSaveOption(c.PostSave)).Lock(keys...).Main(func(ctx context.Context, repo Repository) error {
+	return e.WithOption(PostSaveOption(c.PostSave)).Lock(keys...).Main(func(ctx context.Context, repo *Repository) error {
 		buildRoots, err := c.Build(ctx, DomainBuilder{stage: repo.stage})
 		if err != nil {
 			return err
@@ -752,7 +766,7 @@ func (e *Stage) Run(ctx context.Context, cmd interface{}) *Result {
 			options = append(options, PostSaveOption(cmdPostSave.PostSave))
 		}
 		return e.WithOption(options...).Lock(keys...).Main(c.Main).Save(ctx)
-	case func(ctx context.Context, repo Repository) error:
+	case func(ctx context.Context, repo *Repository) error:
 		return e.Main(c).Save(ctx)
 	}
 	panic("cmd is invalid")
@@ -1078,9 +1092,13 @@ func (e *Stage) do(ctx context.Context) *Result {
 	// 创建聚合
 	var err error
 	if e.main != nil {
-		if err := e.main(ctx, Repository{stage: e}); err != nil {
+		repo := &Repository{stage: e}
+		if err := e.main(ctx, repo); err != nil {
 			return ResultErrOrBreak(err)
 		}
+		if err := repo.getError(); err != nil {
+			return ResultError(err)
+		}
 	}
 
 	err = e.persist(ctx)
diff --git a/engine_test.go b/engine_test.go
index 148ff878d762125e4853763bb3a432f4becf0b4c..8dcae88a5339aa2aa4af94f952fdd96a8136392e 100644
--- a/engine_test.go
+++ b/engine_test.go
@@ -205,7 +205,7 @@ func TestRootContainer(t *testing.T) {
 
 func TestBuildError(t *testing.T) {
 	ctx := context.Background()
-	res := NewEngine(nil, nil, WithoutTransaction).NewStage().Main(func(ctx context.Context, repo Repository) error {
+	res := NewEngine(nil, nil, WithoutTransaction).NewStage().Main(func(ctx context.Context, repo *Repository) error {
 		return fmt.Errorf("test")
 	}).Save(ctx)
 	assert.Error(t, res.Error)
@@ -266,8 +266,8 @@ func TestEntityMove(t *testing.T) {
 	engine := NewEngine(testsuit.NewMemLock(), &MapExecutor{DB: &db})
 	engine.Create(ctx, testOrder1, testOrder2)
 
-	res := engine.NewStage().Main(func(ctx context.Context, repo Repository) error {
-		if err := repo.GetManual(ctx, func(ctx context.Context, root ...IEntity) {}, testOrder1, testOrder2); err != nil {
+	res := engine.NewStage().Main(func(ctx context.Context, repo *Repository) error {
+		if err := repo.CustomGet(ctx, func(ctx context.Context, root ...IEntity) {}, testOrder1, testOrder2); err != nil {
 			return err
 		}
 		testOrder1.Products = nil
@@ -443,12 +443,13 @@ func TestEventPersist(t *testing.T) {
 				ID:   event.ID,
 				Name: string(event.Type),
 			}, nil
-		})).Run(ctx, func(ctx context.Context, repo Repository) error {
+		})).Run(ctx, func(ctx context.Context, repo *Repository) error {
 		testOrder = &order{
 			Title: "order1",
 		}
 		testOrder.AddEvent(&testEvent{Data: "hello"})
-		return repo.Create(testOrder)
+		repo.Add(testOrder)
+		return nil
 	})
 	assert.NoError(t, res.Error)
 	assert.NotEmpty(t, testOrder.GetEvents())
@@ -499,8 +500,8 @@ func BenchmarkUpdateOrders(b *testing.B) {
 		}
 		title := fmt.Sprintf("update %d", j)
 		newID := fmt.Sprintf("new%d", j)
-		res := NewEngine(nil, &MapExecutor{DB: &db}).Run(context.Background(), func(ctx context.Context, repo Repository) error {
-			if err := repo.GetManual(ctx, func(ctx context.Context, root ...IEntity) {}, testOrder); err != nil {
+		res := NewEngine(nil, &MapExecutor{DB: &db}).Run(context.Background(), func(ctx context.Context, repo *Repository) error {
+			if err := repo.CustomGet(ctx, func(ctx context.Context, root ...IEntity) {}, testOrder); err != nil {
 				return err
 			}
 			testOrder.Title = title
diff --git a/eventbus/mysql/eventbus_test.go b/eventbus/mysql/eventbus_test.go
index 4b1a8a6a58d83e6973a04d9b9a4e61e26a851f30..3ceb115cb9c5b97b639c29ba36ba4a1c226ef4d4 100644
--- a/eventbus/mysql/eventbus_test.go
+++ b/eventbus/mysql/eventbus_test.go
@@ -266,11 +266,12 @@ func TestEngine(t *testing.T) {
 	eventBus.Start(ctx)
 
 	engine := dddfirework.NewEngine(nil, exec_mysql.NewExecutor(db), eventBus.Options()...)
-	res := engine.NewStage().Main(func(ctx context.Context, repo dddfirework.Repository) error {
+	res := engine.NewStage().Main(func(ctx context.Context, repo *dddfirework.Repository) error {
 		e := &testEntity{Name: "hello"}
 		e.AddEvent(&testEvent{EType: "test_engine", Data: e.Name})
 		e.AddEvent(&testEvent{EType: "test_engine_tx", Data: e.Name}, dddfirework.WithSendType(dddfirework.SendTypeTransaction))
-		return repo.Create(e)
+		repo.Add(e)
+		return nil
 	}).Save(ctx)
 
 	wg.Wait()
@@ -312,10 +313,11 @@ func TestTXChecker(t *testing.T) {
 	eventBus.Start(ctx)
 
 	engine := dddfirework.NewEngine(nil, &mockExecutor{}, eventBus.Options()...)
-	res := engine.NewStage().Main(func(ctx context.Context, repo dddfirework.Repository) error {
+	res := engine.NewStage().Main(func(ctx context.Context, repo *dddfirework.Repository) error {
 		e := &testEntity{Name: "test_commit_failed"}
 		e.AddEvent(&testEvent{EType: "test_commit_failed", Data: e.Name}, dddfirework.WithSendType(dddfirework.SendTypeTransaction))
-		return repo.Create(e)
+		repo.Add(e)
+		return nil
 	}).Save(ctx)
 
 	time.Sleep(time.Second * 1)
diff --git a/example/biz/sale/application/command/add_coupon.go b/example/biz/sale/application/command/add_coupon.go
index 6b88b455c98d4c2391d8d1a29b2969df8950fcb3..b426a7a9a2cf23697fea57f28a97f37c9b9b97ad 100644
--- a/example/biz/sale/application/command/add_coupon.go
+++ b/example/biz/sale/application/command/add_coupon.go
@@ -20,13 +20,10 @@ import (
 
 	ddd "github.com/bytedance/dddfirework"
 	"github.com/bytedance/dddfirework/example/biz/sale/domain"
-	"github.com/bytedance/dddfirework/example/biz/sale/infrastructure/repo"
 	"github.com/bytedance/dddfirework/example/common/dto/sale"
 )
 
 type AddCouponCommand struct {
-	ddd.Command
-
 	orderID string
 	coupon  *sale.Coupon
 }
@@ -42,16 +39,11 @@ func (c *AddCouponCommand) Init(ctx context.Context) ([]string, error) {
 	return []string{c.orderID}, nil
 }
 
-func (c *AddCouponCommand) Build(ctx context.Context, builder ddd.DomainBuilder) (roots []ddd.IEntity, err error) {
-	order, err := repo.GetOrder(ctx, builder, c.orderID)
-	if err != nil {
-		return nil, err
+func (c *AddCouponCommand) Main(ctx context.Context, repo *ddd.Repository) error {
+	order := &domain.Order{ID: c.orderID}
+	if err := repo.Get(ctx, order); err != nil {
+		return err
 	}
-	return []ddd.IEntity{order}, nil
-}
-
-func (c *AddCouponCommand) Act(ctx context.Context, container ddd.RootContainer, roots ...ddd.IEntity) error {
-	order := roots[0].(*domain.Order)
 	if err := order.AddCoupon(c.coupon.ID, c.coupon.Rule, c.coupon.Discount); err != nil {
 		return err
 	}
diff --git a/example/biz/sale/application/command/add_saleitem.go b/example/biz/sale/application/command/add_saleitem.go
index 4e5d5c2a78d9159bdea81b9e910af7ac4d635d33..8a7e27c850508fc7716975ff26b48e63c24cf7a0 100644
--- a/example/biz/sale/application/command/add_saleitem.go
+++ b/example/biz/sale/application/command/add_saleitem.go
@@ -20,13 +20,10 @@ import (
 
 	"github.com/bytedance/dddfirework"
 	"github.com/bytedance/dddfirework/example/biz/sale/domain"
-	"github.com/bytedance/dddfirework/example/biz/sale/infrastructure/repo"
 	"github.com/bytedance/dddfirework/example/common/dto/sale"
 )
 
 type AddSaleItemCommand struct {
-	dddfirework.Command
-
 	orderID string
 	item    *sale.SaleItem
 }
@@ -39,16 +36,11 @@ func (c *AddSaleItemCommand) Init(ctx context.Context) ([]string, error) {
 	return []string{c.orderID}, nil
 }
 
-func (c *AddSaleItemCommand) Build(ctx context.Context, h dddfirework.DomainBuilder) (roots []dddfirework.IEntity, err error) {
-	order, err := repo.GetOrder(ctx, h, c.orderID)
-	if err != nil {
-		return nil, err
+func (c *AddSaleItemCommand) Main(ctx context.Context, repo *dddfirework.Repository) error {
+	order := &domain.Order{ID: c.orderID}
+	if err := repo.Get(ctx, order); err != nil {
+		return err
 	}
-	return []dddfirework.IEntity{order}, nil
-}
-
-func (c *AddSaleItemCommand) Act(ctx context.Context, container dddfirework.RootContainer, roots ...dddfirework.IEntity) error {
-	order := roots[0].(*domain.Order)
 	order.AddSaleItem(c.item.Code, c.item.Name, c.item.Price, c.item.Count)
 	return nil
 }
diff --git a/example/biz/sale/application/command/create_order.go b/example/biz/sale/application/command/create_order.go
index 433410f3cd95db5a857ad9bbdd9d90073230f17f..e476a1e9fd0b4ed2aea878898a1e46eb135c4cdf 100644
--- a/example/biz/sale/application/command/create_order.go
+++ b/example/biz/sale/application/command/create_order.go
@@ -28,8 +28,6 @@ type CreateOrderResult struct {
 }
 
 type CreateOrderCommand struct {
-	dddfirework.Command
-
 	userID  string
 	items   []*sale.SaleItem
 	coupons []*sale.Coupon
@@ -45,26 +43,18 @@ func NewCreateOrderCommand(userID string, items []*sale.SaleItem, coupons []*sal
 	}
 }
 
-func (c *CreateOrderCommand) Act(ctx context.Context, container dddfirework.RootContainer, roots ...dddfirework.IEntity) error {
+func (c *CreateOrderCommand) Main(ctx context.Context, repo *dddfirework.Repository) error {
 	order, err := domain.NewOrder(c.userID, c.items, c.coupons)
 	if err != nil {
 		return err
 	}
 
-	container.Add(order)
+	repo.Add(order)
 
-	//Commit 操作为可选项，目的是为了即刻获得待新建的订单 ID，构造返回值
-	if err := c.Commit(ctx); err != nil {
+	// 持久化后为新实体自动设置 ID
+	if err := repo.Save(ctx); err != nil {
 		return err
 	}
-	c.Output(order)
+	c.Result = &CreateOrderResult{OrderID: order.ID}
 	return nil
 }
-
-func (c *CreateOrderCommand) PostSave(ctx context.Context, res *dddfirework.Result) {
-	if res.Error == nil {
-		c.Result = &CreateOrderResult{
-			OrderID: res.Output.(*domain.Order).ID,
-		}
-	}
-}
diff --git a/example/biz/sale/application/command/remove_order.go b/example/biz/sale/application/command/remove_order.go
index 8d2339d5699a8ec590e36a9dc8b117a85dddbf02..be849f6a062d2218c35d0c4f32d28cfa98f5c42c 100644
--- a/example/biz/sale/application/command/remove_order.go
+++ b/example/biz/sale/application/command/remove_order.go
@@ -23,8 +23,6 @@ import (
 )
 
 type DeleteOrderCommand struct {
-	dddfirework.Command
-
 	orderID string
 }
 
@@ -34,11 +32,7 @@ func NewDeleteOrderCommand(orderID string) *DeleteOrderCommand {
 	}
 }
 
-func (c *DeleteOrderCommand) Build(ctx context.Context, builder dddfirework.DomainBuilder) (roots []dddfirework.IEntity, err error) {
-	return []dddfirework.IEntity{&domain.Order{ID: c.orderID}}, nil
-}
-
-func (c *DeleteOrderCommand) Act(ctx context.Context, container dddfirework.RootContainer, roots ...dddfirework.IEntity) error {
-	container.Remove(roots[0])
+func (c *DeleteOrderCommand) Main(ctx context.Context, repo *dddfirework.Repository) error {
+	repo.Remove(&domain.Order{ID: c.orderID})
 	return nil
 }
diff --git a/example/biz/sale/application/command/update_order.go b/example/biz/sale/application/command/update_order.go
index 8120f6cd581255e977cc587f7a7bc963cf24525e..073e4535a06f901a26b331e8b3580ed350df7014 100644
--- a/example/biz/sale/application/command/update_order.go
+++ b/example/biz/sale/application/command/update_order.go
@@ -20,7 +20,6 @@ import (
 
 	"github.com/bytedance/dddfirework"
 	"github.com/bytedance/dddfirework/example/biz/sale/domain"
-	"github.com/bytedance/dddfirework/example/biz/sale/infrastructure/repo"
 )
 
 type UpdateOrderOpt struct {
@@ -28,8 +27,6 @@ type UpdateOrderOpt struct {
 }
 
 type UpdateOrderCommand struct {
-	dddfirework.Command
-
 	orderID string
 	opt     UpdateOrderOpt
 }
@@ -45,16 +42,11 @@ func (c *UpdateOrderCommand) Init(ctx context.Context) (lockIDs []string, err er
 	return []string{c.orderID}, nil
 }
 
-func (c *UpdateOrderCommand) Build(ctx context.Context, builder dddfirework.DomainBuilder) (roots []dddfirework.IEntity, err error) {
-	order, err := repo.GetOrder(ctx, builder, c.orderID)
-	if err != nil {
-		return nil, err
+func (c *UpdateOrderCommand) Main(ctx context.Context, repo *dddfirework.Repository) error {
+	order := &domain.Order{ID: c.orderID}
+	if err := repo.Get(ctx, order); err != nil {
+		return err
 	}
-	return []dddfirework.IEntity{order}, nil
-}
-
-func (c *UpdateOrderCommand) Act(ctx context.Context, container dddfirework.RootContainer, roots ...dddfirework.IEntity) error {
-	order := roots[0].(*domain.Order)
 	order.Update(domain.UpdateOrderOpt{Remark: c.opt.Remark})
 	return nil
 }
diff --git a/example/handler/order.go b/example/handler/order.go
index bd39f90c97233634addbeeb3af5887260e569a8a..ee6f3dee3c3ee4e3b3eace105033031057a764d6 100644
--- a/example/handler/order.go
+++ b/example/handler/order.go
@@ -39,7 +39,7 @@ func (s *SaleServiceImpl) CreateOrder(ctx context.Context, req *CreateOrderReque
 	cmd := command.NewCreateOrderCommand(
 		req.User, req.Items, req.Coupons,
 	)
-	res := s.engine.RunCommand(ctx, cmd)
+	res := s.engine.Run(ctx, cmd)
 	if res.Error != nil {
 		return nil, res.Error
 	}
@@ -48,7 +48,7 @@ func (s *SaleServiceImpl) CreateOrder(ctx context.Context, req *CreateOrderReque
 
 // UpdateOrder implements the SaleServiceImpl interface.
 func (s *SaleServiceImpl) UpdateOrder(ctx context.Context, req *UpdateOrderRequest) (resp *UpdateOrderResponse, err error) {
-	if err := s.engine.RunCommand(ctx, command.NewUpdateOrderCommand(
+	if err := s.engine.Run(ctx, command.NewUpdateOrderCommand(
 		req.ID, command.UpdateOrderOpt{Remark: req.Remark},
 	)).Error; err != nil {
 		return nil, err
@@ -58,7 +58,7 @@ func (s *SaleServiceImpl) UpdateOrder(ctx context.Context, req *UpdateOrderReque
 
 // DeleteOrder implements the SaleServiceImpl interface.
 func (s *SaleServiceImpl) DeleteOrder(ctx context.Context, req *DeleteOrderRequest) (resp *DeleteOrderResponse, err error) {
-	if err := s.engine.RunCommand(ctx, command.NewDeleteOrderCommand(
+	if err := s.engine.Run(ctx, command.NewDeleteOrderCommand(
 		req.ID,
 	)).Error; err != nil {
 		return nil, err
@@ -100,7 +100,7 @@ func (s *SaleServiceImpl) GetOrderList(ctx context.Context, req *GetOrderListReq
 
 // AddSaleItem implements the SaleServiceImpl interface.
 func (s *SaleServiceImpl) AddSaleItem(ctx context.Context, req *AddSaleItemRequest) (resp *AddSaleItemResponse, err error) {
-	if err := s.engine.RunCommand(ctx, command.NewAddSaleItemCommand(
+	if err := s.engine.Run(ctx, command.NewAddSaleItemCommand(
 		req.OrderID, req.Item,
 	)).Error; err != nil {
 		return nil, err
@@ -110,7 +110,7 @@ func (s *SaleServiceImpl) AddSaleItem(ctx context.Context, req *AddSaleItemReque
 
 // AddCoupon implements the SaleServiceImpl interface.
 func (s *SaleServiceImpl) AddCoupon(ctx context.Context, req *AddCouponRequest) (resp *AddCouponResponse, err error) {
-	if err := s.engine.RunCommand(ctx, command.NewAddCouponCommand(
+	if err := s.engine.Run(ctx, command.NewAddCouponCommand(
 		req.OrderID, req.Coupon,
 	)).Error; err != nil {
 		return nil, err
diff --git a/executor/mysql/executor_test.go b/executor/mysql/executor_test.go
index 322c8c448776eb91cf1e377d704dc09a3b4c0c34..7e155993778b75ff07bbeb95c0eea362808a3eb0 100644
--- a/executor/mysql/executor_test.go
+++ b/executor/mysql/executor_test.go
@@ -341,14 +341,15 @@ func TestRollback(t *testing.T) {
 	ctx := context.Background()
 
 	engine := ddd.NewEngine(testsuit.NewMemLock(), NewExecutor(db))
-	res := engine.Run(ctx, func(ctx context.Context, repo ddd.Repository) error {
-		return repo.Create(&order{
+	res := engine.Run(ctx, func(ctx context.Context, repo *ddd.Repository) error {
+		repo.Add(&order{
 			ID:    "testrollback",
 			Title: "testrollback",
 		}, &order{
 			ID:    "testrollback",
 			Title: "testrollback",
 		})
+		return nil
 	})
 	assert.NotNil(t, res.Error)
 	assert.ErrorIs(t, db.First(&orderPO{}, "id = ?", "testrollback").Error, gorm.ErrRecordNotFound)
@@ -405,7 +406,7 @@ func BenchmarkUpdate(b *testing.B) {
 
 	engine := ddd.NewEngine(nil, NewExecutor(db))
 	for i := 0; i < b.N; i++ {
-		if res := engine.NewStage().Main(func(ctx context.Context, repo ddd.Repository) error {
+		if res := engine.NewStage().Main(func(ctx context.Context, repo *ddd.Repository) error {
 			testOrder := &order{
 				ID:       "order1",
 				User:     &user{},