implement new worker

cmd/demoworker/main.go

@@ -52,6 +52,42 @@ func main() {
 		log.WithField("endpoint", endpoint).Fatal("register worker failed", "err", err)
 		return
 	}
+	minerPubkey := "0x04d4c8fdd1729112459ba4e4a3c66e1cf480dfc1b06e250880234dc0578cb765280114711dd73c86ddec0e264d5a49b49558827d67d0dcdac6207bdfb1264c7461"
+
+	reg := &omanager.WorkerMessage{
+		Message: &omanager.WorkerMessage_RegisteMessage{
+			RegisteMessage: &omanager.RegisteMessage{
+				DeviceIp:    "192.168.1.112",
+				MinerPubkey: minerPubkey,
+			},
+		},
+	}
+	managerClient.Send(reg)
+	dinfo := &omanager.DeviceInfo{
+		DeviceType:  "gpu-0",
+		DeviceModel: "rtx-3080",
+		DeviceParam: "device param",
+		DevicePower: 100,
+	}
+
+	devInfo := &omanager.WorkerMessage{
+		Message: &omanager.WorkerMessage_DeviceInfo{
+			DeviceInfo: &omanager.DeviceInfoMessage{
+				Devices:         []*omanager.DeviceInfo{dinfo},
+				DeviceSignature: make([]byte, 0),
+			},
+		},
+	}
+	managerClient.Send(devInfo)
+
+	resource := &omanager.WorkerMessage{
+		Message: &omanager.WorkerMessage_ResourceMap{
+			ResourceMap: &omanager.SubmitResourceMap{
+				ResourceMap: make([]byte, 10),
+			},
+		},
+	}
+	managerClient.Send(resource)
 
 	for {
 		data, err := managerClient.Recv()
@@ -73,6 +109,21 @@ func main() {
 				log.WithField("endpoint", endpoint).Fatal("response heartbeat failed", "err", err)
 				return
 			}
+		case *omanager.ManagerMessage_NodeInfoRequest:
+			log.WithField("endpoint", endpoint).Info("got node info request")
+			msg := &omanager.WorkerMessage{
+				Message: &omanager.WorkerMessage_NodeInfo{
+					NodeInfo: &omanager.NodeInfoResponse{
+						MinerPubkey:    minerPubkey,
+						BenefitAddress: "0x0Fb196385c8826e3806ebA2cA2cb78B26E08fEEe",
+						DeviceIp:       "192.168.1.112",
+					},
+				},
+			}
+			if err := managerClient.Send(msg); err != nil {
+				log.WithField("endpoint", endpoint).Fatal("response heartbeat failed", "err", err)
+				return
+			}
 
 		case *omanager.ManagerMessage_DeviceUsage:
 			log.WithField("endpoint", endpoint).Info("got device usage")
@@ -116,10 +167,12 @@ func main() {
 			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),
+						TaskId:              b.PushTaskMessage.TaskId,
+						ContainerSignature:  make([]byte, 65),
+						MinerSignature:      make([]byte, 65),
+						TaskResultBody:      []byte(demoResult),
+						IsSuccessed:         true,
+						TaskExecuteDuration: 10000,
 					},
 				},
 			}