Golang操作MongoDB：mongo-driver

lena7

已于 2022-02-18 14:45:45 修改

阅读量3.4k

点赞数 1

分类专栏：数据库 Golang 文章标签： mongodb golang 数据库

于 2022-02-18 14:42:13 首次发布

本文链接：https://blog.csdn.net/lena7/article/details/123003131

版权

本文介绍了如何在Golang中使用mongo-driver库连接和操作MongoDB数据库，包括获取依赖、建立连接、创建文档实体类以及执行数据库操作。在初始化阶段，通过在包的init方法中设置数据库和集合的连接，确保后续操作的顺利进行。同时提供了官方API文档和参考文章链接以供深入学习。

摘要由CSDN通过智能技术生成

1、获取依赖：go get "go.mongodb.org/mongo-driver"

2、连接数据库
前提：确保mongo进程已启动：ps -ef|grep mongo

import (
	"context"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
	"log"
)

var db *mongo.Database			// 数据库
var stuColl *mongo.Collection	// 集合

func init() {
   
	option := options.Client().ApplyURI("mongodb://localhost:27017")
	client, err := mongo.Connect(context.TODO(), option)
	if err != nil {
   
		log.Fatal(err)
	}
	err = client.Ping(context.TODO(), nil)
	if err != nil {
   
		log.Fatal(err)
	}
	log.Print("connect mongo success")
	db = client.Database("test")
	stuColl = db.Collection("student")
}

这里我在当前包的init方法中完成数据库和集合的初始化，这样每次执行该包的方法都会先执行init()函数，保证了数据库和集合的连接。后面进行操作就直接调用全局变量即可。

3、构建文档对应实体类

import (
	"go.mongodb.org/mongo-driver/bson/primitive"
)

type Student struct {
   
	Id primitive.ObjectID `bson:"_id,omitempty"`
	Name string
	Age int
}

4、操作数据库

import (
	"context"
	"fmt"
	"go.mongodb.org/mongo-driver/bson"
	"log"
)

// InsertOne 添加一条记录
func InsertOne(stu Student) {
   
	one, err := stuColl.InsertOne(context.TODO(), stu)
	if err != nil {
   
		log.Fatal(err)
	}
	fmt.Println("insert one success :",one.InsertedID)
}

// InsertMany 添加多条记录
func InsertMany(stu []interface{
   }) {
   
	// InsertMany 所需的第二个参数是interface{}型切片
	many