Skip to content

Instantly share code, notes, and snippets.

@zjhiphop
Forked from shellmanme/main.go
Created March 19, 2019 10:38
Show Gist options
  • Save zjhiphop/8bcc25d157d8cbe253feda01eb7f441c to your computer and use it in GitHub Desktop.
Save zjhiphop/8bcc25d157d8cbe253feda01eb7f441c to your computer and use it in GitHub Desktop.
mongo-go-driver Demo
package main
import (
"context"
"github.com/mongodb/mongo-go-driver/bson"
"github.com/mongodb/mongo-go-driver/bson/objectid"
"github.com/mongodb/mongo-go-driver/bson/primitive"
"github.com/mongodb/mongo-go-driver/mongo"
"log"
"time"
)
type User struct {
ID objectid.ObjectID "_id,omitempty"
Name string `bson:"dbname",json:"jsonname"`
Phone string
}
func main() {
log.SetFlags(log.Lshortfile)
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
client, err := mongo.Connect(ctx, "mongodb://127.0.0.1")
if err != nil {
log.Fatal(err)
}
db := client.Database("demo")
// Drop database
err = db.Drop(ctx)
if err != nil {
log.Fatal(err)
}
userColl := db.Collection("user")
// Insert one
if result, err := userColl.InsertOne(ctx, User{Name: "UserName", Phone: "1234567890"}); err == nil {
log.Println(result)
} else {
log.Fatal(err)
}
// Insert many
{
users := []interface{}{
User{Name: "UserName_0", Phone: "123"},
User{Name: "UserName_1", Phone: "456"},
User{Name: "UserName_2", Phone: "789"},
}
if result, err := userColl.InsertMany(ctx, users); err == nil {
log.Println(result)
} else {
log.Fatal(err)
}
}
// Find one
{
result := userColl.FindOne(ctx, bson.M{"phone": "1234567890"})
var user User
if err := result.Decode(&user); err != nil {
log.Fatal(err)
}
log.Println(user)
}
// Find
if cur, err := userColl.Find(ctx, bson.M{"phone": primitive.Regex{Pattern: "456", Options: ""}}); err == nil {
defer cur.Close(ctx)
for cur.Next(ctx) {
var user User
if err := cur.Decode(&user); err != nil {
log.Fatal(err)
}
log.Println(user)
}
} else {
log.Fatal(err)
}
// Update one
if result, err := userColl.UpdateOne(
ctx, bson.M{"phone": "123"},
bson.M{"$set": bson.M{"dbname": "UserName_changed"}}); err == nil {
log.Println(result)
} else {
log.Fatal(err)
}
// Update many
if result, err := userColl.UpdateMany(
ctx, bson.M{"phone": primitive.Regex{Pattern: "456", Options: ""}},
bson.M{"$set": bson.M{"dbname": "UserName_changed"}}); err == nil {
log.Println(result)
} else {
log.Fatal(err)
}
// Replace one
{
user := User{Name: "UserName_2_replaced", Phone: "789"}
if result, err := userColl.ReplaceOne(ctx, bson.M{"phone": "789"}, user); err == nil {
log.Println(result)
} else {
log.Fatal(err)
}
}
// Delete one
if result, err := userColl.DeleteOne(ctx, bson.M{"phone": "123"}); err == nil {
log.Println(result)
} else {
log.Fatal(err)
}
// Delete many
if result, err := userColl.DeleteMany(ctx, bson.M{"phone": primitive.Regex{Pattern: "456", Options: ""}}); err == nil {
log.Println(result)
} else {
log.Fatal(err)
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment