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,6 +38,7 @@ func main() {
 		return
 	}
 	minerPubkey := "0x04d4c8fdd1729112459ba4e4a3c66e1cf480dfc1b06e250880234dc0578cb765280114711dd73c86ddec0e264d5a49b49558827d67d0dcdac6207bdfb1264c7461"
+	if idx == 0 {
 
 		reg := &omanager.WorkerMessage{
 			Message: &omanager.WorkerMessage_RegisteMessage{
@@ -88,7 +74,7 @@ func main() {
 			},
 		}
 		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"
 }