自定义字段DefaultField 在使用 UpdateId 等更新集合时，无法自动更新

[colinlabs]

https://github.com/qiniu/qmgo/blob/master/collection.go#L240:L250

这里的代码如果不传递参数是无法执行的，所以，默认情况下，无法自动更新DefaultField下的所有字段。即，没有触发middleware.Do的方法

与自定义InsertOne 等代码逻辑不一样。

https://github.com/qiniu/qmgo/blob/master/collection.go#L150-L155

[ljluestc]

package main

import (
 "context"
 "fmt"
 "log"
 "time"

 "github.com/qiniu/qmgo"
 "github.com/qiniu/qmgo/options"
 "go.mongodb.org/mongo-driver/bson"
 "go.mongodb.org/mongo-driver/bson/primitive"
)

// Document represents a MongoDB document with default fields
type Document struct {
 ID primitive.ObjectID `bson:"_id"`
 Name string `bson:"name"`
 UpdatedAt time.Time `bson:"updated_at"`
}

// CustomCollection wraps qmgo.Collection with a fixed UpdateOne
type CustomCollection struct {
 *qmgo.Collection
}

// UpdateOneFixed ensures middleware is called with default fields
func (c *CustomCollection) UpdateOneFixed(ctx context.Context, filter interface{}, update interface{}, opts ...*options.UpdateOptions) error {
 // Ensure update is a bson.M or convert it
 updateDoc, ok := update.(bson.M)
 if !ok {
 // If update is not bson.M, wrap it in $set
 updateDoc = bson.M{"$set": update}
 }

 // Add default fields if not present
 if _, exists := updateDoc["$set"]; !exists {
 updateDoc["$set"] = bson.M{}
 }
 setFields, _ := updateDoc["$set"].(bson.M)
 if _, exists := setFields["updated_at"]; !exists {
 setFields["updated_at"] = time.Now()
 }

 // Create UpdateOption if none provided
 updateOpt := &qmgo.UpdateOption{}
 for _, opt := range opts {
 if opt != nil {
 updateOpt.Update(opt)
 }
 }

 // Call middleware.Do with the modified update
 return qmgo.GetMiddleware().Do(ctx, c.Collection, filter, updateDoc, updateOpt, qmgo.UpdateOne)
}

func main() {
 ctx := context.Background()

 // Initialize qmgo client
 cfg := &qmgo.Config{Uri: "mongodb://localhost:27017", Database: "test_db", Coll: "test_coll"}
 client, err := qmgo.NewClient(ctx, cfg)
 if err != nil {
 log.Fatalf("Failed to create client: %v", err)
 }
 defer client.Close(ctx)

 // Create a collection
 coll := client.Database(cfg.Database).Collection(cfg.Coll)
 customColl := &CustomCollection{Collection: coll}

 // Insert a test document
 doc := Document{
 ID: primitive.NewObjectID(),
 Name: "Test",
 UpdatedAt: time.Now(),
 }
 _, err = coll.InsertOne(ctx, doc)
 if err != nil {
 log.Fatalf("Failed to insert document: %v", err)
 }
 fmt.Println("Document inserted successfully")

 // Test original UpdateOne (demonstrates issue)
 fmt.Println("\nTesting original UpdateOne...")
 err = coll.UpdateOne(ctx, bson.M{"_id": doc.ID}, bson.M{"$set": bson.M{"name": "Updated"}})
 if err != nil {
 log.Printf("Original UpdateOne failed: %v", err)
 }

 // Retrieve and check document
 var result Document
 err = coll.Find(ctx, bson.M{"_id": doc.ID}).One(&result)
 if err != nil {
 log.Fatalf("Failed to retrieve document: %v", err)
 }
 fmt.Printf("After original UpdateOne: Name=%s, UpdatedAt=%s\n", result.Name, result.UpdatedAt.Format(time.RFC3339))

 // Test fixed UpdateOne
 fmt.Println("\nTesting fixed UpdateOne...")
 err = customColl.UpdateOneFixed(ctx, bson.M{"_id": doc.ID}, bson.M{"$set": bson.M{"name": "Fixed Update"}})
 if err != nil {
 log.Printf("Fixed UpdateOne failed: %v", err)
 }

 // Retrieve and check document again
 err = coll.Find(ctx, bson.M{"_id": doc.ID}).One(&result)
 if err != nil {
 log.Fatalf("Failed to retrieve document: %v", err)
 }
 fmt.Printf("After fixed UpdateOne: Name=%s, UpdatedAt=%s\n", result.Name, result.UpdatedAt.Format(time.RFC3339))
}