admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2025-12-31 16:05:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix recovery (#5417)

Browse Source 
Signed-off-by: sunby <bingyi.sun@zilliz.com>
This commit is contained in:
sunby 2021-05-26 12:04:38 +08:00 committed by zhenshan.cao
parent 81ac20143b
commit b9cf4c1095
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
internal/dataservice/cluster_session_manager.go
View File

@ -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
}

7
internal/dataservice/server_test.go
View File

@ -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())