Go每日一库之199：go-mongox（简单高效、链式调用的mongodb库）

// 示例代码，不是最佳的创建方式
func newCollection() *mongo.Collection {
  client, err := mongo.Connect(context.Background(), options.Client().ApplyURI("mongodb://localhost:27017").SetAuth(options.Credential{
    Username:   "test",
    Password:   "test",
    AuthSource: "db-test",
  }))
  if err != nil {
    panic(err)
  }
  err = client.Ping(context.Background(), readpref.Primary())
  if err != nil {
    panic(err)
  }
  collection := client.Database("db-test").Collection("test_post")
  return collection
}


type Post struct {
  Id      string `bson:"_id"`
  Title   string `bson:"title"`
        Author  string `bson:"author"`
  Content string `bson:"content"`
}




// 你需要预先创建一个 *mongo.Collection 对象
mongoCollection := newCollection()
// 使用 Post 结构体作为泛型参数创建一个 collection
postCollection := mongox.NewCollection[Post](mongoCollection)

通过 Finder() 方法创建一个查询器，基于该查询器我们可以进行相关的查询操作。设置查询条件的方法是Filter，设置 *options.FindOneOptions 参数的方法有：OneOptions 和 Options。


如果查询条件 filter 非常简单，我们可以通过 bsonx.M 方法进行构造。如果 filter 过于复杂，我们可以使用 query 包里的 BsonBuilder() 构造器进行构造。


Updater
Updater 是一个更新器，用于执行更新相关的操作。

// 删除单个文档
deleteResult, err := postCollection.Deleter().Filter(bsonx.Id("1")).DeleteOne(context.Background())
// 携带 option 参数
deleteResult, err := postCollection.Deleter().Filter(bsonx.Id("1")).Options(options.Delete().SetComment("test")).DeleteOne(context.Background())


// 删除多个文档
// - 通过 query 包构造复杂的 bson 语句
deleteResult, err = postCollection.Deleter().Filter(query.BsonBuilder().InString("_id", []string{"2", "3", "4"}...).Build()).DeleteMany(context.Background())

// 聚合操作
// - 使用 aggregation 包构造 pipeline
posts, err = postCollection.
  Aggregator().Pipeline(aggregation.StageBsonBuilder().Project(bsonx.M("content", 0)).Build()).
  Aggregate(context.Background())




// 如果我们通过聚合操作更改字段的名称，那么我们可以使用 AggregationWithCallback 方法，然后通过 callback 函数将结果映射到我们预期的结构体中
type DiffPost struct {
  Id          string `bson:"_id"`
  Title       string `bson:"title"`
  Name        string `bson:"name"` // author → name
  Content     string `bson:"content"`
  Outstanding bool   `bson:"outstanding"`
}
result := make([]*DiffPost, 0)
//  将 author 字段更名为 name，排除 content 字段，添加 outstanding 字段，返回结果为 []*DiffPost
err = postCollection.Aggregator().
  Pipeline(aggregation.StageBsonBuilder().Project(
    bsonx.NewD().
      Add("name", "$author").
      Add("author", 1).
      Add("_id", 1).
      Add("title", 1).
      Add("outstanding", aggregation.BsonBuilder().Eq("$author", "陈明勇").Build()).Build(),
  ).Build(),
  ).
  AggregateWithCallback(context.Background(), func(ctx context.Context, cursor *mongo.Cursor) error {
    return cursor.All(ctx, &result)
  })