mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-05 21:39:19 +08:00
6a722396bd
Signed-off-by: wayblink <anyang.wang@zilliz.com>
42 lines
763 B
Go
42 lines
763 B
Go
package indexnode
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/milvus-io/milvus/internal/types"
|
|
)
|
|
|
|
type mockIndexNodeComponent struct {
|
|
IndexNode
|
|
}
|
|
|
|
var _ types.IndexNodeComponent = &mockIndexNodeComponent{}
|
|
|
|
func NewMockIndexNodeComponent(ctx context.Context) (types.IndexNodeComponent, error) {
|
|
Params.Init()
|
|
factory := &mockFactory{
|
|
chunkMgr: &mockChunkmgr{},
|
|
}
|
|
|
|
node := NewIndexNode(ctx, factory)
|
|
|
|
startEmbedEtcd()
|
|
etcdCli := getEtcdClient()
|
|
node.SetEtcdClient(etcdCli)
|
|
node.storageFactory = &mockStorageFactory{}
|
|
if err := node.Init(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err := node.Start(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err := node.Register(); err != nil {
|
|
return nil, err
|
|
}
|
|
return &mockIndexNodeComponent{
|
|
IndexNode: *node,
|
|
}, nil
|
|
}
|