重构 callback

Author: chenmingyong0423

hook/field/field.go

@@ -21,8 +21,22 @@ import (
 	"github.com/chenmingyong0423/go-mongox/operation"
 )
 
+func getPayload(opCtx *operation.OpContext, opType operation.OpType) any {
+	switch opType {
+	case operation.OpTypeBeforeInsert, operation.OpTypeAfterInsert:
+		return opCtx.Doc
+	case operation.OpTypeBeforeUpdate, operation.OpTypeAfterUpdate:
+		return opCtx.Updates
+	case operation.OpTypeBeforeDelete, operation.OpTypeAfterDelete:
+		return opCtx.Filter
+	case operation.OpTypeBeforeUpsert, operation.OpTypeAfterUpsert:
+		return opCtx.Replacement
+	}
+	return nil
+}
+
 func Execute(ctx context.Context, opCtx *operation.OpContext, opType operation.OpType, opts ...any) error {
-	payLoad := opCtx.GetPayload(opType)
+	payLoad := getPayload(opCtx, opType)
 	if payLoad == nil {
 		return nil
 	}

operation/operation_type.go

@@ -41,17 +41,3 @@ type OpContext struct {
 	Updates     any
 	Replacement any
 }
-
-func (opCtx OpContext) GetPayload(opType OpType) any {
-	switch opType {
-	case OpTypeBeforeInsert, OpTypeAfterInsert:
-		return opCtx.Doc
-	case OpTypeBeforeUpdate, OpTypeAfterUpdate:
-		return opCtx.Updates
-	case OpTypeBeforeDelete, OpTypeAfterDelete:
-		return opCtx.Filter
-	case OpTypeBeforeUpsert, OpTypeAfterUpsert:
-		return opCtx.Replacement
-	}
-	return nil
-}