update demoworker

cmd/demoworker/main.go

@@ -26,22 +26,7 @@ func newManagerClient(endpoint string) (omanager.NodeManagerServiceClient, error
 	return omanager.NewNodeManagerServiceClient(client), nil
 }
 
-func main() {
-	flag.Parse()
-	seed, err := newManagerClient(*manager)
-	if err != nil {
-		log.Fatal("connect seed manager failed", "err", err)
-	}
-	nmlist, err := seed.ManagerList(context.Background(), new(omanager.ManagerListRequest))
-	if err != nil {
-		log.Fatal("get manager list failed", "err", err)
-	}
-	log.WithField("count", len(nmlist.Managers)).Info("got manager list")
-	if len(nmlist.Managers) == 0 {
-		log.Fatal("no manager found")
-	}
-	endpoint := nmlist.Managers[0].Endpoint
-
+func withManager(idx int, endpoint string) {
 	client, err := newManagerClient(endpoint)
 	if err != nil {
 		log.WithField("endpoint", endpoint).Fatal("connect manager failed", "err", err)
@@ -53,42 +38,43 @@ func main() {
 		return
 	}
 	minerPubkey := "0x04d4c8fdd1729112459ba4e4a3c66e1cf480dfc1b06e250880234dc0578cb765280114711dd73c86ddec0e264d5a49b49558827d67d0dcdac6207bdfb1264c7461"
+	if idx == 0 {
 
-	reg := &omanager.WorkerMessage{
-		Message: &omanager.WorkerMessage_RegisteMessage{
-			RegisteMessage: &omanager.RegisteMessage{
-				DeviceIp:    "192.168.1.112",
-				MinerPubkey: minerPubkey,
+		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,
-	}
+		}
+		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),
+		devInfo := &omanager.WorkerMessage{
+			Message: &omanager.WorkerMessage_DeviceInfo{
+				DeviceInfo: &omanager.DeviceInfoMessage{
+					Devices:         []*omanager.DeviceInfo{dinfo},
+					DeviceSignature: make([]byte, 0),
+				},
 			},
-		},
-	}
-	managerClient.Send(devInfo)
+		}
+		managerClient.Send(devInfo)
 
-	resource := &omanager.WorkerMessage{
-		Message: &omanager.WorkerMessage_ResourceMap{
-			ResourceMap: &omanager.SubmitResourceMap{
-				ResourceMap: make([]byte, 10),
+		resource := &omanager.WorkerMessage{
+			Message: &omanager.WorkerMessage_ResourceMap{
+				ResourceMap: &omanager.SubmitResourceMap{
+					ResourceMap: make([]byte, 10),
+				},
 			},
-		},
+		}
+		managerClient.Send(resource)
 	}
-	managerClient.Send(resource)
-
 	for {
 		data, err := managerClient.Recv()
 		if err != nil {
@@ -185,6 +171,27 @@ func main() {
 			log.WithField("endpoint", endpoint).Info("got proof task result")
 
 		}
+	}
+}
 
+func main() {
+	flag.Parse()
+	seed, err := newManagerClient(*manager)
+	if err != nil {
+		log.Fatal("connect seed manager failed", "err", err)
+	}
+	nmlist, err := seed.ManagerList(context.Background(), new(omanager.ManagerListRequest))
+	if err != nil {
+		log.Fatal("get manager list failed", "err", err)
+	}
+	log.WithField("count", len(nmlist.Managers)).Info("got manager list")
+	if len(nmlist.Managers) == 0 {
+		log.Fatal("no manager found")
+	}
+	for i, nm := range nmlist.Managers {
+		endpoint := nm.Endpoint
+		go withManager(i, endpoint)
 	}
+	var wait = make(chan struct{})
+	<-wait
 }

nmregister/register.go

@@ -11,7 +11,6 @@ import (
 	"github.com/odysseus/service-registry/registry"
 	"github.com/redis/go-redis/v9"
 	log "github.com/sirupsen/logrus"
-	"os"
 	"sync"
 	"time"
 )
@@ -43,11 +42,6 @@ func (s *NMRegister) ServiceType() common.ServiceType {
 	return common.SERVICE_NODE_MANAGER
 }
 
-func (s *NMRegister) Instance() string {
-	hname, _ := os.Hostname()
-	return fmt.Sprintf("%s:%s", s.conf.RemoteHost, hname)
-}
-
 func (s *NMRegister) Status() string {
 	return "running"
 }