271 lines
6.4 KiB
Go
271 lines
6.4 KiB
Go
package mongo
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"testing"
|
||
"time"
|
||
|
||
"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"
|
||
)
|
||
|
||
func init() {
|
||
mongo_uri = "mongodb://192.168.0.254:27017"
|
||
mongo_database = "sample"
|
||
mongo_username = "aaa"
|
||
mongo_password = "bbb123"
|
||
}
|
||
|
||
func TestPing(t *testing.T) {
|
||
if err := ping(); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
}
|
||
|
||
type Student struct {
|
||
ID primitive.ObjectID `bson:"_id,omitempty" json:"id,omitempty"` //确保在 BSON 编码/解码时，当 _id 为空时不会将其包含在文档中。在插入文档时，可以不设置 ID 字段，MongoDB 驱动会自动为其生成一个唯一的 ObjectID。
|
||
Name string `json:"name,omitempty"`
|
||
Sex string `json:"sex,omitempty"`
|
||
CourseID int `bson:"courseid,omitempty" json:"course_id,omitempty"`
|
||
StudentID int `json:"student_id,omitempty"`
|
||
}
|
||
|
||
func TestInsert(t *testing.T) {
|
||
client, err := newConnect()
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
defer client.Disconnect(context.Background())
|
||
|
||
for i := 0; i < 10000; i++ {
|
||
collection, err := newCollection(client, "", "student")
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
n := fmt.Sprintf("张小凡%d", i)
|
||
ret, err := collection.InsertOne(context.Background(), &Student{Name: n, Sex: "女", StudentID: i, CourseID: i % 10})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
fmt.Println(i, ret)
|
||
fmt.Println(ret.InsertedID.(primitive.ObjectID).Hex())
|
||
}
|
||
}
|
||
|
||
func TestInsertCourse(t *testing.T) {
|
||
client, err := newConnect()
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
defer client.Disconnect(context.Background())
|
||
|
||
for i := 2; i < 10; i++ {
|
||
collection, err := newCollection(client, "", "course")
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
n := fmt.Sprintf("sports%d", i)
|
||
ret, err := collection.InsertOne(context.Background(), &Course{ID: i, Name: n, Title: "sports"})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
fmt.Println(i, ret)
|
||
}
|
||
}
|
||
|
||
func TestDelete(t *testing.T) {
|
||
client, err := newConnect()
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
defer client.Disconnect(context.Background())
|
||
col, err := newCollection(client, "", "student")
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
if err := col.FindOneAndDelete(context.Background(), bson.D{{"name", "张惠"}}).Err(); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
ret, err := col.DeleteOne(context.Background(), bson.D{{"name", "张惠"}})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
fmt.Println(ret)
|
||
ret, err = col.DeleteMany(context.Background(), bson.D{{"name", "张惠"}})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
fmt.Println(ret)
|
||
}
|
||
|
||
func TestSingleResult(t *testing.T) {
|
||
client, err := newConnect()
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
|
||
defer cancel()
|
||
defer client.Disconnect(ctx)
|
||
col, err := Collection(client, "student")
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
var result Student
|
||
// if err := col.FindOne(ctx, bson.D{{"name", "张惠"}}).Decode(&result); err != nil {
|
||
if err := col.FindOne(ctx, bson.M{"name": "张惠0"}).Decode(&result); err != nil {
|
||
if err != mongo.ErrNoDocuments {
|
||
t.Fatal(err)
|
||
}
|
||
}
|
||
fmt.Println(result)
|
||
}
|
||
|
||
func TestFindAll(t *testing.T) {
|
||
client, err := newConnect()
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
|
||
defer cancel()
|
||
defer client.Disconnect(ctx)
|
||
col, err := newCollection(client, "", "student")
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
cur, err := col.Find(ctx, bson.D{})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
defer cur.Close(ctx)
|
||
for cur.Next(ctx) {
|
||
var result Student
|
||
if err := cur.Decode(&result); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
fmt.Println(result, result.ID.String(), result.ID.Timestamp(), result.ID.Timestamp().Local())
|
||
}
|
||
if err := cur.Err(); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
}
|
||
|
||
func TestFind(t *testing.T) {
|
||
client, err := newConnect()
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
|
||
defer cancel()
|
||
defer client.Disconnect(ctx)
|
||
col, err := newDefaultCollection(client, "student")
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
cur, err := col.Find(ctx, bson.M{"name": "张小凡1"}) //注意这里的格式$ne不相等,$gt大于,$gte大于等于,$in in,$nin no in ,$exists是否包含这个键...
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
defer cur.Close(ctx)
|
||
for cur.Next(ctx) {
|
||
var result Student
|
||
if err := cur.Decode(&result); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
fmt.Println(result, result.ID.String(), result.ID.Timestamp(), result.ID.Timestamp().Local())
|
||
}
|
||
if err := cur.Err(); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
}
|
||
|
||
func TestPage(t *testing.T) {
|
||
client, err := newConnect()
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
|
||
defer cancel()
|
||
defer client.Disconnect(ctx)
|
||
col, err := newDefaultCollection(client, "student")
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
cur, err := col.Find(ctx, bson.M{}, options.Find().SetSkip(1000).SetLimit(1000))
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
defer cur.Close(ctx)
|
||
for cur.Next(ctx) {
|
||
var result Student
|
||
if err := cur.Decode(&result); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
fmt.Println(result, result.ID.String(), result.ID.Timestamp(), result.ID.Timestamp().Local())
|
||
}
|
||
if err := cur.Err(); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
}
|
||
|
||
type Course struct {
|
||
ID int `bson:"_id,omitempty" json:"id,omitempty"`
|
||
Name string `json:"name,omitempty"`
|
||
Title string `json:"title,omitempty"`
|
||
}
|
||
|
||
// 关连表查询
|
||
func TestAggregate(t *testing.T) {
|
||
// 建立 MongoDB 连接
|
||
client, err := newConnect()
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
defer client.Disconnect(context.Background())
|
||
|
||
// 选择数据库和集合
|
||
studentsCollection, err := newDefaultCollection(client, "student")
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
// 执行关联查询
|
||
pipeline := bson.A{
|
||
bson.D{
|
||
{"$match", bson.D{
|
||
{"name", "张小凡1"},
|
||
}},
|
||
},
|
||
bson.D{
|
||
{"$lookup", bson.D{
|
||
{"from", "course"},
|
||
{"localField", "courseid"},
|
||
{"foreignField", "_id"},
|
||
{"as", "courses"},
|
||
}},
|
||
},
|
||
}
|
||
|
||
// 执行聚合查询，结果OK
|
||
cursor, err := studentsCollection.Aggregate(context.Background(), pipeline)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
defer cursor.Close(context.Background())
|
||
|
||
// 解码查询结果
|
||
var result []bson.M
|
||
if err := cursor.All(context.Background(), &result); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
// 打印查询结果
|
||
fmt.Println("Result:")
|
||
for _, doc := range result {
|
||
fmt.Println(doc)
|
||
}
|
||
}
|