add msg resp

controllers/NodeController.go

@@ -64,8 +64,10 @@ func (c *NodeController) SetBenefitAddress() {
 		err = utils.WriteBenefitFile(nm.HistoryBenefitAcc)
 		if err != nil {
 			c.ResponseInfo(500, "Write benefit file failed", "")
+			return
 		}
 	}
+	nm.IsUpdateBenefitAddr = true
 	c.ResponseInfo(200, "set benefit address successful", "")
 }
 

nm/api.go

@@ -13,13 +13,15 @@ type NodeManager struct {
 }
 
 var (
-	HistoryBenefitAcc []*models.BenefitAddressStruct
-	RunningState      *models.RunningState
-	IsRecvTask        bool
+	HistoryBenefitAcc   []*models.BenefitAddressStruct
+	RunningState        *models.RunningState
+	IsRecvTask          bool
+	IsUpdateBenefitAddr bool
 )
 
 func init() {
 	IsRecvTask = false
+	IsUpdateBenefitAddr = false
 	HistoryBenefitAcc = make([]*models.BenefitAddressStruct, 0)
 	RunningState = &models.RunningState{
 		RunningTime:        time.Now().Unix(),

nm/monitor.go

@@ -81,10 +81,15 @@ func (m *MonitorNm) monitorNmClient() {
 				for {
 					if !IsRecvTask {
 						log.Warn("User set recv task status is false")
-						msgRespWorker.RegisterMsgResp(nodeManager, worker, RegisterInfoResp, nil)
+						msgRespWorker.RegisterMsgResp(nodeManager, worker, GoodbyeResp, nil)
 						nodeManager.UpdateStatus(false)
 						return
 					}
+					if IsUpdateBenefitAddr {
+						benefitAddrUpdateRespParam := utils.BuildParams(conf.GetConfig().BenefitAddress)
+						msgRespWorker.RegisterMsgResp(nodeManager, worker, BenefitAddrUpdateResp, benefitAddrUpdateRespParam)
+						IsUpdateBenefitAddr = false
+					}
 					sub := time.Now().Sub(nodeManager.GetLastHeartTime()).Seconds()
 					log.WithField("time(uint seconds)", sub).Info("Handler nm msg thread monitor heartbeat time")
 					rev, err := worker.Recv()

nm/msg_resp.go

@@ -344,6 +344,20 @@ func AddModelRunningResp(params ...interface{}) *nodemanagerV2.WorkerMessage {
 	return addModelRunningRes
 }
 
+func BenefitAddrUpdateResp(params ...interface{}) *nodemanagerV2.WorkerMessage {
+	log.Info("Benefit addr update response received params:", params)
+	addr := params[0].(string)
+	benefitAddrUpdateResp := &nodemanagerV2.WorkerMessage{
+		Message: &nodemanagerV2.WorkerMessage_BenefitAddrUpdate{
+			BenefitAddrUpdate: &nodemanagerV2.BenefitAddrUpdate{
+				BenefitAddress: addr,
+			},
+		},
+	}
+	log.Info("---------------------------------------Send Benefit addr update response msg ------------------------------------")
+	return benefitAddrUpdateResp
+}
+
 func RunningModelStatusResp(params ...interface{}) *nodemanagerV2.WorkerMessage {
 	log.Info("Add model running response received params:", params)
 	info := params[0].(*models.ModelInfo)