Skip to content

Latest commit

 

History

History
246 lines (220 loc) · 5.71 KB

5.4.md

File metadata and controls

246 lines (220 loc) · 5.71 KB
Raw

Go 操作mongodb

添加mongodb驱动程序

用于go get将 Go 驱动程序添加为依赖项。

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

使用方法

创建main.go 文件

package main

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

// MongoDB 连接池
var MongoDBClient *mongo.Database

// pool 连接池模式
func ConnectToDBPool() {
	user := "admin"
	password := "12345678"
	host := "127.0.0.1"
	port := "27017"
	dbName := "demo"
	timeOut := 2
	maxNum := 50

	uri := fmt.Sprintf("mongodb://%s:%s@%s:%s/%s?w=majority", user, password, host, port, dbName)
	// 设置连接超时时间
	ctx, cancel := context.WithTimeout(context.Background(), time.Duration(timeOut))
	defer cancel()
	// 通过传进来的uri连接相关的配置
	o := options.Client().ApplyURI(uri)
	// 设置最大连接数 - 默认是100 ，不设置就是最大 max 64
	o.SetMaxPoolSize(uint64(maxNum))
	// 发起链接
	client, err := mongo.Connect(ctx, o)
	if err != nil {
		fmt.Println("ConnectToDB", err)
		return
	}
	// 判断服务是不是可用
	if err = client.Ping(context.Background(), readpref.Primary()); err != nil {
		fmt.Println("ConnectToDB", err)
		return
	}
	// 返回 client
	MongoDBClient = client.Database(dbName)
}
func ConnectToDB() {
	clientOptions := options.Client().ApplyURI("mongodb://admin:12345678@localhost:27017")
	var ctx = context.TODO()
	// Connect to MongoDB
	client, err := mongo.Connect(ctx, clientOptions)
	if err != nil {
		log.Fatal(err)
	}
	// Check the connection
	err = client.Ping(ctx, nil)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println("Connected to MongoDB!")

	// 返回 client
	MongoDBClient = client.Database("demo")

	//defer client.Disconnect(ctx)
}

// 插入单条数据
func insertOne() {
	ash := Member{"13212345678", "123456", []string{"abc1", "efg1", "hij1"}}
	insertResult, err := MongoDBClient.Collection("test1").InsertOne(context.TODO(), ash)
	if err != nil {
		fmt.Println(err)
	}
	println("Inserted a single document: ", insertResult.InsertedID)
}

// 插入多条数据
func insert() {
	var ash []interface{}
	ash = append(ash, Member{"13222222222", "123456", []string{"aaa", "bbb", "ccc"}})
	ash = append(ash, Member{"13333333333", "123456", []string{"aaa1", "bbb1", "ccc1"}})
	fmt.Println(ash)
	insertResult, err := MongoDBClient.Collection("test1").InsertMany(context.TODO(), ash)
	if err != nil {
		fmt.Println(err)
	}
	println("Inserted Multiple document: ", insertResult.InsertedIDs)
}

// 查询单条
func findOne() {
	var result bson.M
	err := MongoDBClient.Collection("test1").FindOne(context.TODO(), bson.D{{"info", "aaa1"}}).Decode(&result)
	if err != nil {
		if err == mongo.ErrNoDocuments {
			//This error means your query did not match any documents.
			return
		}
		panic(err)
	}
	fmt.Println(result)

}

// 查询多条数据
func find() {
	findOptions := options.Find()
	findOptions.SetLimit(10)
	cur, err := MongoDBClient.Collection("test1").Find(context.TODO(), bson.D{{"phone", "13333333333"}}, findOptions)
	if err != nil {
		fmt.Println(err)
	}
	var results []*Member
	for cur.Next(context.TODO()) {
		// create a value into which the single document can be decoded
		var elem Member
		err := cur.Decode(&elem)
		if err != nil {
			fmt.Println(err)
		}

		results = append(results, &elem)
	}

	if err := cur.Err(); err != nil {
		fmt.Println(err)
	}
	//fmt.Println(results)
	for _, v := range results {
		fmt.Println(v.Phone)
		fmt.Println(v.Name)
		fmt.Println(v.Info)
	}
}
func updateOne() {
	//如果过滤的文档不存在，则插入新的文档
	opts := options.Update().SetUpsert(true)
	id, _ := primitive.ObjectIDFromHex("633b02b6e082e5046001d0b9")
	filter := bson.D{{"_id", id}}
	update := bson.D{{"$set", bson.D{{"phone", "1444444444444"}}}}
	result, err := MongoDBClient.Collection("test1").UpdateOne(context.TODO(), filter, update, opts)
	//result, err := MongoDBClient.Collection("test1").UpdateOne(context.TODO(), filter, update)
	if err != nil {
		panic(err)
	}
	fmt.Println(result)
}

func update() {
	filter := bson.D{{"name", "123456"}}
	update := bson.D{{"$set", bson.D{{"name", "张三"}}}}
	result, err := MongoDBClient.Collection("test1").UpdateMany(context.TODO(), filter, update)
	if err != nil {
		panic(err)
	}
	fmt.Println(result)
}

// 替换文档
func replaceOne() {
	filter := bson.D{{"phone", "13222222222"}}
	replacement := bson.D{{"phone", "16666666666"}}
	result, err := MongoDBClient.Collection("test1").ReplaceOne(context.TODO(), filter, replacement)
	if err != nil {
		panic(err)
	}
	fmt.Println(result)
}

// 删除单个文件
func deleteOne() {
	filter := bson.D{{"phone", "16666666666"}}
	result, err := MongoDBClient.Collection("test1").DeleteOne(context.TODO(), filter)
	if err != nil {
		panic(err)
	}
	fmt.Println(result)
}

// 删除多个
func delete() {
	//filter := bson.D{{"runtime", bson.D{{"$gt", 800}}}}
	filter := bson.D{{"phone", "16666666666"}}
	results, err := MongoDBClient.Collection("test1").DeleteMany(context.TODO(), filter)
	if err != nil {
		panic(err)
	}
	fmt.Println(results)
}

type Member struct {
	Phone string
	Name  string
	Info  []string
}

func main() {
	//连接数据库
	ConnectToDB()
	//连接池连接数据库
	//ConnectToDBPool()
	//插入单条数据
	//insertOne()
	//插入多条数据
	//insert()
	//查找单条数据
	//findOne()
	//查找多条数据
	find()
	//修改单条数据
	//updateOne()
	//修改多条数据
	//update()
	//替换文档
	//replaceOne()
	// 删除多个
	//deleteOne()
	// 删除多个
	//delete()
}

links

https://www.mongodb.com/docs/drivers/go/current/

  • 目录
  • 上一节：
  • 下一节：