mirror of https://github.com/milvus-io/milvus.git
parent
81ac20143b
commit
b9cf4c1095
internal/dataservice
|
@ -33,7 +33,10 @@ func newClusterSessionManager() *clusterSessionManager {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *clusterSessionManager) createSession(addr string) error {
|
func (m *clusterSessionManager) createSession(addr string) error {
|
||||||
cli := grpcdatanodeclient.NewClient(addr)
|
cli, err := grpcdatanodeclient.NewClient(addr, 0, []string{}, 0)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
if err := cli.Init(); err != nil {
|
if err := cli.Init(); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
|
@ -619,10 +619,11 @@ func TestDataNodeTtChannel(t *testing.T) {
|
||||||
})
|
})
|
||||||
|
|
||||||
ch := make(chan interface{}, 1)
|
ch := make(chan interface{}, 1)
|
||||||
svr.createDataNodeClient = func(addr string) types.DataNode {
|
svr.createDataNodeClient = func(addr string, serverID int64) (types.DataNode, error) {
|
||||||
cli := newMockDataNodeClient(0)
|
cli, err := newMockDataNodeClient(0)
|
||||||
|
assert.Nil(t, err)
|
||||||
cli.ch = ch
|
cli.ch = ch
|
||||||
return cli
|
return cli, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
ttMsgStream, err := svr.msFactory.NewMsgStream(context.TODO())
|
ttMsgStream, err := svr.msFactory.NewMsgStream(context.TODO())
|
||||||
|
|
Loading…
Reference in New Issue