mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-01 03:18:29 +08:00
58 lines
1.7 KiB
Go
58 lines
1.7 KiB
Go
package test
|
|
|
|
import (
|
|
//"context"
|
|
//"sync"
|
|
"testing"
|
|
|
|
"github.com/milvus-io/milvus/internal/util/typeutil"
|
|
)
|
|
|
|
type Timestamp = typeutil.Timestamp
|
|
|
|
//func GetInsertMsg(collectionName string, partitionTag string, entityId UniqueID) *msgpb.InsertOrDeleteMsg {
|
|
// return &msgpb.InsertOrDeleteMsg{
|
|
// CollectionName: collectionName,
|
|
// PartitionTag: partitionTag,
|
|
// SegmentID: UniqueID(entityId / 100),
|
|
// Uid: UniqueID(entityId),
|
|
// Timestamp: Timestamp(entityId),
|
|
// ClientId: 0,
|
|
// }
|
|
//}
|
|
//
|
|
//func GetDeleteMsg(collectionName string, entityId UniqueID) *msgpb.InsertOrDeleteMsg {
|
|
// return &msgpb.InsertOrDeleteMsg{
|
|
// CollectionName: collectionName,
|
|
// Uid: entityId,
|
|
// Timestamp: Timestamp(entityId + 100),
|
|
// }
|
|
//}
|
|
|
|
func TestInsert(t *testing.T) {
|
|
// TODO: fix test
|
|
//return
|
|
//ctx := context.Background()
|
|
//var topics []string
|
|
//topics = append(topics, "test")
|
|
//topics = append(topics, "test1")
|
|
//writerNode, _ := writer.NewWriteNode(ctx, "null", topics, 0)
|
|
//var insertMsgs []*msgpb.InsertOrDeleteMsg
|
|
//for i := 0; i < 120; i++ {
|
|
// insertMsgs = append(insertMsgs, GetInsertMsg("collection0", "tag01", UniqueID(i)))
|
|
//}
|
|
//wg := sync.WaitGroup{}
|
|
//wg.Add(3)
|
|
////var wg sync.WaitGroup
|
|
//writerNode.InsertBatchData(ctx, insertMsgs, &wg)
|
|
//var insertMsgs2 []*msgpb.InsertOrDeleteMsg
|
|
//for i := 120; i < 200; i++ {
|
|
// insertMsgs2 = append(insertMsgs2, GetInsertMsg("collection0", "tag02", UniqueID(i)))
|
|
//}
|
|
//writerNode.InsertBatchData(ctx, insertMsgs2, &wg)
|
|
//var deleteMsgs []*msgpb.InsertOrDeleteMsg
|
|
//deleteMsgs = append(deleteMsgs, GetDeleteMsg("collection0", 2))
|
|
//deleteMsgs = append(deleteMsgs, GetDeleteMsg("collection0", 120))
|
|
//writerNode.DeleteBatchData(ctx, deleteMsgs)
|
|
}
|