Go语言学习笔记之golang操作MongoDB数据库

更新时间：2023年05月25日 11:48:41 作者：PPPsych

MongoDB是Nosql中常用的一种数据库,这篇文章主要给大家介绍了关于Go语言学习笔记之golang操作MongoDB数据库的相关资料,文中通过实例代码介绍的非常详细,需要的朋友可以参考下

一下载安装驱动

官方文档

https://pkg.go.dev/go.mongodb.org/mongo-driver/mongo

下载地址

https://www.mongodb.com/try/download/community

打开客户端

mongo.exe

创建数据库

use golang_db;

创建集合

db.createCollection("student");

下载驱动

go get go.mongodb.org/mongo-driver/mongo

二 go连接到MongoDB

package main

import (
	"context"
	"fmt"
	"log"

	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
)

var client *mongo.Client

func initDB() {
	// 设置客户端连接配置
	clientOptions := options.Client().ApplyURI("mongodb://localhost:27017")

	// 连接到MongoDB
	var err error
	c, err := mongo.Connect(context.TODO(), clientOptions)
	if err != nil {
		log.Fatal(err)
	}

	// 检查连接
	err = c.Ping(context.TODO(), nil)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println("MongoDB 连接成功")
	client = c
}

func main() {
	initDB()
}

运行结果：

[Running] go run "e:\golang开发学习\go_pro\test.go"
MongoDB 连接成功

[Done] exited with code=0 in 2.819 seconds

三 BSON简介

MongoDB中的JSON文档存储在名为BSON（二进制编码的JSON）的二进制表中。与其他将JSON数据存储为简单字符串和数字的数据库不同，BSON编码扩展了JSON表示，使其包含额外的类型，如int、long、date、浮点数和decimal128。这使得应用程序更容易可靠地处理、排序和比较数据。

连接MongoDB地Go驱动程序中有两大类型表示BSON数据：D和Raw。

类型D家族被用来简洁地构建使用本地Go类型地BSON对象。这对于构造传递给MongoDB地命令特别有用。D家族包括四大类：

D：一个BSON文档。这种类型应该在顺序重要的情况下使用，比如MongoDB命令。
M：一张无序的map。它和D是一样的，只是它不保持顺序。
A：一个BSON数组。
E：D里面的一个元素。

要使用BSON，需要先导入下面的包：

go.mongodb.org/mongo-driver/bson

下面是一个使用D类型构建地过滤器文档的例子，它可以用来查找name字段与‘张三’或‘李四’匹配的文档：

bson.D{{
	"name",
	bson.D{{
		"$in",
		bson.A{"张三","李四"},
	}},
}}

Raw类型家族用于验证字节切片。还可以使用Lookup()从原始类型检索单个元素。如果不想将BSON反序列化成另一种类型的开销，那么这是非常有用的。

本文只使用D类型。

四添加文档

4.1 创建一个结构体

type Student struct {
	Name string
	Age  int
}

4.2 添加单个文档

方法：

func (coll *Collection) InsertOne(ctx context.Context, document interface{},
	opts ...*options.InsertOneOptions) (*InsertOneResult, error)

实例演示：

package main

import (
	"context"
	"fmt"
	"log"

	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
)

var client *mongo.Client

func initDB() {
	// 设置客户端连接配置
	clientOptions := options.Client().ApplyURI("mongodb://localhost:27017")

	// 连接到MongoDB
	var err error
	c, err := mongo.Connect(context.TODO(), clientOptions)
	if err != nil {
		log.Fatal(err)
	}

	// 检查连接
	err = c.Ping(context.TODO(), nil)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println("MongoDB 连接成功")
	client = c
}

type Student struct {
	Name string
	Age  int
}

func insertOne(s Student) {
	initDB()

	collection := client.Database("golang_db").Collection("student")
	insertResult, err := collection.InsertOne(context.TODO(), s)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Printf("添加了一条文档记录: %v\n", insertResult.InsertedID)
}

func main() {
	s := Student{Name: "Psych", Age: 18}
	insertOne(s)
}

运行结果：

[Running] go run "e:\golang开发学习\go_pro\test.go"
MongoDB 连接成功
添加了一条文档记录: ObjectID("62ee3202c3c3f019d63c34ff")

[Done] exited with code=0 in 3.518 seconds

MongoDB中查看：

4.3 添加多个文档

方法：

func (coll *Collection) InsertMany(ctx context.Context, documents []interface{},
	opts ...*options.InsertManyOptions) (*InsertManyResult, error)

实例演示：

package main

import (
	"context"
	"fmt"
	"log"

	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
)

var client *mongo.Client

func initDB() {
	// 设置客户端连接配置
	clientOptions := options.Client().ApplyURI("mongodb://localhost:27017")

	// 连接到MongoDB
	var err error
	c, err := mongo.Connect(context.TODO(), clientOptions)
	if err != nil {
		log.Fatal(err)
	}

	// 检查连接
	err = c.Ping(context.TODO(), nil)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println("MongoDB 连接成功")
	client = c
}

type Student struct {
	Name string
	Age  int
}

func insertMore(students []interface{}) {
	initDB()

	collection := client.Database("golang_db").Collection("student")
	insertManyResult, err := collection.InsertMany(context.TODO(), students)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Printf("添加了多条文档记录: %v\n", insertManyResult.InsertedIDs)
}

func main() {
	s1 := Student{Name: "Klee", Age: 8}
	s2 := Student{Name: "Morax", Age: 1000}
	s3 := Student{Name: "Baal", Age: 500}
	student := []interface{}{s1, s2, s3}
	insertMore(student)
}

运行结果：

[Running] go run "e:\golang开发学习\go_pro\test.go"
MongoDB 连接成功
添加了多条文档记录: [ObjectID("62ee337502174a1366f86db2") ObjectID("62ee337502174a1366f86db3") ObjectID("62ee337502174a1366f86db4")]

[Done] exited with code=0 in 2.558 seconds

MongoDB中查看：

五查找文档

方法：

func (coll *Collection) Find(ctx context.Context, filter interface{},
	opts ...*options.FindOptions) (cur *Cursor, err error)

实例演示：

package main

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

	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
)

var client *mongo.Client

func initDB() {
	// 设置客户端连接配置
	clientOptions := options.Client().ApplyURI("mongodb://localhost:27017")

	// 连接到MongoDB
	var err error
	c, err := mongo.Connect(context.TODO(), clientOptions)
	if err != nil {
		log.Fatal(err)
	}

	// 检查连接
	err = c.Ping(context.TODO(), nil)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println("MongoDB 连接成功")
	client = c
}

func findData() {
	initDB()

	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
	defer cancel()

	collection := client.Database("golang_db").Collection("student")

	// c, err := collection.Find(ctx, bson.D{{Key: "name", Value: "Psych"}})
	// 单个数据查询
	// c, err := collection.Find(ctx, bson.D{{Key: "name", Value: "Morax"}})
	c, err := collection.Find(ctx, bson.D{})
	if err != nil {
		log.Fatal(err)
	}
	defer c.Close(ctx)

	for c.Next(ctx) {
		var result bson.D
		err := c.Decode(&result)
		if err != nil {
			log.Fatal(err)
		}
		fmt.Printf("查找结果为: %v\n", result)
		fmt.Printf("result.Map(): %v\n", result.Map()["name"])
	}
	if err := c.Err(); err != nil {
		log.Fatal(err)
	}

}

func main() {
	findData()
}

注释内容为单个数据查询，可自行测试

运行结果：

[Running] go run "e:\golang开发学习\go_pro\test.go"
MongoDB 连接成功
查找结果为: [{_id ObjectID("62ee3202c3c3f019d63c34ff")} {name Psych} {age 18}]
result.Map(): Psych
查找结果为: [{_id ObjectID("62ee337502174a1366f86db2")} {name Klee} {age 8}]
result.Map(): Klee
查找结果为: [{_id ObjectID("62ee337502174a1366f86db3")} {name Morax} {age 1000}]
result.Map(): Morax
查找结果为: [{_id ObjectID("62ee337502174a1366f86db4")} {name Baal} {age 500}]
result.Map(): Baal

[Done] exited with code=0 in 2.482 seconds

六更新文档

方法：

func (coll *Collection) UpdateOne(ctx context.Context, filter interface{}, update interface{},
	opts ...*options.UpdateOptions) (*UpdateResult, error) 
func (coll *Collection) UpdateMany(ctx context.Context, filter interface{}, update interface{},
	opts ...*options.UpdateOptions) (*UpdateResult, error)

实例演示：

package main

import (
	"context"
	"fmt"
	"log"

	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
)

var client *mongo.Client

func initDB() {
	// 设置客户端连接配置
	clientOptions := options.Client().ApplyURI("mongodb://localhost:27017")

	// 连接到MongoDB
	var err error
	c, err := mongo.Connect(context.TODO(), clientOptions)
	if err != nil {
		log.Fatal(err)
	}

	// 检查连接
	err = c.Ping(context.TODO(), nil)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println("MongoDB 连接成功")
	client = c
}

func updateData() {
	initDB()

	ctx := context.TODO()
	defer client.Disconnect(ctx)

	c := client.Database("golang_db").Collection("student")

	update := bson.D{{"$set", bson.D{{Key: "name", Value: "钟离"}, {Key: "age", Value: 1200}}}}
	// 更新条件 name Morax
	// 更新为 name 钟离 age 1200
	ur, err := c.UpdateMany(ctx, bson.D{{Key: "name", Value: "Morax"}}, update)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Printf("修改文档数量: %v\n", ur.ModifiedCount)
}

func main() {
	updateData()
}

运行结果：

[Running] go run "e:\golang开发学习\go_pro\test.go"
MongoDB 连接成功
修改文档数量: 1

[Done] exited with code=0 in 2.529 seconds

MongoDB中查看：

七删除文档

方法：

func (coll *Collection) DeleteMany(ctx context.Context, filter interface{},
	opts ...*options.DeleteOptions) (*DeleteResult, error)
func (coll *Collection) DeleteOne(ctx context.Context, filter interface{},
	opts ...*options.DeleteOptions) (*DeleteResult, error)

实例演示：

package main

import (
	"context"
	"fmt"
	"log"

	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
)

var client *mongo.Client

func initDB() {
	// 设置客户端连接配置
	clientOptions := options.Client().ApplyURI("mongodb://localhost:27017")

	// 连接到MongoDB
	var err error
	c, err := mongo.Connect(context.TODO(), clientOptions)
	if err != nil {
		log.Fatal(err)
	}

	// 检查连接
	err = c.Ping(context.TODO(), nil)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println("MongoDB 连接成功")
	client = c
}

func deleteData() {
	initDB()

	ctx := context.TODO()
	defer client.Disconnect(ctx)

	c := client.Database("golang_db").Collection("student")
	dr, err := c.DeleteMany(ctx, bson.D{{Key: "name", Value: "Psych"}})
	if err != nil {
		log.Fatal(err)
	}
	fmt.Printf("删除文档数量: %v\n", dr.DeletedCount)
}

func main() {
	deleteData()
}

运行结果：

[Running] go run "e:\golang开发学习\go_pro\test.go"
MongoDB 连接成功
删除文档数量: 1

[Done] exited with code=0 in 3.698 seconds

MongoDB中查看：

总结

到此这篇关于Go语言学习笔记之golang操作MongoDB数据库的文章就介绍到这了,更多相关golang操作MongoDB数据库内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

golang中cache组件的使用及groupcache源码解析
本篇主要解析groupcache源码中的关键部分， lru的定义以及如何做到同一个key只加载一次。缓存填充以及加载抑制的实现方法，本文重点给大家介绍golang中cache组件的使用及groupcache源码解析，感兴趣的朋友一起看看吧
2021-06-06
关于Go 是传值还是传引用?
这篇文章主要讨论Go语言是传值还是传引用?文章先从Go 官方的定义展开，随后是传值和传引用得介绍到map 和 slice得区别,需要的小伙伴可以参考一下文章得具体内容
2021-10-10
详解Golang如何动态获取配置文件
项目中经常获取一些常用配置文件,当有配置文件中的参数被修改后,如何的实时获取配置文件就很关键了,下面小编就来讲讲如何利用viper实时获取配置文件吧
2023-08-08
go语言分布式id生成器及分布式锁介绍
这篇文章主要为大家介绍了go语言分布式id生成器及分布式锁介绍，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
2023-04-04
Go语言实现简单留言板的方法
这篇文章主要介绍了Go语言实现简单留言板的方法,涉及数据库、模板页面元素等留言板相关技巧,具有一定参考借鉴价值,需要的朋友可以参考下
2015-02-02
golang1.21新特性全面讲解
经过了半年左右的开发，golang 1.21 最近正式发布了，这个版本中有不少重要的新特性和变更，尤其是在泛型相关的代码上，下面小编就来和大家好好唠唠吧
2023-08-08
ubuntu安装golang并设置goproxy的方法步骤
在Ubuntu系统上安装Go语言（Golang）有多种方法,包括使用包管理器、从源代码编译安装以及使用版本管理工具如gvm,安装完成后,为了方便管理Go语言项目依赖,需要设置GOPATH环境变量并配置Go代理,本文介绍ubuntu安装golang并设置goproxy的方法,感兴趣的朋友一起看看吧
2024-10-10
Go 验证字符串中是否包含中文(推荐)
这篇文章主要介绍了Go 验证字符串中是否包含中文,本文通过实例代码给大家介绍的非常详细，具有一定的参考借鉴价值，需要的朋友可以参考下
2020-01-01
Go gRPC服务端流式RPC教程示例
这篇文章主要为大家介绍了Go gRPC服务端流式RPC教程示例，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
2022-06-06
golang 日志库ZAP[uber-go zap]示例详解
ZAP是由Uber开源的高性能Go语言日志库,支持多种日志级别及基本信息打印,虽然ZAP本身不支持日志分割,但可以结合lumberjack进行日志切割,实现日志按文件大小、时间或间隔切割等功能,ZAP提供Logger和SugaredLogger两种日志记录器
2024-10-10