golang连接MongoDB数据库及数据库操作指南

go get -u github.com/mongodb/mongo-go-driver

go mod init 模块名称

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!")
    }
}

bson.D{{"foo", "bar"}, {"hello", "world"}, {"pi", 3.14159}}

bson.M{"foo": "bar", "hello": "world", "pi": 3.14159}

bson.A{"hello", "world", 3.14159, bson.D{{"abcde", 12345}}}

import "go.mongodb.org/mongo-driver/bson"

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)
}

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
}