Fix recovery (#5417)

Signed-off-by: sunby <bingyi.sun@zilliz.com>
pull/5779/head
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

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
}

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