You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
251 lines
6.1 KiB
251 lines
6.1 KiB
// Copyright © 2023 OpenIM. All rights reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package controller
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"math/rand"
|
|
"strconv"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/OpenIMSDK/tools/log"
|
|
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
|
|
"github.com/openimsdk/open-im-server/v3/pkg/common/config"
|
|
unrelationtb "github.com/openimsdk/open-im-server/v3/pkg/common/db/table/unrelation"
|
|
"github.com/openimsdk/open-im-server/v3/pkg/common/db/unrelation"
|
|
)
|
|
|
|
func Test_BatchInsertChat2DB(t *testing.T) {
|
|
conf := config.NewGlobalConfig()
|
|
conf.Mongo.Address = []string{"192.168.44.128:37017"}
|
|
// conf.Mongo.Timeout = 60
|
|
conf.Mongo.Database = "openIM"
|
|
// conf.Mongo.Source = "admin"
|
|
conf.Mongo.Username = "root"
|
|
conf.Mongo.Password = "openIM123"
|
|
conf.Mongo.MaxPoolSize = 100
|
|
conf.RetainChatRecords = 3650
|
|
conf.ChatRecordsClearTime = "0 2 * * 3"
|
|
|
|
mongo, err := unrelation.NewMongo(conf)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
err = mongo.GetDatabase(conf.Mongo.Database).Client().Ping(context.Background(), nil)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
db := &commonMsgDatabase{
|
|
msgDocDatabase: unrelation.NewMsgMongoDriver(mongo.GetDatabase(conf.Mongo.Database)),
|
|
}
|
|
|
|
//ctx := context.Background()
|
|
//msgs := make([]*sdkws.MsgData, 0, 1)
|
|
//for i := 0; i < cap(msgs); i++ {
|
|
// msgs = append(msgs, &sdkws.MsgData{
|
|
// Content: []byte(fmt.Sprintf("test-%d", i)),
|
|
// SendTime: time.Now().UnixMilli(),
|
|
// })
|
|
//}
|
|
//err = db.BatchInsertChat2DB(ctx, "test", msgs, 0)
|
|
//if err != nil {
|
|
// panic(err)
|
|
//}
|
|
|
|
_ = db.BatchInsertChat2DB
|
|
c := mongo.GetDatabase(conf.Mongo.Database).Collection("msg")
|
|
|
|
ch := make(chan int)
|
|
rand.Seed(time.Now().UnixNano())
|
|
|
|
index := 10
|
|
|
|
var wg sync.WaitGroup
|
|
for i := 0; i < 1000; i++ {
|
|
wg.Add(1)
|
|
go func(channelID int) {
|
|
defer wg.Done()
|
|
<-ch
|
|
var arr []string
|
|
for i := 0; i < 500; i++ {
|
|
arr = append(arr, strconv.Itoa(i+1))
|
|
}
|
|
rand.Shuffle(len(arr), func(i, j int) {
|
|
arr[i], arr[j] = arr[j], arr[i]
|
|
})
|
|
for j, s := range arr {
|
|
if j == 0 {
|
|
fmt.Printf("channnelID: %d, arr[0]: %s\n", channelID, arr[j])
|
|
}
|
|
filter := bson.M{"doc_id": "test:0"}
|
|
update := bson.M{
|
|
"$addToSet": bson.M{
|
|
fmt.Sprintf("msgs.%d.del_list", index): bson.M{"$each": []string{s}},
|
|
},
|
|
}
|
|
_, err := c.UpdateOne(context.Background(), filter, update)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
}(i)
|
|
}
|
|
|
|
for i := 0; i < 1000; i++ {
|
|
wg.Add(1)
|
|
go func() {
|
|
defer wg.Done()
|
|
<-ch
|
|
var arr []string
|
|
for i := 0; i < 500; i++ {
|
|
arr = append(arr, strconv.Itoa(1001+i))
|
|
}
|
|
rand.Shuffle(len(arr), func(i, j int) {
|
|
arr[i], arr[j] = arr[j], arr[i]
|
|
})
|
|
for _, s := range arr {
|
|
filter := bson.M{"doc_id": "test:0"}
|
|
update := bson.M{
|
|
"$addToSet": bson.M{
|
|
fmt.Sprintf("msgs.%d.read_list", index): bson.M{"$each": []string{s}},
|
|
},
|
|
}
|
|
_, err := c.UpdateOne(context.Background(), filter, update)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
}()
|
|
}
|
|
|
|
time.Sleep(time.Second * 2)
|
|
|
|
close(ch)
|
|
|
|
wg.Wait()
|
|
|
|
}
|
|
|
|
func GetDB() *commonMsgDatabase {
|
|
conf := config.NewGlobalConfig()
|
|
conf.Mongo.Address = []string{"203.56.175.233:37017"}
|
|
// conf.Mongo.Timeout = 60
|
|
conf.Mongo.Database = "openim_v3"
|
|
// conf.Mongo.Source = "admin"
|
|
conf.Mongo.Username = "root"
|
|
conf.Mongo.Password = "openIM123"
|
|
conf.Mongo.MaxPoolSize = 100
|
|
conf.RetainChatRecords = 3650
|
|
conf.ChatRecordsClearTime = "0 2 * * 3"
|
|
|
|
mongo, err := unrelation.NewMongo(conf)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
err = mongo.GetDatabase(conf.Mongo.Database).Client().Ping(context.Background(), nil)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return &commonMsgDatabase{
|
|
msgDocDatabase: unrelation.NewMsgMongoDriver(mongo.GetDatabase(conf.Mongo.Database)),
|
|
}
|
|
}
|
|
|
|
func Test_Insert(t *testing.T) {
|
|
db := GetDB()
|
|
ctx := context.Background()
|
|
var arr []any
|
|
for i := 0; i < 345; i++ {
|
|
if i%2 == 0 {
|
|
arr = append(arr, (*unrelationtb.MsgDataModel)(nil))
|
|
continue
|
|
}
|
|
arr = append(arr, &unrelationtb.MsgDataModel{
|
|
Seq: int64(i),
|
|
Content: fmt.Sprintf("test-%d", i),
|
|
})
|
|
}
|
|
if err := db.BatchInsertBlock(ctx, "test", arr, updateKeyMsg, 1); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func Test_Revoke(t *testing.T) {
|
|
db := GetDB()
|
|
ctx := context.Background()
|
|
var arr []any
|
|
for i := 0; i < 456; i++ {
|
|
arr = append(arr, &unrelationtb.RevokeModel{
|
|
UserID: "uid_" + strconv.Itoa(i),
|
|
Nickname: "uname_" + strconv.Itoa(i),
|
|
Time: time.Now().UnixMilli(),
|
|
})
|
|
}
|
|
if err := db.BatchInsertBlock(ctx, "test", arr, updateKeyRevoke, 123); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func Test_FindBySeq(t *testing.T) {
|
|
if err := log.InitFromConfig("", "", 6, true, false, "", 2, 1); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
db := GetDB()
|
|
ctx := context.Background()
|
|
fmt.Println(
|
|
db.msgDocDatabase.(*unrelation.MsgMongoDriver).GetMsgBySeqIndexIn1Doc(ctx, "100", "si_100_101:0", []int64{1}),
|
|
)
|
|
//res, err := db.msgDocDatabase.GetMsgBySeqIndexIn1Doc(ctx, "123456", "test:0", []int64{1, 2, 3})
|
|
//if err != nil {
|
|
// t.Fatal(err)
|
|
//}
|
|
//db.GetMsgBySeqs(ctx, "100", "si_100_101:0", []int64{6})
|
|
//data, _ := json.Marshal(res)
|
|
//fmt.Println(string(data))
|
|
}
|
|
|
|
//func Test_Delete(t *testing.T) {
|
|
// db := GetDB()
|
|
// ctx := context.Background()
|
|
// var arr []any
|
|
// for i := 0; i < 123; i++ {
|
|
// arr = append(arr, []string{"uid_1", "uid_2"})
|
|
// }
|
|
// if err := db.BatchInsertBlock(ctx, "test", arr, updateKeyDel, 210); err != nil {
|
|
// t.Fatal(err)
|
|
// }
|
|
//}
|
|
|
|
func TestName(t *testing.T) {
|
|
db := GetDB()
|
|
var seqs []int64
|
|
for i := int64(1); i <= 50; i++ {
|
|
seqs = append(seqs, i)
|
|
}
|
|
msgs, err := db.getMsgBySeqsRange(context.Background(), "4931176757", "si_3866692501_4931176757", seqs, seqs[0], seqs[len(seqs)-1])
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
t.Log(msgs)
|
|
|
|
}
|