docs: update hook docs

liasica · Sep 1, 2021 · 875d115 · 875d115
1 parent c0e8dff
commit 875d115
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 9 deletions.
diff --git a/README.md b/README.md
@@ -230,11 +230,11 @@ Or
         Name         string    `bson:"name"`
         Age          int       `bson:"age"`
     }
-    func (u *User) BeforeInsert() error {
+    func (u *User) BeforeInsert(ctx context.Context) error {
         fmt.Println("before insert called")
         return nil
     }
-    func (u *User) AfterInsert() error {
+    func (u *User) AfterInsert(ctx context.Context) error {
         fmt.Println("after insert called")
         return nil
     }

diff --git a/README_ZH.md b/README_ZH.md
@@ -230,11 +230,11 @@ go get github.com/qiniu/qmgo
         Name         string    `bson:"name"`
         Age          int       `bson:"age"`
     }
-    func (u *User) BeforeInsert() error {
+    func (u *User) BeforeInsert(ctx context.Context) error {
         fmt.Println("before insert called")
         return nil
     }
-    func (u *User) AfterInsert() error {
+    func (u *User) AfterInsert(ctx context.Context) error {
         fmt.Println("after insert called")
         return nil
     }

diff --git a/middleware/middleware_test.go b/middleware/middleware_test.go
@@ -13,16 +13,16 @@ func TestMiddleware(t *testing.T) {
 	ast := require.New(t)
 	ctx := context.Background()
 	// not register
-	ast.NoError(Do("success", operator.BeforeInsert, ctx))
+	ast.NoError(Do(ctx, "success", operator.BeforeInsert))
 
 	// valid register
 	Register(callbackTest)
-	ast.NoError(Do("success", operator.BeforeInsert, ctx))
-	ast.Error(Do("failure", operator.BeforeUpsert, ctx))
-	ast.NoError(Do("failure", operator.BeforeUpdate, ctx, "success"))
+	ast.NoError(Do(ctx, "success", operator.BeforeInsert))
+	ast.Error(Do(ctx, "failure", operator.BeforeUpsert))
+	ast.NoError(Do(ctx, "failure", operator.BeforeUpdate, "success"))
 }
 
-func callbackTest(doc interface{}, opType operator.OpType, ctx context.Context, opts ...interface{}) error {
+func callbackTest(ctx context.Context, doc interface{}, opType operator.OpType, opts ...interface{}) error {
 	if doc.(string) == "success" && opType == operator.BeforeInsert {
 		return nil
 	}