Fix proxy server

Signed-off-by: neza2017 <yefu.chen@zilliz.com>
This commit is contained in:
neza2017 2021-01-26 16:30:36 +08:00 committed by yefu.chen
parent fd0227b69a
commit b6a788373b
2 changed files with 9 additions and 1 deletions

View File

@ -49,6 +49,10 @@ func (c *Client) InvalidateCollectionMetaCache(request *proxypb.InvalidateCollMe
}
func (c *Client) GetTimeTickChannel() (string, error) {
err := c.tryConnect()
if err != nil {
return "", err
}
response, err := c.proxyServiceClient.GetTimeTickChannel(c.ctx, &commonpb.Empty{})
if err != nil {
return "", err
@ -57,6 +61,10 @@ func (c *Client) GetTimeTickChannel() (string, error) {
}
func (c *Client) GetComponentStates() (*internalpb2.ComponentStates, error) {
err := c.tryConnect()
if err != nil {
return nil, err
}
return c.proxyServiceClient.GetComponentStates(c.ctx, &commonpb.Empty{})
}

View File

@ -35,7 +35,7 @@ func (tt *TimeTickImpl) Start() error {
select {
case <-tt.ctx.Done():
log.Println("time tick loop was canceled by context!")
break
return
default:
current, err := tt.ttBarrier.GetTimeTick()
if err != nil {