Fix proxy server

Signed-off-by: neza2017 <yefu.chen@zilliz.com>

internal/distributed/proxyservice/client.go

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

internal/proxyservice/timetick.go

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