Loading [MathJax]/jax/output/CommonHTML/config.js
前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >专栏 >Go 语言 mongox 库:简化操作、安全、高效、可扩展、BSON 构建

Go 语言 mongox 库:简化操作、安全、高效、可扩展、BSON 构建

原创
作者头像
陈明勇
发布于 2024-06-08 13:53:15
发布于 2024-06-08 13:53:15
3150
举报
文章被收录于专栏:Go 技术Go 技术Go技术干货

前言

Go 语言里使用 MongoDB 官方提供的 mongo-go-driver 库进行集合操作时,你是否感到结构体与 MongoDB 集合之间的映射,以及构建 BSON 数据这些操作极其繁琐?特别是在构建 BSON 数据时,各种字段、逗号和括号的排列组合是否让你觉得仿佛在进行一场复杂的拼图游戏?

如果你有同感,那么你并不孤单,我也是。因此我在想,有没有一个能让我丝滑,高效操作 MongoDB 的第三方库呢,遗憾的是,并没有找到符合我预期的库,索性我就自己动手开发了一个,这就是 go mongox 库的由来。

如果你也有类似我的这种感受,相信 go mongox 库能给你带来不一样的体验。

Go Mongox

go mongox 是一个基于泛型的库,扩展了 MongoDB 的官方库。通过泛型技术,它实现了结构体与 MongoDB 集合的绑定,旨在提供类型安全和简化的数据操作。

go mongox 还引入链式调用,让文档操作更流畅,并且提供了丰富的 BSON 构建器和内置函数,简化了 BSON 数据的构建。

此外,它还支持插件化编程和内置多种钩子函数,为数据库操作前后的自定义逻辑提供灵活性,增强了应用的可扩展性和可维护性。

仓库地址:https://github.com/chenmingyong0423/go-mongox

官网文档:https://go-mongox.dev

欢迎体验 go mongox 库,也期待您的贡献。如果您觉得这个库对您有帮助,请给它一个 Star 支持!

功能特性

  • 泛型的 MongoDB 集合
  • 支持 BSON 数据的构建
  • 文档的 CRUD 操作
  • 聚合操作
  • 内置基本的 Model 结构体,自动化更新默认的 field 字段
  • 支持结构体 tag 校验
  • Hooks
  • 支持插件化编程

安装

go get github.com/chenmingyong0423/go-mongox@latest

Collection 集合操作

基于泛型的 Collection 形态初始化

代码语言:go
AI代码解释
复制
package main

import (
	"context"
	"fmt"

	"github.com/chenmingyong0423/go-mongox"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
	"go.mongodb.org/mongo-driver/mongo/readpref"
)

type User struct {
	mongox.Model `bson:"inline"`
	Name         string `bson:"name"`
	Age          int    `bson:"age"`
}

func main() {
	// 你需要预先创建一个 *mongo.Collection 对象
	mongoColl := newCollection()
	// 使用 User 结构体作为泛型参数创建一个 collection
	userColl := mongox.NewCollection[User](mongoColl)
	fmt.Println(userColl != nil)
}

// 示例代码,不是最佳的创建方式
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
}

通过 mongox.NewCollection 函数,我们可以指定泛型参数并创建一个泛型的 Collection 对象。这样我们就可以使用 userColl 对象来操作 User 类型的文档了。

后面的操作将基于 userColl 对象进行举例。

Creator 创造器

通过 Creator() 方法获取一个新的泛型的创建器对象,即 Creator[T],通过 Creator[T] 的方法,我们能够执行相关的创建操作。

示例

代码语言:go
AI代码解释
复制
// 插入单个文档
insertOneResult, err := userColl.Creator().InsertOne(context.Background(), &User{Name: "陈明勇", Age: 18})

users := []*User{
	{Name: "Mingyong Chen", Age: 18},
	{Name: "cmy", Age: 18},
}
// 插入多个文档
insertManyResult, err := userColl.Creator().InsertMany(context.Background(), users)

InsertOneInsertMany 方法与官方的 API 同名,前者用于插入单个文档,后者用于插入多个文档。如果我们需要设置 options 参数,可以将其作为这两个方法的第三个参数传递。

更多用法可前往官网文档 Creator 创造器 | go mongox 进行查看。

Finder 查询器

通过 Finder() 方法获取一个新的泛型的查询器对象,即 Finder[T]。通过 Finder[T] 的方法,我们能够执行相关的查询操作。

示例

代码语言:go
AI代码解释
复制
// 查询单个文档
post, err := userColl.Finder().Filter(query.Id(primitive.NewObjectID())).FindOne(context.Background())

// - 复杂条件查询
// -- 使用 query 包构建复杂的 bson
// -- bson.D{bson.E{Key:"name", Value:bson.D{bson.E{Key:"$eq", Value:"陈明勇"}}}, bson.E{Key:"age", Value:bson.D{bson.E{Key:"$eq", Value:18}}}}
post, err = userColl.Finder().
	Filter(query.NewBuilder().Eq("name", "陈明勇").Eq("age", 18).Build()).
	FindOne(context.Background())

// 查询多个文档
posts, err := userColl.Finder().
	Filter(query.In("name", []string{"陈明勇", "Mingyong Chen"}...)).
	Find(context.Background())

FindOneFind 方法与官方的 API 同名,前者用于查询单个文档,后者用于查询多个文档。我们可以通过 Filter 方法设置 查询条件,如果我们需要设置 options 参数,可以将其作为这两个方法的第二个参数传递。

对于简单的查询条件,我们可以使用 query 包提供的函数进行构建,例如 query.(id);对于复杂的查询条件,我们可以使用 query 包提供的 Builder 构建器进行构建。query 包的用法会在下面的内容进行详细地介绍。

更多用法可前往官网文档 Finder 查询器 | go mongox 进行查看。

Updater 更新器

通过 Updater() 方法获取一个新的泛型的更新器对象,即 Updater[T]。通过 Updater[T] 的方法,我们能够执行相关的更新操作。

示例

代码语言:go
AI代码解释
复制
// 更新单个文档
// 通过 update 包构建 bson 更新语句
updateResult, err := userColl.Updater().
	Filter(query.Id(primitive.NewObjectID())).
	Updates(update.Set("age", 19)).
	UpdateOne(context.Background())

// 更新多个文档
updateResult, err = userColl.Updater().
	Filter(query.In("name", []string{"陈明勇", "Mingyong Chen"}...)).
	Updates(update.Set("age", "19")).
	UpdateMany(context.Background())

UpdateOneUpdateMany 方法与官方的 API 同名,前者用于更新单个文档,后者用于更新多个文档。我们可以通过 Filter 方法设置 文档匹配的条件,如果我们需要设置 options 参数,可以将其作为这两个方法的第二个参数传递。

对于更新参数,我们可以使用 Updates 方法进行设置。该方法接收 bsonmap 等合法的更新操作语句。上面的例子使用了 update 包里的 Set 对更新操作语句进行构建。

更多用法可前往官网文档 Updater 更新器 | go mongox 进行查看。

Deleter 删除器

通过 Deleter() 方法获取一个新的泛型的删除器对象,即 Deleter[T]。通过 Deleter[T] 的方法,我们能够执行相关的删除操作。

示例

代码语言:go
AI代码解释
复制
// 删除单个文档
deleteResult, err := userColl.Deleter().Filter(query.Id(primitive.NewObjectID())).DeleteOne(context.Background())

// 删除多个文档
deleteResult, err = userColl.Deleter().
	Filter(query.In("name", "Mingyong Chen", "cmy")).
	DeleteMany(context.Background())

DeleteOneDeleteMany 方法与官方的 API 同名,前者用于删除单个文档,后者用于删除多个文档。我们可以通过 Filter 方法设置 文档匹配的条件。如果我们需要设置 options 参数,可以将其作为这两个方法的第二个参数传递。

更多用法可前往官网文档 Deleter 删除器 | go mongox 进行查看。

Aggregator 聚合器

通过 Aggregator() 获取一个新的泛型的聚合器对象,即 Aggregator[T any],通过 Aggregator[T any] 的方法,我们能够执行相关的聚合操作。

聚合器实现了三个方法:

  • Pipeline() 用于设置聚合管道。
  • Aggregate() 用于执行聚合操作,返回的查询结果类型与 T 一致。
  • AggregateWithParse() 也是用于执行聚合操作,但使用场景不一样。当聚合结果的类型与 T 不一致时,使用 AggregateWithParse() 方法可以将结果解析到指定的对象里。
代码语言:go
AI代码解释
复制
// 忽略年龄字段,只查询名字
users, err := userColl.Aggregator().
	Pipeline(aggregation.NewStageBuilder().Project(bsonx.M("age", 0)).Build()).
	Aggregate(context.Background())


// 字段重命名,聚合查询并解析结果
type RealUser struct {
	mongox.Model `bson:"inline"`
	RealName     string `bson:"real_name"`
	Age          int    `bson:"age"`
}
var results []*RealUser
err = userColl.Aggregator().
	Pipeline(aggregation.NewStageBuilder().Project(
		bsonx.NewD().Add("real_name", "$name").Add("age", 1).Build(),
	).Build()).
	AggregateWithParse(context.Background(), &results)

更多用法可前往官网文档 Aggregator 聚合器 | go mongox 进行查看。

BSON 数据构建

Go mongox 设计了多种不同类型的 BSON 构建器和函数,为我们开发者在不同场景下构建 BSON 数据提供了强大的支持。无论是数据查询、更新,还是执行复杂的聚合操作,开发者都可以在 bsonxqueryupdate 以及 aggregation 专门的包中找到合适的构建器或函数。这些 BSON 构建器和函数不仅优化了代码的编写过程,还显著提高了开发效率,使得处理复杂的 BSON 数据变得既简单又高效。

bsonx 包

bsonx 提供提供了一系列便捷的函数和构建器去构建 BSON 数据,旨在简化 BSON 数据的构建过程。

代码语言:go
AI代码解释
复制
d := bsonx.NewD().Add("name", "陈明勇").Add("age", "18").Build()
m := bsonx.M("name", "陈明勇")
e := bsonx.E("name", "陈明勇")
d = bsonx.D("name", "陈明勇")
a := bsonx.A("Mingyong Chen", "陈明勇")

更多用法可前往官网文档 bsonx 包 | go mongox 进行查看。

query 查询构建

query 包为构建 MongoDB 查询条件提供了便捷方法。它包括一系列函数和构建器,旨在简化查询条件的构建。对于复杂查询条件,构建器通过链式调用方式,提供了灵活的构建手段;对于简单需求,则可直接利用函数来快速构建查询条件。

代码语言:go
AI代码解释
复制
// bson.D{bson.E{Key:"_id", Value:"12345678"}}
/*
   {
     "_id": "12345678"
   }
*/
query.Id("12345678")

// bson.D{bson.E{Key:"name", Value:bson.D{bson.E{Key:"$regex", Value:".*cmy.*"}, bson.E{Key:"$options", Value:"i"}}}}
/*
   {
     "name": {
       "$regex": ".*cmy.*",
       "$options": "i"
     }
   }
*/
query.RegexOptions("name", ".*cmy.*", "i")

// bson.D{bson.E{Key:"age", Value:bson.D{bson.E{Key:"$gte", Value:18}, bson.E{Key:"$lte", Value:25}}}, bson.E{Key:"name", Value:bson.D{bson.E{Key:"$in", Value:[]interface {}{"陈明勇", "chenmingyong"}}}}}
/*
   {
     "age": {
       "$gte": 18,
       "$lte": 25
     },
     "name": {
       "$in": ["陈明勇", "Mingyong Chen"]
     }
   }
*/
query.NewBuilder().Gte("age", 18).Lte("age", 25).In("name", "陈明勇", "Mingyong Chen").Build()

// bson.D{bson.E{Key:"lastLogin", Value:bson.D{bson.E{Key:"$gte", Value:time.Date(2024, time.March, 0, 0, 0, 0, 189625000, time.Local)}}}, bson.E{Key:"$or", Value:[]interface {}{bson.D{bson.E{Key:"status", Value:bson.D{bson.E{Key:"$eq", Value:"active"}}}}, bson.D{bson.E{Key:"loginAttempts", Value:bson.D{bson.E{Key:"$gte", Value:5}}}}}}}
/*
   {
     "lastLogin": {
       "$gte": "2024-03-08T00:00:00.189Z"
     },
     "$or": [
       {
         "status": {
           "$eq": "active"
         }
       },
       {
         "loginAttempts": {
           "$gte": 5
         }
       }
     ]
   }
*/
query.NewBuilder().Gte("lastLogin", time.Now().Add(-30*24*time.Hour)).Or(
    query.Eq("status", "active"),
    query.Gte("loginAttempts", 5),
).Build()

// bson.D{bson.E{Key:"name", Value:bson.D{bson.E{Key:"$eq", Value:"陈明勇"}}}, bson.E{Key:"hobbies", Value:bson.D{bson.E{Key:"$elemMatch", Value:primie.D{bson.E{Key:"name", Value:bson.D{bson.E{Key:"$eq", Value:"coding"}}}, bson.E{Key:"level", Value:bson.D{bson.E{Key:"$gte", Value:5}}}}}}}}
/*
   {
     "name": {
       "$eq": "陈明勇"
     },
     "hobbies": {
       "$elemMatch": {
         "name": {
           "$eq": "coding"
         },
         "level": {
           "$gte": 5
         }
       }
     }
   }
*/
query.NewBuilder().
    Eq("name", "陈明勇").
    ElemMatch("hobbies", query.NewBuilder().Eq("name", "coding").Gte("level", 5).Build()).
    Build()

更多用法可前往官网文档 query 包 | go mongox 进行查看。

update 更新构建器

update 包为构建 MongoDB 更新文档提供了便捷方法。它包括一系列函数和构建器,旨在简化更新文档的构建。对于复杂的更新文档,构建器通过链式调用方式,提供了灵活的构建手段;对于简单需求,则可直接利用函数来快速构建更新文档。

代码语言:go
AI代码解释
复制
// bson.D{bson.E{Key:"$set", Value:bson.D{bson.E{Key:"name", Value:"陈明勇"}}}}
/*
   {
	 "$set": {
	   "name": "陈明勇"
	 }
   }
*/
update.Set("name", "陈明勇")

// bson.D{bson.E{Key:"$inc", Value:bson.D{bson.E{Key:"money", Value:"100000"}}}}
/*
   {
	 "$inc": {
	   "money": 100000
	 }
   }
*/
update.Inc("money", "100000")

// bson.D{bson.E{Key:"$push", Value:bson.D{bson.E{Key:"tags", Value:"golang"}}}}
/*
   {
	 "$push": {
	   "tags": "golang"
	 }
   }
*/
update.Push("tags", "golang")

// bson.D{bson.E{Key:"$set", Value:bson.D{bson.E{Key:"name", Value:"陈明勇"}, bson.E{Key:"age", Value:18}}}}
/*{
	"$set": {
	  "name": "陈明勇",
	  "age": 18
	}
  }
*/
update.NewBuilder().Set("name", "陈明勇").Set("age", 18).Build()

// bson.D{bson.E{Key:"$set", Value:bson.D{bson.E{Key:"update_at", Value:time.Date(2024, time.April, 7, 3, 13, 23, 958924000, time.Local)}}}, bson.E{Key:"$inc", Value:bson.D{bson.E{Key:"view", Value:1}}}}
/*{
	"$set": {
	  "update_at": "2024-04-07T00:00:00.958Z"
	},
	"$inc": {
	  "view": 1
	}
  }
*/
update.NewBuilder().Set("update_at", time.Now()).Inc("view", 1).Build()

// bson.D{bson.E{Key:"$push", Value:bson.D{bson.E{Key:"comments", Value:"新评论"}}}, bson.E{Key:"$inc", Value:bson.D{bson.E{Key:"commentCount", Value:}}}
/*
   {
	 "$push": {
	   "comments": "新评论"
	 },
	 "$inc": {
	   "commentCount": 1
	 }
   }
*/
update.NewBuilder().Push("comments", "新评论").Inc("commentCount", 11).Build()

更多用法可前往官网文档 Update 包 | go mongox 进行查看。

aggregation 聚合构建器

aggregation 包提供了方便的方法来构建MongoDB聚合管道(pipeline)结构。它包括多个函数和构建器,简化了管道构建过程。对于复杂场景,构建器支持链式调用,使得构建过程更加灵活;而对于基础需求,可以直接使用函数快速完成构建。

aggregation 包提供了两种构建器:

  • aggregation.StageBuilder:用于轻松构建聚合管道的各个阶段(Pipeline Stages),如$group$match等。通过 aggregation.StageBsonBuilder() 创建一个新的构建器实例,然后调用相应的方法来构建阶段。
  • aggregation.Builder:用于构建管道阶段内部使用的复杂表达式(Pipeline Expressions),例如条件逻辑、数学运算等。通过 aggregation.BsonBuilder() 创建一个新的构建器实例,然后调用相应的方法来构建表达式。

聚合管道阶段

聚合阶段构建器用于轻松构建聚合管道的各个阶段(Pipeline Stages),如 $group$match 等。

通过 aggregation.NewStageBuilder() 创建一个新的构建器实例,然后调用相应的方法来构建阶段。

代码语言:go
AI代码解释
复制
// mongo.Pipeline{bson.D{bson.E{Key:"$group", Value:bson.D{bson.E{Key:"_id", Value:"$age"}, bson.E{Key:"count", Value:bson.D{bson.E{Key:"$sum", Value:1}}}, bson.E{Key:"names", Value:bson.D{bson.E{Key:"$push", Value:"$name"}}}}}}}
/*[
	{
	  "$group": {
		"_id": "$age",
		"count": { "$sum": 1 },
		"names": { "$push": "$name" }
	  }
	}
  ]
*/
aggregation.NewStageBuilder().Group("$age",
	aggregation.NewBuilder().Sum("count", 1).Push("names", "$name").Build()...,
).Build()

// mongo.Pipeline{bson.D{bson.E{Key:"$addFields", Value:bson.D{bson.E{Key:"isAdult", Value:bson.D{bson.E{Key:"$gte", Value:[]interface {}{"$age", 18}}}}}}}, bson.D{bson.E{Key:"$replaceWith", Value:bson.D{bson.E{Key:"name", Value:"$name"}, bson.E{Key:"isAdult", Value:"$isAdult"}}}}}
/*
   [
	 {
	   "$addFields": {
		 "isAdult": {
		   "$gte": ["$age", 18]
		 }
	   }
	 },
	 {
	   "$replaceWith": {
		 "name": "$name",
		 "isAdult": "$isAdult"
	   }
	 }
   ]
*/
aggregation.NewStageBuilder().
	AddFields(aggregation.Gte("isAdult", "$age", 18)).
	ReplaceWith(bsonx.NewD().Add("name", "$name").Add("isAdult", "$isAdult").Build()).Build()

// mongo.Pipeline{bson.D{bson.E{Key:"$bucket", Value:bson.D{bson.E{Key:"groupBy", Value:"$age"}, bson.E{Key:"boundaries", Value:[]interface {}{0, 19, 31, 46, +Inf}}, bson.E{Key:"default", Value:"Other"}, bson.E{Key:"output", Value:bson.D{bson.E{Key:"count", Value:bson.D{bson.E{Key:"$sum", Value:1}}}, bson.E{Key:"names", Value:bson.D{bson.E{Key:"$push", Value:"$name"}}}}}}}}}
/*
   [
	 {
	   $bucket: {
		 groupBy: "$age",  // 指定分组的依据字段
		 boundaries: [0, 19, 31, 46, Infinity],  // 定义年龄分组的边界
		 default: "Other",  // 对于不满足任何边界条件的文档,将其分配到一个默认的桶
		 output: {
		   "count": { $sum: 1 },  // 计算每个桶中的文档数
		   "names": { $push: "$name" }  // 收集每个桶中所有用户的名字
		 }
	   }
	 }
   ]
*/
aggregation.NewStageBuilder().Bucket(
	"$age",
	[]any{0, 19, 31, 46, math.Inf(1)},
	&aggregation.BucketOptions{
		DefaultKey: "Other",
		Output:     aggregation.NewBuilder().Sum("count", 1).Push("names", "$name").Build(),
	},
).Build()

聚合表达式

聚合表达式构建器用于轻松构建聚合管道的各个表达式(Expressions),如 $add, $subtract 等。

通过 aggregation.BsonBuilder() 创建一个新的构建器实例,然后调用相应的方法来构建表达式。

代码语言:go
AI代码解释
复制
// bson.D{bson.E{Key:"isAdult", Value:bson.D{bson.E{Key:"$gte", Value:[]interface {}{"$age", 18}}}}}
/*
   {
	 "isAdult": {
	   "$gte": ["$age", 18]
	 }
   }
*/
aggregation.Gte("isAdult", "$age", 18)

// bson.D{bson.E{Key:"birthYear", Value:bson.D{bson.E{Key:"$subtract", Value:[]interface {}{2024, "$age"}}}}}
/*
   {
	 "birthYear": {
	   "$subtract": [2024, "$age"]
	 }
   }
*/
aggregation.Subtract("birthYear", 2024, "$age")

// bson.D{bson.E{Key:"age", Value:bson.D{bson.E{Key:"$gt", Value:[]interface {}{18}}}}}
/*
   {
	 "age": {
	   "$gt": 18
	 }
   }
*/
aggregation.Gt("age", 18)

// bson.D{bson.E{Key:"isAdult", Value:bson.D{bson.E{Key:"$gte", Value:[]interface {}{"$age", 18}}}}, bson.E{Key:"birthYear", Value:bson.D{bson.E{Key:"$subtract", Value:[]interface {}{2024, "$age"}}}}}
/*
   {
	 "isAdult": {
	   "$gte": ["$age", 18]
	 },
	 "birthYear": {
	   "$subtract": [2024, "$age"]
	 }
   }
*/
aggregation.NewBuilder().
	Gte("isAdult", "$age", 18).
	Subtract("birthYear", 2024, "$age").Build()

// bson.D{bson.E{Key:"count", Value:bson.D{bson.E{Key:"$sum", Value:1}}}, bson.E{Key:"names", Value:bson.D{bson.E{Key:"$push", Value:"$name"}}}}
/*
   {
	 "count": {
	   "$sum": 1
	 },
	 "names": {
	   "$push": "$name"
	 }
   }
*/
aggregation.NewBuilder().Sum("count", 1).Push("names", "$name").Build()

// bson.D{bson.E{Key:"count", Value:bson.D{bson.E{Key:"$sum", Value:1}}}, bson.E{Key:"averageAge", Value:bson.D{bson.E{Key:"$avg", Value:"$age"}}}, bson.E{Key:"names", Value:bson.D{bson.E{Key:"$push", Value:"$name"}}}}
/*{
	"count": {
	  "$sum": 1
	},
	"averageAge": {
	  "$avg": "$age"
	},
	"names": {
	  "$push": "$name"
	}
  }
*/
aggregation.NewBuilder().
	Sum("count", 1).
	Avg("averageAge", "$age").
	Push("names", "$name").Build()

更多用法可前往官网文档 Aggregation 包 | go mongox 进行查看。

插件化编程

go mongox 支持插件化编程,它提供了一种灵活的方式在数据库操作的前后插入自定义的逻辑,从而增强应用的可扩展性和可维护性。

go mongox 提供了 RegisterPluginUnregisterPlugin 方法来注册和删除插件。

代码语言:go
AI代码解释
复制
type User struct {
	mongox.Model `bson:"inline"`
	Name         string `bson:"name"`
	Age          int    `bson:"age"`
}

// 注册插件
mongox.RegisterPlugin("after find", func(ctx context.Context, opCtx *operation.OpContext, opts ...any) error {
	if user, ok := opCtx.Doc.(*User); ok {
		fmt.Println(user)
	}
	if users, ok := opCtx.Doc.([]*User); ok {
		fmt.Println(users)
	}
	return nil
}, operation.OpTypeAfterFind)

// 移除插件
mongox.RemovePlugin("after find", operation.OpTypeAfterFind)

激活内置的插件(钩子)

go mongox 库内置了三个实用的 hook 钩子:

  • field 钩子:自动化更新默认的 field 字段
  • model 钩子:针对模型(结构体)设置钩子函数,这些钩子函数会在 MongoDB 的集合操作前后被调用。
  • validator 钩子:利用结构体的标签(tag)去对字段值进行校验。

go mongox 库默认不激活这些钩子,如果你想激活它们,可以参考以下代码:

代码语言:go
AI代码解释
复制
mongox.InitPlugin(&mongox.PluginConfig{
	EnableDefaultFieldHook: true,
	EnableModelHook:        true,
	EnableValidationHook:   true,
	// 覆盖默认的校验器,当 EnableValidationHook 为 true 时生效
	Validate: nil,
})

这三个内置钩子的内容会在下面进行介绍。

更多用法可前往官网文档 插件化编程 | go mongox 进行查看。

结构体 tag 校验

go mongox 库支持利用结构体的标签(tag)去对字段值进行校验,这一校验功能基于 playground/validator 库提供的所有结构体校验规则。

代码语言:go
AI代码解释
复制
type User struct {
	mongox.Model   `bson:"inline"`
	Name           string `bson:"name"`
	Age            uint8  `validate:"gte=0,lte=130"`                // 确保年龄在 0 到 130 岁之间
	Email          string `json:"e-mail" validate:"required,email"` // 表示这个字段在数据验证时是必需的,并且必须符合电子邮箱的格式。
	FavouriteColor string `validate:"hexcolor|rgb|rgba"`            // 确保提供的颜色值要么是十六进制颜色码,要么是RGB或RGBA格式。
}

结构体 tag 校验功能默认是关闭的,如需开启,请使用 mongox.InitPlugin 函数。

更多用法可前往官网文档 结构体校验 | go mongox 进行查看。

内置 Model

go mongox 内置了一个 Model 结构体,它包含了 IDCreatedAtUpdatedAt 三个字段。

代码语言:go
AI代码解释
复制
type Model struct {
	ID        primitive.ObjectID `bson:"_id,omitempty"`
	CreatedAt time.Time          `bson:"created_at"`
	UpdatedAt time.Time          `bson:"updated_at"`
}

func (m *Model) DefaultId() {
	if m.ID.IsZero() {
		m.ID = primitive.NewObjectID()
	}
}

func (m *Model) DefaultCreatedAt() {
	if m.CreatedAt.IsZero() {
		m.CreatedAt = time.Now().Local()
	}
}

func (m *Model) DefaultUpdatedAt() {
	m.UpdatedAt = time.Now().Local()
}

这个结构体实现了 DefaultModelHook 接口,如果初始化插件时将 EnableDefaultFieldHook 设置为 true(详情请参考 启用内置插件-钩子 ),go mongox 库将自动化地处理文档的创建、更新操作中的 ID 和时间的赋值。

更多用法可前往官网文档 内置 Model | go mongox 进行查看。

Hook 钩子

模型钩子(Model Hooks)

你可以针对模型(结构体)设置钩子函数,这些钩子函数会在 MongoDB 的集合操作前后被调用。例如,你可以在插入文档前后对文档进行处理,或者在查询文档后对文档进行处理。

代码语言:go
AI代码解释
复制
type User struct {
	mongox.Model `bson:"inline"`
	Name         string `bson:"name"`
	Age          int    `bson:"age"`
}

func (u *User) BeforeInsert(ctx context.Context) error {
	fmt.Println("BeforeInsert called")
	return nil
}

func (u *User) AfterInsert(ctx context.Context) error {
	fmt.Println("AfterInsert called")
	return nil
}

func (u *User) AfterFind(ctx context.Context) error {
	fmt.Println("AfterFind called")
	return nil
}

func (u *User) BeforeUpsert(ctx context.Context) error {
	fmt.Println("BeforeUpsert called")
	return nil
}

func (u *User) AfterUpsert(ctx context.Context) error {
	fmt.Println("AfterUpsert called")
	return nil
}

更多用法可前往官网文档 模型钩子(Model Hooks) | go mongox 进行查看。

一次性钩子

go mongox 支持一次性钩子,你可以在查询、插入、删除、更新和保存文档的前后执行一些操作。

代码语言:go
AI代码解释
复制
// 插入操作
_, err = userColl.Creator().
	RegisterBeforeHooks(func(ctx context.Context, opContext *creator.OpContext[User], opts ...any) error {
		fmt.Println("BeforeHook called")
		fmt.Println(opContext.Doc)
		fmt.Println(opContext.Col != nil)
		return nil
	}).
	RegisterAfterHooks(func(ctx context.Context, opContext *creator.OpContext[User], opts ...any) error {
		fmt.Println("AfterHook called")
		fmt.Println(opContext.Doc)
		fmt.Println(opContext.Col != nil)
		return nil
	}).
	InsertOne(context.Background(), &User{Name: "Mingyong Chen", Age: 18})

// 查询操作
_, err = userColl.Finder().
	RegisterBeforeHooks(func(ctx context.Context, opContext *finder.OpContext, opts ...any) error {
		fmt.Println("BeforeHook called")
		fmt.Println(opContext.Filter)
		return nil
	}).
	RegisterAfterHooks(func(ctx context.Context, opContext *finder.AfterOpContext[User], opts ...any) error {
		fmt.Println("AfterHook called")
		fmt.Println(opContext.Filter)
		fmt.Println(opContext.Doc)
		return nil
	}).
	Filter(query.Eq("name", "陈明勇")).FindOne(context.Background())

// 更新操作
_, err = userColl.Updater().
	RegisterBeforeHooks(func(ctx context.Context, opContext *updater.BeforeOpContext, opts ...any) error {
		fmt.Println("BeforeHook called")
		fmt.Println(opContext.Filter)
		fmt.Println(opContext.Updates)
		fmt.Println(opContext.Col != nil)
		return nil
	}).
	RegisterAfterHooks(func(ctx context.Context, opContext *updater.AfterOpContext, opts ...any) error {
		fmt.Println("AfterHook called")
		fmt.Println(opContext.Filter)
		fmt.Println(opContext.Updates)
		fmt.Println(opContext.Col != nil)
		return nil
	}).
	Filter(query.Eq("name", "陈明勇")).
	Updates(update.Set("age", 19)).
	UpdateOne(context.Background())

// 删除操作
_, err = userColl.Deleter().
	RegisterBeforeHooks(func(ctx context.Context, opContext *deleter.BeforeOpContext, opts ...any) error {
		fmt.Println("BeforeHook called")
		fmt.Println(opContext.Filter)
		fmt.Println(opContext.Col != nil)
		return nil
	}).
	RegisterAfterHooks(func(ctx context.Context, opContext *deleter.AfterOpContext, opts ...any) error {
		fmt.Println("AfterHook called")
		fmt.Println(opContext.Filter)
		fmt.Println(opContext.Col != nil)
		return nil
	}).
	Filter(query.Eq("name", "Mingyong Chen")).
	DeleteOne(context.Background())

更多用法可前往官网文档 一次性钩子 | go mongox 进行查看。

小结

本文详细介绍了 go mongox 库的关键模块,包括创建指定约束类型的泛型 Collection、灵活的 BSON 构建器、基础的 CRUD 操作、聚合操作、以及插件和钩子机制,并提供了相应的使用示例。

仓库地址:https://github.com/chenmingyong0423/go-mongox

官网文档:https://go-mongox.dev

欢迎体验 go mongox 库,也期待您的贡献。如果您觉得这个库对您有帮助,请给它一个 Star 支持!

我正在参与2024腾讯技术创作特训营最新征文,快来和我瓜分大奖!

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

评论
登录后参与评论
暂无评论
推荐阅读
编辑精选文章
换一批
JAVA Class类与反射
Class对象是Class类的实例,类至少包含以下信息,因此class类又可以被 解构为如下部分:
全栈程序员站长
2022/08/30
4420
JAVA Class类与反射
深入理解Java中的反射机制和使用原理!详细解析invoke方法的执行和使用
newInstance方法可以根据传入的参数来调用对应的Constructor创建对象的实例
攻城狮Chova
2022/02/24
2.1K0
深入理解Java中的反射机制和使用原理!详细解析invoke方法的执行和使用
Java反射(完)类加载和反射获取信息
1.目的是为了确保 Class 文件的字节流中包含的信息符合当前虚拟机的要求,并且不会危害虚拟机自身的安全。2.包括:文件格式验证(是否以魔数 oxcafebabe 开头)、元数据验证、字节码验证和符号引用验证 3.可以考虑使用-Xverify:none 参数来关闭大部分的类验证措施,缩短虚拟机类加载的时间。
浅辄
2022/12/07
5730
Java反射(完)类加载和反射获取信息
Java基础(二十三):反射机制
(1)class:外部类,成员(成员内部类,静态内部类),局部内部类,匿名内部类 (2)interface:接口 (3)[]:数组 (4)enum:枚举在这里插入代码片 (5)annotation:注解@interface (6)primitive type:基本数据类型 (7)void
Java微观世界
2025/01/21
1150
Java基础(二十三):反射机制
ReflectionUtils反射工具:精要介绍与实战应用指南
org.springframework.util.ReflectionUtils 是 Spring 框架提供的一个反射工具类,它封装了 Java 反射 API 的一些常用操作,使得我们能够更加方便、简洁地使用反射功能。。
公众号:码到三十五
2024/04/02
5050
Java中的反射(通过反射获取类的结构、invoke方法、获取注解)
1)获取该类型的Class对象 2)调用Class对象的newInstance()方法创建对象
鱼找水需要时间
2023/05/09
4.8K0
Java中的反射(通过反射获取类的结构、invoke方法、获取注解)
Android反射机制简单理解,ReflectionUtils  反射工具类
JAVA反射机制是在运行状态中,对于任意一个类,都能够知道这个类的所有属性和方法;对于任意一个对象,都能够调用它的任意一个方法和属性;这种动态获取的信息以及动态调用对象的方法的功能称为java语言的反射机制(注意关键词:运行状态)换句话说,Java程序可以加载一个运行时才得知名称的class,获悉其完整构造(但不包括methods定义),并生成其对象实体、或对其fields设值、或唤起其methods
zhangjiqun
2024/12/16
2020
Java反射详解
这样的需求在学习框架时特别多,即通过外部文件配置,在不修改源码情况下。来控制程序,也符合设计模式的ocp原则(开闭原则:不修改源码,扩容功能)。
timerring
2023/05/24
3390
Java反射详解
反射入门_入门教程
类(外部类,内部类,匿名内部类等等),接口,数组,enum,注解,基本数据类型 ,void都算Class对象
全栈程序员站长
2022/09/21
2600
Java反射基础
Java反射(Reflection)机制就是在运行状态中,对于任意一个类,都能够知道这个类的属性和方法。对于任意一个对象能够调用它的任意一个属性和方法。这种动态获取的信息和动态调用对象的方法的功能称为Java语言的反射机制。Java程序中一般的对象的类型都是在编译期就确定下来的,而Java反射机制可以动态地创建对象并调用其属性,这样的对象的类型在编译期是未知的。所以我们可以通过反射机制直接创建对象,即使这个对象的类型在编译期是未知的。
Abalone
2022/07/14
3010
【Java基础】JavaCore核心-反射技术
互联网小阿祥
2023/05/28
3230
【Java基础】JavaCore核心-反射技术
Java反射机制
        JAVA反射机制是在运行状态中,对于任意一个实体类,都能够知道这个类的所有属性和方法;对于任意一个对象,都能够调用它的任意方法和属性;这种动态获取信息以及动态调用对象方法的功能称为java语言的反射机制。
sunonzj
2022/06/21
6090
Java反射机制
java中的反射
  –对象照镜子后可以得到的信息:某个类的数据成员名、方法和构造器、某个类到底实现了哪些接口。对于每个类而言,JRE 都为其保留一个不变的 Class 类型的对象。一个 Class 对象包含了特定某个类的有关信息。
全栈程序员站长
2022/09/06
1.8K0
java中的反射
玩转 ReflectionUtils 工具类,离大佬又近一步
ReflectionUtils是Spring框架中的反射工具类,它提供了一系列静态方法,可以方便地进行类、对象、方法、字段等反射操作。
终码一生
2023/08/22
1.9K0
玩转 ReflectionUtils 工具类,离大佬又近一步
Java ”框架 = 注解 + 反射 + 设计模式“ 之 反射详解
Reflection (反射)是被视为 **动态语言的关键** ,反射机制允许程序在执行期借助 **Reflection API** 取得任何类的内部信息,并能直接操作任意对象内部属性以及方法。
RainbowSea
2023/02/23
6890
从Class源码看反射
日常敲码中,如果想要在程序运行阶段访问某个类的所有信息,并支持修改类的状态或者行为的话,肯定会用到反射,而反射靠的就是Class类。Java的动态代理也用到了这个东西,所以了解其基本操作在苦逼的CRUD中会添加一丝丝乐趣(有点意思)。
Liusy
2020/09/01
8230
从Class源码看反射
java-reflection
以上方法返回值的类型是一个 Class 类,此类是Java反射的源头,实际上所谓反射从程序的运行结果来看也很好理解,即可以通过对象反射求出类的名称。
Remember_Ray
2020/09/15
4570
java-reflection
TIII-Android技术篇之反射
对于一个类来说,最重要的是什么? 构造+属性+方法+访问权限 待测试的类 /** * 作者:张风捷特烈 * 时间:2018/5/6:10:40 * 邮箱:1981462002@qq.com
张风捷特烈
2018/09/29
4730
JDK源码分析 反射
对于JDK源码分析的文章,仅仅记录我认为重要的地方。源码的细节实在太多,不可能面面俱到地写清每个逻辑。所以我的JDK源码分析,着重在JDK的体系架构层面,具体源码可以参考:http://www.cnblogs.com/skywang12345/category/455711.html。
Yano_nankai
2018/10/08
4410
大数据必学Java基础(九十):通过反射获取运行时类的完整结构
问题1:创建Person的对象,以后用new Person()创建,还是用反射创建?
Lansonli
2022/12/13
5190
大数据必学Java基础(九十):通过反射获取运行时类的完整结构
推荐阅读
相关推荐
JAVA Class类与反射
更多 >
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档