215 lines
6.3 KiB
Go
215 lines
6.3 KiB
Go
package mgo_test
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"testing"
|
||
|
||
"go_dreamfactory/lego/core"
|
||
"go_dreamfactory/lego/sys/mgo"
|
||
|
||
"go.mongodb.org/mongo-driver/bson"
|
||
"go.mongodb.org/mongo-driver/mongo"
|
||
"go.mongodb.org/mongo-driver/mongo/options"
|
||
"go.mongodb.org/mongo-driver/x/bsonx"
|
||
)
|
||
|
||
const (
|
||
Sql_UserDataTable core.SqlTable = "userdata" //用户
|
||
)
|
||
|
||
type TestData struct {
|
||
Id uint32 `bson:"_id"` //用户id
|
||
NiceName string //昵称
|
||
Sex int32 //性别 1男 2女
|
||
HeadUrl string //头像Id
|
||
}
|
||
|
||
//测试 系统 直连模式
|
||
func Test_sys(t *testing.T) {
|
||
if err := mgo.OnInit(map[string]interface{}{
|
||
"MongodbUrl": "mongodb://127.0.0.1:37017",
|
||
"MongodbDatabase": "testdb",
|
||
}); err == nil {
|
||
if _, err := mgo.UpdateOne(Sql_UserDataTable,
|
||
bson.M{"_id": 10001},
|
||
bson.M{"$set": bson.M{
|
||
"nicename": "liwei1dao",
|
||
"headurl": "http://test.web.com",
|
||
"sex": 1,
|
||
}},
|
||
new(options.UpdateOptions).SetUpsert(true)); err != nil {
|
||
fmt.Printf("UpdateOne errr:%v", err)
|
||
}
|
||
|
||
data := &TestData{}
|
||
if err := mgo.FindOne(Sql_UserDataTable, bson.M{"_id": 10002}).Decode(data); err != nil {
|
||
fmt.Printf("FindOne errr:%v", err)
|
||
} else {
|
||
fmt.Printf("FindOne data:%+v", data)
|
||
}
|
||
} else {
|
||
fmt.Printf("OnInit errr:%v", err)
|
||
}
|
||
}
|
||
|
||
//测试 事务 副本集模式
|
||
func Test_Affair(t *testing.T) {
|
||
if err := mgo.OnInit(map[string]interface{}{
|
||
"MongodbUrl": "mongodb://root:123@127.0.0.1:37017,127.0.0.1:37018,127.0.0.1:37019/admin?replicaSet=mongoReplSet",
|
||
"MongodbDatabase": "testdb",
|
||
}); err == nil {
|
||
err = mgo.UseSession(func(sessionContext mongo.SessionContext) error {
|
||
err := sessionContext.StartTransaction()
|
||
if err != nil {
|
||
fmt.Println(err)
|
||
return err
|
||
}
|
||
col := mgo.Collection(Sql_UserDataTable)
|
||
|
||
//在事务内写一条id为“222”的记录
|
||
_, err = col.InsertOne(sessionContext, bson.M{"_id": 10004, "nicename": "liwei2dao", "headurl": "http://test1.web.com", "sex": 2})
|
||
if err != nil {
|
||
fmt.Println(err)
|
||
return err
|
||
}
|
||
//在事务内写一条id为“333”的记录
|
||
_, err = col.InsertOne(sessionContext, bson.M{"_id": 10005, "nicename": "liwei3dao", "headurl": "http://test1.web.com", "sex": 2})
|
||
if err != nil {
|
||
sessionContext.AbortTransaction(sessionContext)
|
||
return err
|
||
} else {
|
||
sessionContext.CommitTransaction(sessionContext)
|
||
}
|
||
return nil
|
||
})
|
||
fmt.Printf("FindOne errr:%v", err)
|
||
} else {
|
||
fmt.Printf("FindOne errr:%v", err)
|
||
}
|
||
}
|
||
|
||
//测试创建索引 过期索引
|
||
func Test_CreateTTLIndex(t *testing.T) {
|
||
sys, err := mgo.NewSys(mgo.SetMongodbUrl("mongodb://47.90.84.157:9094"), mgo.SetMongodbDatabase("square"))
|
||
if err != nil {
|
||
fmt.Printf("start sys Fail err:%v", err)
|
||
return
|
||
}
|
||
|
||
indexModel := mongo.IndexModel{
|
||
Keys: bsonx.Doc{{"expire_date", bsonx.Int32(1)}}, // 设置TTL索引列"expire_date"
|
||
Options: options.Index().SetExpireAfterSeconds(1 * 24 * 3600), // 设置过期时间1天，即，条目过期一天过自动删除
|
||
}
|
||
str, err := sys.CreateIndex(core.SqlTable("dynamics"), indexModel)
|
||
if err != nil {
|
||
fmt.Printf("CreateIndex err:%v", err)
|
||
} else {
|
||
fmt.Printf("CreateIndex str:%v", str)
|
||
}
|
||
}
|
||
|
||
//测试创建索引 地图索引
|
||
func Test_CreateIndex(t *testing.T) {
|
||
sys, err := mgo.NewSys(mgo.SetMongodbUrl("mongodb://47.90.84.157:9094"), mgo.SetMongodbDatabase("square"))
|
||
if err != nil {
|
||
fmt.Printf("start sys Fail err:%v", err)
|
||
return
|
||
}
|
||
|
||
indexModel := mongo.IndexModel{
|
||
Keys: bson.M{"location": "2dsphere"},
|
||
Options: options.Index().SetBits(11111),
|
||
}
|
||
str, err := sys.CreateIndex(core.SqlTable("dynamics"), indexModel)
|
||
if err != nil {
|
||
fmt.Printf("CreateIndex err:%v", err)
|
||
} else {
|
||
fmt.Printf("CreateIndex str:%v", str)
|
||
}
|
||
}
|
||
|
||
//创建复合索引
|
||
func Test_CreateCompoundIndex(t *testing.T) {
|
||
sys, err := mgo.NewSys(mgo.SetMongodbUrl("mongodb://47.90.84.157:9094"), mgo.SetMongodbDatabase("lego_yl"))
|
||
if err != nil {
|
||
fmt.Printf("start sys Fail err:%v", err)
|
||
return
|
||
}
|
||
str, err := sys.CreateIndex(core.SqlTable("unreadmsg"), mongo.IndexModel{Keys: bson.M{"channeltype": 1, "targetid": 1, "uid": 1, "unreadmsg.sendtime": 1}}, nil)
|
||
if err != nil {
|
||
fmt.Printf("CreateIndex err:%v", err)
|
||
} else {
|
||
fmt.Printf("CreateIndex str:%v", str)
|
||
}
|
||
}
|
||
|
||
//测试删除索引 地图索引
|
||
func Test_DeleteIndex(t *testing.T) {
|
||
sys, err := mgo.NewSys(mgo.SetMongodbUrl("mongodb://47.90.84.157:9094"), mgo.SetMongodbDatabase("square"))
|
||
if err != nil {
|
||
fmt.Printf("start sys Fail err:%v", err)
|
||
return
|
||
}
|
||
_, err = sys.DeleteIndex(core.SqlTable("dynamics"), "location_2dsphere", nil)
|
||
if err != nil {
|
||
fmt.Printf("DeleteIndex err:%v", err)
|
||
} else {
|
||
fmt.Printf("DeleteIndex succ")
|
||
}
|
||
}
|
||
|
||
//测试附近查找
|
||
/*
|
||
db.<collection>.find( { <location field> :
|
||
{ $near :
|
||
{ $geometry :
|
||
{ type : "Point" ,coordinates : [ <longitude> , <latitude> ] } ,
|
||
$maxDistance : <distance in meters>
|
||
}
|
||
}
|
||
})
|
||
*/
|
||
func Test_QueryRound(t *testing.T) {
|
||
sys, err := mgo.NewSys(mgo.SetMongodbUrl("mongodb://47.90.84.157:9094"), mgo.SetMongodbDatabase("square"))
|
||
if err != nil {
|
||
fmt.Printf("start sys Fail err:%v", err)
|
||
return
|
||
}
|
||
filter := bson.M{
|
||
"location": bson.D{
|
||
{"$near", bson.D{
|
||
{"$geometry", bson.M{"type": "Point", "coordinates": []float64{113.867584, 22.56532}}},
|
||
{"$maxDistance", 10000},
|
||
}},
|
||
},
|
||
}
|
||
_, err = sys.Find(core.SqlTable("dynamics"), filter)
|
||
if err != nil {
|
||
fmt.Printf("CreateIndex err:%v", err)
|
||
} else {
|
||
fmt.Printf("CreateIndex succ")
|
||
}
|
||
return
|
||
}
|
||
|
||
//聚合查询演示
|
||
func Test_Aggregate(t *testing.T) {
|
||
sys, err := mgo.NewSys(mgo.SetMongodbUrl("mongodb://47.90.84.157:9094"), mgo.SetMongodbDatabase("square"))
|
||
if err != nil {
|
||
fmt.Printf("start sys Fail err:%v", err)
|
||
return
|
||
}
|
||
result := make([]bson.M, 0)
|
||
var cursor *mongo.Cursor
|
||
matchStage := bson.D{{"$match", bson.M{"_id": 979332}}}
|
||
unwindStage := bson.D{{"$unwind", "$chatmsg"}}
|
||
match2Stage := bson.D{{"$match", bson.M{"chatmsg.sendtime": bson.M{"$gt": 1611890940100, "$lt": 1611893135929}}}}
|
||
groupStage := bson.D{{"$group", bson.M{"_id": 979332, "chatmsg": bson.M{"$push": "$chatmsg"}}}}
|
||
projectStage := bson.D{{"$project", bson.M{"_id": 1, "chatmsg": 1}}}
|
||
if cursor, err = sys.Aggregate(core.SqlTable("grouprecorddata"), mongo.Pipeline{matchStage, unwindStage, match2Stage, groupStage, projectStage}); err == nil {
|
||
err = cursor.All(context.Background(), &result)
|
||
}
|
||
return
|
||
}
|