mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-07 01:28:27 +08:00
38 lines
967 B
Go
38 lines
967 B
Go
package controller
|
|
|
|
import (
|
|
"strconv"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/zilliztech/milvus-distributed/internal/conf"
|
|
"github.com/zilliztech/milvus-distributed/internal/kv"
|
|
"github.com/zilliztech/milvus-distributed/internal/proto/internalpb"
|
|
"go.etcd.io/etcd/clientv3"
|
|
)
|
|
|
|
func newKvBase() *kv.EtcdKV {
|
|
//etcdAddr := conf.Config.Etcd.Address
|
|
//etcdAddr += ":"
|
|
//etcdAddr += strconv.FormatInt(int64(conf.Config.Etcd.Port), 10)
|
|
etcdAddr := "127.0.0.1:" + strconv.Itoa(int(conf.Config.Etcd.Port))
|
|
cli, _ := clientv3.New(clientv3.Config{
|
|
Endpoints: []string{etcdAddr},
|
|
DialTimeout: 5 * time.Second,
|
|
})
|
|
kvbase := kv.NewEtcdKV(cli, conf.Config.Etcd.Rootpath)
|
|
return kvbase
|
|
}
|
|
|
|
func TestComputeClosetTime(t *testing.T) {
|
|
kvbase := newKvBase()
|
|
var news internalpb.SegmentStats
|
|
for i := 0; i < 10; i++ {
|
|
news = internalpb.SegmentStats{
|
|
SegmentID: UniqueID(6875940398055133887),
|
|
MemorySize: int64(i * 1000),
|
|
}
|
|
ComputeCloseTime(news, kvbase)
|
|
}
|
|
}
|