84 lines
2.0 KiB
Go
84 lines
2.0 KiB
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"go_dreamfactory/comm"
|
|
"go_dreamfactory/lego/core"
|
|
"go_dreamfactory/lego/sys/mgo"
|
|
"go_dreamfactory/pb"
|
|
"math/rand"
|
|
"sync"
|
|
"testing"
|
|
|
|
"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 Test_Id(t *testing.T) {
|
|
wg := new(sync.WaitGroup)
|
|
for i := 0; i < 3; i++ {
|
|
wg.Add(1)
|
|
go func(_wg *sync.WaitGroup) {
|
|
for i := 0; i < 10; i++ {
|
|
id := primitive.NewObjectID().Hex()
|
|
fmt.Printf(id + "\n")
|
|
}
|
|
_wg.Done()
|
|
}(wg)
|
|
}
|
|
wg.Wait()
|
|
}
|
|
|
|
func Test_2D(t *testing.T) {
|
|
if sys, err := mgo.NewSys(
|
|
mgo.SetMongodbUrl("mongodb://10.0.0.9:10013"),
|
|
mgo.SetMongodbDatabase("dreamfactory2"),
|
|
); err != nil {
|
|
fmt.Printf("err:%v", err)
|
|
return
|
|
} else {
|
|
//创建uid索引
|
|
_index, err := sys.CreateIndex(core.SqlTable(comm.TableArena), mongo.IndexModel{
|
|
Keys: bson.M{"loc": "2dsphere"},
|
|
})
|
|
fmt.Printf("_index:%v err:%v \n", _index, err)
|
|
port := []float64{2, float64(rand.Int31n(100)) / 1000.0}
|
|
fmt.Printf("port:%v \n", port)
|
|
if cursor, err := sys.Find(comm.TableArena, bson.M{
|
|
"dan": 2,
|
|
// "loc": bson.M{
|
|
// "$geoWithin": bson.M{
|
|
// "$center": bson.A{port, 100},
|
|
// },
|
|
"loc": bson.M{
|
|
"$near": bson.M{
|
|
"$geometry": bson.M{"type": "Point", "coordinates": port},
|
|
"$maxDistance": 100000,
|
|
},
|
|
},
|
|
// bson.A{[]int32{2 * 1000, rand.Int31n(100)}, 100}
|
|
// "loc": bson.M{
|
|
// "$near": bson.M{
|
|
// "$geometry": bson.M{"type": "Point", "coordinates": []int32{dan * 1000, rand.Int31n(100)}},
|
|
// "$maxDistance": 100,
|
|
// },
|
|
// },
|
|
}, options.Find().SetSkip(0).SetLimit(int64(5))); err != nil {
|
|
fmt.Printf("err:%v", err)
|
|
return
|
|
} else {
|
|
for cursor.Next(context.Background()) {
|
|
temp := &pb.DBArenaUser{}
|
|
if err = cursor.Decode(temp); err != nil {
|
|
fmt.Printf("err:%v", err)
|
|
return
|
|
}
|
|
fmt.Printf("temp:%v\n", temp.Uid)
|
|
}
|
|
}
|
|
}
|
|
}
|