开发环境

  1. $ docker run -d  --name mongo \
  2.     -p 27017:27017 \
  3.     -e MONGO_INITDB_ROOT_USERNAME=admin \
  4.     -e MONGO_INITDB_ROOT_PASSWORD=654321 \
  5.     mongo:4.0

go.mod 设置

  1. go 1.14
  3. require go.mongodb.org/mongo-driver v1.4.6

连接

  1. package main
  3. import (
  4.     "context"
  5.     "log"
  6.     "time"
  8.     "go.mongodb.org/mongo-driver/mongo"
  9.     "go.mongodb.org/mongo-driver/mongo/options"
  10.     "go.mongodb.org/mongo-driver/mongo/readpref"
  11. )
  13. func main() {
  14.     ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
  15.     defer cancel()
  16.     client, err := mongo.Connect(ctx, options.Client().ApplyURI("mongodb://localhost:27017"))
  17.     err = client.Ping(ctx, readpref.Primary())
  18.     if err != nil {
  19.         log.Fatal(err)
  20.     }
  21.     log.Println("mongodb connect success")
  22. }

连接和集合

  1. //  database
  2.     database := client.Database("my_db")
  3.     //  collection
  4.     collection := database.Collection("my_collection")
  5.     log.Println(collection.Name())

插入一条数据

  1. package main
  3. import (
  4.     "context"
  5.     "log"
  6.     "time"
  8.     "go.mongodb.org/mongo-driver/bson/primitive"
  9.     "go.mongodb.org/mongo-driver/mongo"
  10.     "go.mongodb.org/mongo-driver/mongo/options"
  11.     "go.mongodb.org/mongo-driver/mongo/readpref"
  12. )
  14. type Student struct {
  15.     Name   string `bson:"name"` // 名字
  16.     Gender string `bson:gender` // 性别
  17.     Age    int8   `bson:"age"`  // 年龄
  18. }
  20. func main() {
  21.     clientOptions := options.Client().SetHosts([]string{"localhost:27017"}).
  22.         SetConnectTimeout(5 * time.Second).
  23.         SetAuth(options.Credential{Username: "admin", Password: "654321"})
  24.     ctx := context.TODO()
  25.     client, err := mongo.Connect(ctx, clientOptions)
  26.     err = client.Ping(ctx, readpref.Primary())
  27.     if err != nil {
  28.         log.Fatal(err)
  29.     }
  30.     log.Println("mongodb connect success")
  31.     //  database
  32.     database := client.Database("my_db")
  33.     //  collection
  34.     collection := database.Collection("my_collection")
  36.     s := &Student{
  37.         Name:   "张三",
  38.         Gender: "男",
  39.         Age:    20,
  40.     }
  41.     res, err := collection.InsertOne(context.TODO(), s)
  42.     if err != nil {
  43.         log.Fatal(err)
  44.     }
  45.     docId := res.InsertedID.(primitive.ObjectID)
  46.     recordId := docId.Hex()
  47.     log.Println("insert one ID str is :", recordId)
  48. }

插入多条数据

  1. package main
  3. import (
  4.     "context"
  5.     "log"
  6.     "time"
  8.     "go.mongodb.org/mongo-driver/mongo"
  9.     "go.mongodb.org/mongo-driver/mongo/options"
  10.     "go.mongodb.org/mongo-driver/mongo/readpref"
  11. )
  13. type Student struct {
  14.     Name   string `bson:"name"` // 名字
  15.     Gender string `bson:gender` // 性别
  16.     Age    int8   `bson:"age"`  // 年龄
  17. }
  19. func main() {
  20.     clientOptions := options.Client().SetHosts([]string{"localhost:27017"}).
  21.         SetConnectTimeout(5 * time.Second).
  22.         SetAuth(options.Credential{Username: "admin", Password: "654321"})
  23.     ctx := context.TODO()
  24.     client, err := mongo.Connect(ctx, clientOptions)
  25.     err = client.Ping(ctx, readpref.Primary())
  26.     if err != nil {
  27.         log.Fatal(err)
  28.     }
  29.     log.Println("mongodb connect success")
  31.     collection := client.Database("my_db").Collection("my_collection")
  33.     s1 := &Student{
  34.         Name:   "李四",
  35.         Gender: "男",
  36.         Age:    18,
  37.     }
  39.     s2 := &Student{
  40.         Name:   "王五",
  41.         Gender: "女",
  42.         Age:    30,
  43.     }
  44.     res, err := collection.InsertMany(context.TODO(), []interface{}{s1, s2})
  45.     if err != nil {
  46.         log.Fatal(err)
  47.     }
  48.     log.Println("insert manyID is :", res)
  49. }

查找单条数据

  1. package main
  3. import (
  4.     "context"
  5.     "go.mongodb.org/mongo-driver/bson"
  6.     "log"
  7.     "time"
  9.     "go.mongodb.org/mongo-driver/mongo"
  10.     "go.mongodb.org/mongo-driver/mongo/options"
  11.     "go.mongodb.org/mongo-driver/mongo/readpref"
  12. )
  14. type Student struct {
  15.     Name   string `bson:"name"` // 名字
  16.     Gender string `bson:gender` // 性别
  17.     Age    int8   `bson:"age"`  // 年龄
  18. }
  20. func main() {
  21.     clientOptions := options.Client().SetHosts([]string{"localhost:27017"}).
  22.         SetConnectTimeout(5 * time.Second).
  23.         SetAuth(options.Credential{Username: "admin", Password: "654321"})
  24.     ctx := context.TODO()
  25.     client, err := mongo.Connect(ctx, clientOptions)
  26.     err = client.Ping(ctx, readpref.Primary())
  27.     if err != nil {
  28.         log.Fatal(err)
  29.     }
  30.     log.Println("mongodb connect success")
  32.     collection := client.Database("my_db").Collection("my_collection")
  34.     filter := bson.M{"name": "张三"}
  35.     res := collection.FindOne(ctx, filter)
  36.     if res == nil || res.Err() != nil {
  37.         log.Fatal("find one error ", res.Err().Error())
  38.     }
  39.     var s Student
  40.     err = res.Decode(&s)
  41.     if err != nil {
  42.         log.Fatal("find one failed error is ", err)
  43.     }
  44.     log.Println("find one success, res is ", s)
  45. }

查找多条

  1. package main
  3. import (
  4.     "context"
  5.     "log"
  6.     "time"
  8.     "go.mongodb.org/mongo-driver/bson"
  9.     "go.mongodb.org/mongo-driver/mongo"
  10.     "go.mongodb.org/mongo-driver/mongo/options"
  11.     "go.mongodb.org/mongo-driver/mongo/readpref"
  12. )
  14. type Student struct {
  15.     ID     string `bson:"_id,omitempty"`
  16.     Name   string `bson:"name"` // 名字
  17.     Gender string `bson:gender` // 性别
  18.     Age    int8   `bson:"age"`  // 年龄
  19. }
  21. func main() {
  22.     clientOptions := options.Client().SetHosts([]string{"localhost:27017"}).
  23.         SetConnectTimeout(5 * time.Second).
  24.         SetAuth(options.Credential{Username: "admin", Password: "654321"})
  25.     ctx := context.TODO()
  26.     client, err := mongo.Connect(ctx, clientOptions)
  27.     err = client.Ping(ctx, readpref.Primary())
  28.     if err != nil {
  29.         log.Fatal(err)
  30.     }
  31.     log.Println("mongodb connect success")
  33.     collection := client.Database("my_db").Collection("my_collection")
  35.     filter := bson.M{"age": bson.M{"$gte": 20}}
  36.     cursor, err := collection.Find(ctx, filter)
  37.     if err != nil {
  38.         log.Fatal("find error ", err)
  39.         return
  40.     }
  41.     defer cursor.Close(context.Background())
  42.     list := make([]Student, 0, cursor.RemainingBatchLength())
  43.     for cursor.Next(context.Background()) {
  44.         var s Student
  45.         err := cursor.Decode(&s)
  46.         if err != nil {
  47.             log.Println("decode error is ", err)
  48.             continue
  49.         }
  50.         list = append(list, s)
  51.     }
  52.     log.Println("find result ", list)
  53. }

分页查找

  1. package main
  3. import (
  4.     "context"
  5.     "log"
  6.     "time"
  8.     "go.mongodb.org/mongo-driver/bson"
  9.     "go.mongodb.org/mongo-driver/mongo"
  10.     "go.mongodb.org/mongo-driver/mongo/options"
  11.     "go.mongodb.org/mongo-driver/mongo/readpref"
  12. )
  14. type Student struct {
  15.     ID     string `bson:"_id,omitempty"`
  16.     Name   string `bson:"name"` // 名字
  17.     Gender string `bson:gender` // 性别
  18.     Age    int8   `bson:"age"`  // 年龄
  19. }
  21. func main() {
  22.     clientOptions := options.Client().SetHosts([]string{"localhost:27017"}).
  23.         SetConnectTimeout(5 * time.Second).
  24.         SetAuth(options.Credential{Username: "admin", Password: "654321"})
  25.     ctx := context.TODO()
  26.     client, err := mongo.Connect(ctx, clientOptions)
  27.     err = client.Ping(ctx, readpref.Primary())
  28.     if err != nil {
  29.         log.Fatal(err)
  30.     }
  31.     log.Println("mongodb connect success")
  33.     collection := client.Database("my_db").Collection("my_collection")
  35.     filter := bson.M{"age": bson.M{"$gte": 0}}
  37.     sort := bson.D{{"age", -1}}
  38.     findOptions := options.Find().SetSort(sort)
  39.     //从第1页获取，每次获取5条
  40.     page := 1
  41.     size := 5
  42.     skipTmp := int64((page - 1) * size)
  43.     limitTmp := int64(size)
  44.     findOptions.Skip = &skipTmp
  45.     findOptions.Limit = &limitTmp
  46.     cursor, err := collection.Find(ctx, filter, findOptions)
  47.     defer cursor.Close(context.Background())
  48.     if err != nil {
  49.         log.Fatal("limit page error is ", err)
  50.     }
  51.     for cursor.Next(context.Background()) {
  52.         var s Student
  53.         err := cursor.Decode(&s)
  54.         if err != nil {
  55.             log.Println("data decode error", err)
  56.             continue
  57.         }
  58.         log.Println(s)
  59.     }
  60. }

更新单条数据

  1. package main
  3. import (
  4.     "context"
  5.     "log"
  6.     "time"
  8.     "go.mongodb.org/mongo-driver/bson"
  9.     "go.mongodb.org/mongo-driver/mongo"
  10.     "go.mongodb.org/mongo-driver/mongo/options"
  11.     "go.mongodb.org/mongo-driver/mongo/readpref"
  12. )
  14. type Student struct {
  15.     ID     string `bson:"_id,omitempty"`
  16.     Name   string `bson:"name"` // 名字
  17.     Gender string `bson:gender` // 性别
  18.     Age    int8   `bson:"age"`  // 年龄
  19. }
  21. func main() {
  22.     clientOptions := options.Client().SetHosts([]string{"localhost:27017"}).
  23.         SetConnectTimeout(5 * time.Second).
  24.         SetAuth(options.Credential{Username: "admin", Password: "654321"})
  25.     ctx := context.TODO()
  26.     client, err := mongo.Connect(ctx, clientOptions)
  27.     err = client.Ping(ctx, readpref.Primary())
  28.     if err != nil {
  29.         log.Fatal(err)
  30.     }
  31.     log.Println("mongodb connect success")
  33.     collection := client.Database("my_db").Collection("my_collection")
  35.     filter := bson.M{"name": "李四"}
  36.     value := bson.M{"$set": bson.M{
  37.         "age": 25}}
  39.     res, err := collection.UpdateOne(ctx, filter, value)
  40.     if err != nil {
  41.         log.Println("update data err", err)
  42.         return
  43.     }
  44.     log.Println("update count", res.ModifiedCount)
  45. }

修改多条数据

  1. package main
  3. import (
  4.     "context"
  5.     "log"
  6.     "time"
  8.     "go.mongodb.org/mongo-driver/bson"
  9.     "go.mongodb.org/mongo-driver/mongo"
  10.     "go.mongodb.org/mongo-driver/mongo/options"
  11.     "go.mongodb.org/mongo-driver/mongo/readpref"
  12. )
  14. type Student struct {
  15.     ID     string `bson:"_id,omitempty"`
  16.     Name   string `bson:"name"` // 名字
  17.     Gender string `bson:gender` // 性别
  18.     Age    int8   `bson:"age"`  // 年龄
  19. }
  21. func main() {
  22.     clientOptions := options.Client().SetHosts([]string{"localhost:27017"}).
  23.         SetConnectTimeout(5 * time.Second).
  24.         SetAuth(options.Credential{Username: "admin", Password: "654321"})
  25.     ctx := context.TODO()
  26.     client, err := mongo.Connect(ctx, clientOptions)
  27.     err = client.Ping(ctx, readpref.Primary())
  28.     if err != nil {
  29.         log.Fatal(err)
  30.     }
  31.     log.Println("mongodb connect success")
  33.     collection := client.Database("my_db").Collection("my_collection")
  35.     var names = []string{"张三", "李四"}
  36.     filter := bson.M{"name": bson.M{"$in": names}}
  38.     value := bson.M{"$set": bson.M{"age": 22}}
  39.     result, err := collection.UpdateMany(ctx, filter, value)
  40.     if err != nil {
  41.         log.Fatal("update many error", err)
  42.     }
  43.     log.Println("update count ", result.ModifiedCount)
  44. }

删除

  1. package main
  3. import (
  4.     "context"
  5.     "log"
  6.     "time"
  8.     "go.mongodb.org/mongo-driver/bson"
  9.     "go.mongodb.org/mongo-driver/mongo"
  10.     "go.mongodb.org/mongo-driver/mongo/options"
  11.     "go.mongodb.org/mongo-driver/mongo/readpref"
  12. )
  14. type Student struct {
  15.     ID     string `bson:"_id,omitempty"`
  16.     Name   string `bson:"name"` // 名字
  17.     Gender string `bson:gender` // 性别
  18.     Age    int8   `bson:"age"`  // 年龄
  19. }
  21. func main() {
  22.     clientOptions := options.Client().SetHosts([]string{"localhost:27017"}).
  23.         SetConnectTimeout(5 * time.Second).
  24.         SetAuth(options.Credential{Username: "admin", Password: "654321"})
  25.     ctx := context.TODO()
  26.     client, err := mongo.Connect(ctx, clientOptions)
  27.     err = client.Ping(ctx, readpref.Primary())
  28.     if err != nil {
  29.         log.Fatal(err)
  30.     }
  31.     log.Println("mongodb connect success")
  33.     collection := client.Database("my_db").Collection("my_collection")
  34.     filter := bson.D{{"name", "张三"}}
  35.     res, err := collection.DeleteOne(ctx, filter)
  36.     if err != nil {
  37.         log.Fatal(err)
  38.     }
  39.     log.Println("delete count", res.DeletedCount)
  40. }