后台启动 MongoDB 服务器端:
- nohup ./bin/mongod --dbpath=./data &
- 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"
- "os"
- "time"
- )
- type Howie struct {
- //struct 里面获取 ObjectID
- HowieId primitive.ObjectID `bson:"_id"`
- Name string
- Pwd string
- Age int64
- CreateTime int64
- }
- func main() {
- TestMongo("mongodb://0.0.0.0:27017")
- }
- func TestMongo(url string) {
- var (
- err error
- client *mongo.Client
- collection *mongo.Collection
- insertOneRes *mongo.InsertOneResult
- insertManyRes *mongo.InsertManyResult
- delRes *mongo.DeleteResult
- updateRes *mongo.UpdateResult
- cursor *mongo.Cursor
- howieArray = GetHowieArray()
- howie Howie
- howieArrayEmpty []Howie
- size int64
- )
- // 链接 mongo 服务
- if client, err = mongo.Connect(getContext(), options.Client().ApplyURI(url)); err != nil {
- checkErr(err)
- }
- // 判断服务是否可用
- if err = client.Ping(getContext(), readpref.Primary()); err != nil {
- checkErr(err)
- }
- // 选择数据库和集合
- collection = client.Database("testing_base").Collection("howie")
- // 删除这个集合
- collection.Drop(getContext())
- // 插入一条数据
- if insertOneRes, err = collection.InsertOne(getContext(), howieArray[0]); err != nil {
- checkErr(err)
- }
- fmt.Printf("InsertOne 插入的消息 ID:%v\n", insertOneRes.InsertedID)
- // 批量插入数据
- if insertManyRes, err = collection.InsertMany(getContext(), howieArray[1:]); err != nil {
- checkErr(err)
- }
- fmt.Printf("InsertMany 插入的消息 ID:%v\n", insertManyRes.InsertedIDs)
- var Dinfo = make(map[string]interface{})
- err = collection.FindOne(getContext(), bson.D{{"name", "howie_2"}, {"age", 11}}).Decode(&Dinfo)
- fmt.Println(Dinfo)
- fmt.Println("_id", Dinfo["_id"])
- // 查询单条数据
- if err = collection.FindOne(getContext(), bson.D{{"name", "howie_2"}, {"age", 11}}).Decode(&howie); err != nil {
- checkErr(err)
- }
- fmt.Printf("FindOne 查询到的数据:%v\n", howie)
- // 查询单条数据后删除该数据
- if err = collection.FindOneAndDelete(getContext(), bson.D{{"name", "howie_3"}}).Decode(&howie); err != nil {
- checkErr(err)
- }
- fmt.Printf("FindOneAndDelete 查询到的数据:%v\n", howie)
- // 查询单条数据后修改该数据
- if err = collection.FindOneAndUpdate(getContext(), bson.D{{"name", "howie_4"}}, bson.M{"$set": bson.M{"name": "这条数据我需要修改了"}}).Decode(&howie); err != nil {
- checkErr(err)
- }
- fmt.Printf("FindOneAndUpdate 查询到的数据:%v\n", howie)
- // 查询单条数据后替换该数据 (以前的数据全部清空)
- if err = collection.FindOneAndReplace(getContext(), bson.D{{"name", "howie_5"}}, bson.M{"hero": "这条数据我替换了"}).Decode(&howie); err != nil {
- checkErr(err)
- }
- fmt.Printf("FindOneAndReplace 查询到的数据:%v\n", howie)
- // 一次查询多条数据
- // 查询 createtime>=3
- // 限制取 2 条
- // createtime 从大到小排序的数据
- if cursor, err = collection.Find(getContext(), bson.M{"createtime": bson.M{"$gte": 2}}, options.Find().SetLimit(2), options.Find().SetSort(bson.M{"createtime": -1})); err != nil {
- checkErr(err)
- }
- if err = cursor.Err(); err != nil {
- checkErr(err)
- }
- defer cursor.Close(context.Background())
- for cursor.Next(context.Background()) {
- if err = cursor.Decode(&howie); err != nil {
- checkErr(err)
- }
- howieArrayEmpty = append(howieArrayEmpty, howie)
- }
- for _, v := range howieArrayEmpty {
- fmt.Printf("Find 查询到的数据 ObejectId 值 %s 值:%v\n", v.HowieId.Hex(), v)
- }
- // 查询集合里面有多少数据
- if size, err = collection.CountDocuments(getContext(),bson.D{}); err != nil {
- checkErr(err)
- }
- fmt.Printf("Count 里面有多少条数据:%d\n", size)
- // 查询集合里面有多少数据 (查询 createtime>=3 的数据)
- if size, err = collection.CountDocuments(getContext(), bson.M{"createtime": bson.M{"$gte": 3}}); err != nil {
- checkErr(err)
- }
- fmt.Printf("Count 里面有多少条数据:%d\n", size)
- // 修改一条数据
- if updateRes, err = collection.UpdateOne(getContext(), bson.M{"name": "howie_2"}, bson.M{"$set": bson.M{"name": "我要改了他的名字"}}); err != nil {
- checkErr(err)
- }
- fmt.Printf("UpdateOne 的数据:%d\n", updateRes)
- // 修改多条数据
- if updateRes, err = collection.UpdateMany(getContext(), bson.M{"createtime": bson.M{"$gte": 3}}, bson.M{"$set": bson.M{"name": "我要批量改了他的名字"}}); err != nil {
- checkErr(err)
- }
- fmt.Printf("UpdateMany 的数据:%d\n", updateRes)
- // 删除一条数据
- if delRes, err = collection.DeleteOne(getContext(), bson.M{"name": "howie_1"}); err != nil {
- checkErr(err)
- }
- fmt.Printf("DeleteOne 删除了多少条数据:%d\n", delRes.DeletedCount)
- // 删除多条数据
- if delRes, err = collection.DeleteMany(getContext(), bson.M{"createtime": bson.M{"$gte": 7}}); err != nil {
- checkErr(err)
- }
- fmt.Printf("DeleteMany 删除了多少条数据:%d\n", delRes.DeletedCount)
- }
- func checkErr(err error) {
- if err != nil {
- if err == mongo.ErrNoDocuments {
- fmt.Println("没有查到数据")
- os.Exit(0)
- } else {
- fmt.Println(err)
- os.Exit(0)
- }
- }
- }
- func getContext() (ctx context.Context) {
- ctx, _ = context.WithTimeout(context.Background(), 10*time.Second)
- return
- }
- func GetHowieArray() (data []interface{}) {
- var i int64
- for i = 0; i <= 10; i++ {
- data = append(data, Howie{
- HowieId: primitive.NewObjectID(),
- Name: fmt.Sprintf("howie_%d", i+1),
- Pwd: fmt.Sprintf("pwd_%d", i+1),
- Age: i + 10,
- CreateTime: i + 1,
- })
- }
- return
- }
注意: 从 GitHub.com/MongoDB/mongo-go-driver 下载下来要放到 go.MongoDB.org 目录对应的位置, 不然会报错
来源: http://www.bubuko.com/infodetail-3030870.html