monitor nm heart status

models/node_manager.go

@@ -79,7 +79,7 @@ type BenefitAddressInfo struct {
 
 type NodeManagerClient struct {
 	mutex         sync.Mutex
-	LastHeartTime int64
+	LastHeartTime time.Time
 	PublicKey     string
 	Endpoint      string
 	Client        nodeManagerV1.NodeManagerServiceClient
@@ -88,13 +88,13 @@ type NodeManagerClient struct {
 	IsSelected    bool
 }
 
-func (n *NodeManagerClient) GetLastHeartTime() int64 {
+func (n *NodeManagerClient) GetLastHeartTime() time.Time {
 	n.mutex.Lock()
 	defer n.mutex.Unlock()
 	return n.LastHeartTime
 }
 
-func (n *NodeManagerClient) UpdateLastHeartTime(time int64) {
+func (n *NodeManagerClient) UpdateLastHeartTime(time time.Time) {
 	n.mutex.Lock()
 	n.LastHeartTime = time
 	n.mutex.Unlock()

nm/start.go

@@ -195,7 +195,7 @@ func monitorWorker(op *operate.DockerOp) {
 
 				log.Info("------------------------Start rev msg worker thread------------------------")
 				for {
-					if (time.Now().UnixMilli()-nodeManager.GetLastHeartTime())/conf.GetConfig().HeartRespTimeMillis > conf.GetConfig().HeartRespTimeSecond {
+					if time.Now().Sub(nodeManager.GetLastHeartTime()) == time.Duration(conf.GetConfig().HeartRespTimeSecond) {
 						if nodeManager.Status {
 							nodeManager.UpdateStatus(false)
 						}
@@ -268,7 +268,7 @@ func handlerMsg(nodeManager *models.NodeManagerClient,
 				}
 				heartbeatReq := rev.GetHeartbeatRequest()
 				if heartbeatReq != nil {
-					nodeManager.UpdateLastHeartTime(int64(heartbeatReq.Timestamp))
+					nodeManager.UpdateLastHeartTime(time.Now())
 					params := buildParams(heartbeatReq.Timestamp)
 					msgRespWorker.RegisterMsgResp(nodeManager, worker, HeartbeatResp, params)
 					log.Info("-------------Heart beat req:-------------", heartbeatReq)