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