fix(callback): add missing model hook on InitializeCallbacks() (#97)

Author: chenmingyong0423

callback/callback.go

@@ -17,6 +17,8 @@ package callback
 import (
 	"context"
 
+	"github.com/chenmingyong0423/go-mongox/v2/internal/hook/model"
+
 	"github.com/chenmingyong0423/go-mongox/v2/internal/hook/field"
 
 	"github.com/chenmingyong0423/go-mongox/v2/operation"
@@ -33,30 +35,97 @@ func InitializeCallbacks() *Callback {
 					return field.Execute(ctx, opCtx, operation.OpTypeBeforeInsert, opts...)
 				},
 			},
+			{
+				name: "mongox:model",
+				fn: func(ctx context.Context, opCtx *operation.OpContext, opts ...any) error {
+					return model.Execute(ctx, opCtx, operation.OpTypeBeforeInsert, opts...)
+				},
+			},
+		},
+		afterInsert: []callbackHandler{
+			{
+				name: "mongox:model",
+				fn: func(ctx context.Context, opCtx *operation.OpContext, opts ...any) error {
+					return model.Execute(ctx, opCtx, operation.OpTypeAfterInsert, opts...)
+				},
+			},
 		},
-		afterInsert: make([]callbackHandler, 0),
 		beforeUpdate: []callbackHandler{
 			{
 				name: "mongox:fieds",
 				fn: func(ctx context.Context, opCtx *operation.OpContext, opts ...any) error {
 					return field.Execute(ctx, opCtx, operation.OpTypeBeforeUpdate, opts...)
 				},
 			},
+			{
+				name: "mongox:model",
+				fn: func(ctx context.Context, opCtx *operation.OpContext, opts ...any) error {
+					return model.Execute(ctx, opCtx, operation.OpTypeBeforeUpdate, opts...)
+				},
+			},
+		},
+		afterUpdate: []callbackHandler{
+			{
+				name: "mongox:model",
+				fn: func(ctx context.Context, opCtx *operation.OpContext, opts ...any) error {
+					return model.Execute(ctx, opCtx, operation.OpTypeAfterUpdate, opts...)
+				},
+			},
+		},
+		beforeDelete: []callbackHandler{
+			{
+				name: "mongox:model",
+				fn: func(ctx context.Context, opCtx *operation.OpContext, opts ...any) error {
+					return model.Execute(ctx, opCtx, operation.OpTypeBeforeDelete, opts...)
+				},
+			},
+		},
+		afterDelete: []callbackHandler{
+			{
+				name: "mongox:model",
+				fn: func(ctx context.Context, opCtx *operation.OpContext, opts ...any) error {
+					return model.Execute(ctx, opCtx, operation.OpTypeAfterDelete, opts...)
+				},
+			},
 		},
-		afterUpdate:  make([]callbackHandler, 0),
-		beforeDelete: make([]callbackHandler, 0),
-		afterDelete:  make([]callbackHandler, 0),
 		beforeUpsert: []callbackHandler{
 			{
 				name: "mongox:fieds",
 				fn: func(ctx context.Context, opCtx *operation.OpContext, opts ...any) error {
 					return field.Execute(ctx, opCtx, operation.OpTypeBeforeUpsert, opts...)
 				},
 			},
+			{
+				name: "mongox:model",
+				fn: func(ctx context.Context, opCtx *operation.OpContext, opts ...any) error {
+					return model.Execute(ctx, opCtx, operation.OpTypeBeforeUpsert, opts...)
+				},
+			},
+		},
+		afterUpsert: []callbackHandler{
+			{
+				name: "mongox:model",
+				fn: func(ctx context.Context, opCtx *operation.OpContext, opts ...any) error {
+					return model.Execute(ctx, opCtx, operation.OpTypeAfterUpsert, opts...)
+				},
+			},
+		},
+		beforeFind: []callbackHandler{
+			{
+				name: "mongox:model",
+				fn: func(ctx context.Context, opCtx *operation.OpContext, opts ...any) error {
+					return model.Execute(ctx, opCtx, operation.OpTypeBeforeFind, opts...)
+				},
+			},
+		},
+		afterFind: []callbackHandler{
+			{
+				name: "mongox:model",
+				fn: func(ctx context.Context, opCtx *operation.OpContext, opts ...any) error {
+					return model.Execute(ctx, opCtx, operation.OpTypeAfterFind, opts...)
+				},
+			},
 		},
-		afterUpsert: make([]callbackHandler, 0),
-		beforeFind:  make([]callbackHandler, 0),
-		afterFind:   make([]callbackHandler, 0),
 	}
 }