Commit 694ee11f authored by duanjinfei's avatar duanjinfei

update nm is conn failed handler

parent b27765f2
......@@ -85,6 +85,9 @@ func reportModelInfo(nodeManager *models.NodeManagerClient,
for {
select {
case taskId := <-dockerOp.ModelTaskIdChan:
if !nodeManager.Status {
return
}
params := buildParams(taskId)
msgRespWorker.RegisterMsgResp(nodeManager, worker, SubmitResourceMapRes, params)
}
......
......@@ -47,7 +47,6 @@ func (o *RespMsgWorker) SendMsg() {
err := pool.workerClient.SendMsg(workerMsg)
if err != nil {
log.Error("Send heartbeat msg error:", err)
pool.nodeManager.UpdateStatus(false)
return
}
log.Info("Worker client send message successfully")
......
......@@ -156,9 +156,11 @@ func monitorWorker(op *operate.DockerOp) {
rev, err := worker.Recv()
if err == io.EOF {
log.Errorf("Node manage not work endpoint:%s", nodeManager.Endpoint)
nodeManager.UpdateStatus(false)
params := buildParams(fmt.Sprintf("Node manage not work endpoint:%s", nodeManager.Endpoint))
msgRespWorker.RegisterMsgResp(nodeManager, worker, GoodbyeResp, params)
if nodeManager.Status {
nodeManager.UpdateStatus(false)
}
return
}
if err != nil {
......@@ -187,6 +189,9 @@ func handlerMsg(nodeManager *models.NodeManagerClient,
select {
case rev := <-nodeManagerMsgChan:
{
if !nodeManager.Status {
return
}
heartbeatReq := rev.GetHeartbeatRequest()
if heartbeatReq != nil {
nodeManager.UpdateLastHeartTime(int64(heartbeatReq.Timestamp))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment