fix bug

cmd/demoworker/main.go

@@ -75,6 +75,34 @@ func withManager(idx int, endpoint string) {
 		}
 		managerClient.Send(resource)
 	}
+	taskCh := make(chan *omanager.ManagerMessage_PushTaskMessage, 1000)
+	go func() {
+		for {
+			select {
+			case task, ok := <-taskCh:
+				if !ok {
+					return
+				}
+				demoResult := "a demo task result return"
+				msg := &omanager.WorkerMessage{
+					Message: &omanager.WorkerMessage_SubmitTaskResult{
+						SubmitTaskResult: &omanager.SubmitTaskResult{
+							TaskId:              task.PushTaskMessage.TaskId,
+							ContainerSignature:  make([]byte, 65),
+							MinerSignature:      make([]byte, 65),
+							TaskResultBody:      []byte(demoResult),
+							IsSuccessed:         true,
+							TaskExecuteDuration: 10000,
+						},
+					},
+				}
+				if err := managerClient.Send(msg); err != nil {
+					log.WithField("endpoint", endpoint).Fatal("response device info failed", "err", err)
+				}
+
+			}
+		}
+	}()
 	for {
 		data, err := managerClient.Recv()
 		if err != nil {
@@ -149,23 +177,7 @@ func withManager(idx int, endpoint string) {
 			return
 		case *omanager.ManagerMessage_PushTaskMessage:
 			log.WithField("endpoint", endpoint).Info("got push task message")
-			demoResult := "a demo task result return"
-			msg := &omanager.WorkerMessage{
-				Message: &omanager.WorkerMessage_SubmitTaskResult{
-					SubmitTaskResult: &omanager.SubmitTaskResult{
-						TaskId:              b.PushTaskMessage.TaskId,
-						ContainerSignature:  make([]byte, 65),
-						MinerSignature:      make([]byte, 65),
-						TaskResultBody:      []byte(demoResult),
-						IsSuccessed:         true,
-						TaskExecuteDuration: 10000,
-					},
-				},
-			}
-			if err := managerClient.Send(msg); err != nil {
-				log.WithField("endpoint", endpoint).Fatal("response device info failed", "err", err)
-				return
-			}
+			taskCh <- b
 
 		case *omanager.ManagerMessage_ProofTaskResult:
 			log.WithField("endpoint", endpoint).Info("got proof task result")

server/workerManager.go

@@ -426,7 +426,6 @@ func (wm *WorkerManager) handleWorkerMsg(worker *Worker) {
 
 	workerCheckTicker := time.NewTicker(time.Second * time.Duration(checkDuration))
 	defer workerCheckTicker.Stop()
-	defer close(worker.quit)
 
 	for {
 		select {