Refactor writher

writer/common.go

@@ -0,0 +1,24 @@
+package writer
+
+import (
+	"reflect"
+	"strings"
+)
+
+func FindIdField(modelType reflect.Type) int {
+	return FindField(modelType, "_id")
+}
+func FindField(modelType reflect.Type, bsonName string) int {
+	numField := modelType.NumField()
+	for i := 0; i < numField; i++ {
+		field := modelType.Field(i)
+		bsonTag := field.Tag.Get("bson")
+		tags := strings.Split(bsonTag, ",")
+		for _, tag := range tags {
+			if strings.TrimSpace(tag) == bsonName {
+				return i
+			}
+		}
+	}
+	return -1
+}

writer/updater.go

@@ -5,8 +5,6 @@ import (
 	"go.mongodb.org/mongo-driver/bson"
 	"go.mongodb.org/mongo-driver/mongo"
 	"reflect"
-
-	mgo "github.com/core-go/mongo"
 )
 
 type Updater[T any] struct {
@@ -28,7 +26,7 @@ func NewUpdaterWithId[T any](database *mongo.Database, collectionName string, op
 		modelType = modelType.Elem()
 		isPointer = true
 	}
-	index, _, _ := mgo.FindIdField(modelType)
+	index := FindIdField(modelType)
 	collection := database.Collection(collectionName)
 	return &Updater[T]{collection: collection, idIndex: index, Map: mp, isPointer: isPointer}
 }

writer/writer.go

@@ -6,8 +6,6 @@ import (
 	"go.mongodb.org/mongo-driver/mongo"
 	"go.mongodb.org/mongo-driver/mongo/options"
 	"reflect"
-
-	mgo "github.com/core-go/mongo"
 )
 
 type Writer[T any] struct {
@@ -29,7 +27,7 @@ func NewWriter[T any](database *mongo.Database, collectionName string, options .
 		modelType = modelType.Elem()
 		isPointer = true
 	}
-	index, _, _ := mgo.FindIdField(modelType)
+	index := FindIdField(modelType)
 	collection := database.Collection(collectionName)
 	return &Writer[T]{collection: collection, idIndex: index, Map: mp, isPointer: isPointer}
 }