Advertisement

golang连接MongoDB数据库

阅读量:

golang连接MongoDB数据库

  • 安装配置MongoDB驱动

    • 导入驱动
  • 获得数据库连接

  • 数据库操作

    • 添加文档
    • 查找文档
    • 更新文档
    • 删除文档

安装配置MongoDB驱动

安装驱动

复制代码
    go get -u github.com/mongodb/mongo-go-driver

初始化模块

复制代码
    go mod init 模块名称

执行go mod tidy

复制代码
    go mod tidy

导入驱动

复制代码
    import (
    "context"
    "go.mongodb.org/mongo-driver/bson"    //BOSN解析包
    "go.mongodb.org/mongo-driver/mongo"    //MongoDB的Go驱动包
    "go.mongodb.org/mongo-driver/mongo/options"
    )

获得数据库连接

初始化连接

复制代码
    package main
    
    import (
    "context"
    "fmt"
    "go.mongodb.org/mongo-driver/mongo"
    "go.mongodb.org/mongo-driver/mongo/options"
    )
    
    var client *mongo.Client
    
    func initDB() (err error) {
    // 设置客户端连接配置
    clientOptions := options.Client().ApplyURI("mongodb://ip:port")
    // 连接到MongoDB
    client, err = mongo.Connect(context.TODO(), clientOptions)
    if err != nil {
        return err
    }
    // 检查连接
    err = client.Ping(context.TODO(), nil)
    if err != nil {
        return err
    }
    return nil
    }
    
    func main() {
    err := initDB() // 调用输出化数据库的函数
    if err != nil {
        fmt.Printf("初始化失败!,err:%v\n", err)
        return
    }else{
    	fmt.Println("Connected to MongoDB!")
    }
    }

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

类型D家族用于简洁地构建本地Go类型的BSON对象。对于构造传递给MongoDB的命令而言这一方法非常有效。其中包含四类:如A、B、C、D类别等

D标识符:一种BSON文档。这种数据类型常用于有序性要求高的场景中,例如MongoDB命令

复制代码
    bson.D{{"foo", "bar"}, {"hello", "world"}, {"pi", 3.14159}}

M:一张无序的map。它和D是一样的,只是它不保持顺序。

复制代码
    bson.M{"foo": "bar", "hello": "world", "pi": 3.14159}

A:一个BSON数组。

复制代码
    bson.A{"hello", "world", 3.14159, bson.D{{"abcde", 12345}}}

E:D里面的一个元素。

要使用BSON,需要先导入下面的包:

复制代码
    import "go.mongodb.org/mongo-driver/bson"

Base类型家族被用来验证字节切片。你可以通过Lookup()方法从原始类型中检查单个元素。如果不希望在转换过程中curring another type of overhead, 这是非常有价值的.

数据库操作

添加文档

添加单个文档

使用collection.InsertOne()方法插入一条文档记录:

复制代码
    func insertOne(s interface{}) (err error) {
    collection := client.Database("datebase").Collection("collection")
    insertResult, err := collection.InsertOne(context.TODO(), s)
    if err != nil {
    	return err
    }
    fmt.Println("Inserted a single document: ", insertResult.InsertedID)
    return nil
    }

测试:

复制代码
    s := struct{name string
    	age int
    }{"tom", 20}
    err := insertOne(s)
    if err != nil{
    	fmt.Printf("添加文档失败!,err:%v\n", err)
    }

添加多个文档

使用collection.InsertMany()方法插入多条文档记录:

复制代码
    func insertMore(s []interface{}) (err error) {
    //students := []interface{}{s2, s3}
    collection := client.Database("datebase").Collection("collection")
    insertManyResult, err := collection.InsertMany(context.TODO(), s)
    if err != nil {
        return err
    }
    fmt.Println("Inserted multiple documents: ", insertManyResult.InsertedIDs)
    return nil
    }

测试:

复制代码
    type Student struct {
    name string
    age int
    }
    s1 := Student{name: "tom", age: 20}
    s2 := Student{name: "kite", age: 21}
    s3 := Student{name: "rose", age: 22}
    ss := []interface{}{s1, s2, s3}
    insertMore(ss)
    if err != nil{
    	fmt.Printf("添加多个文档失败!,err:%v\n", err)
    }

查找文档

复制代码
    func find() (err error){
    ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
    defer cancel()
    collection := client.Database("datebase").Collection("collection")
    cur, err := collection.Find(ctx, bson.D{})
    if err != nil {
        return err
    }
    defer cur.Close(ctx)
    for cur.Next(ctx) {
        var result bson.D
        err := cur.Decode(&result)
        if err != nil {
            return err
        }
        fmt.Printf("result: %v\n", result)
    }
    if err := cur.Err(); err != nil {
        return err
    }
    return nil
    }

更新文档

复制代码
    func update() (err error) {
    ctx := context.TODO()
    defer client.Disconnect(ctx)
    c := client.Database("datebase").Collection("collection")
    
    update := bson.D{{"$set", bson.D{{"attr1", "value1"}, {"attr2", "value2"}}}}
    
    ur, err := c.UpdateMany(ctx, bson.D{{"old_attr1", "old_value1"}}, update)
    if err != nil {
        return err
    }
    fmt.Printf("ur.ModifiedCount: %v\n", ur.ModifiedCount)
    return nil
    }

删除文档

复制代码
    func del() (err error) {
    c := client.Database("datebase").Collection("collection")
    ctx := context.TODO()
    
    dr, err := c.DeleteMany(ctx, bson.D{{"attr1", "value1"}})
    
    if err != nil {
        return err
    }
    fmt.Printf("ur.ModifiedCount: %v\n", dr.DeletedCount)
    	return nil
    }

全部评论 (0)

还没有任何评论哟~