💥 v2.0 必须更新面板，新增服务监控

cmd/dashboard/rpc/rpc.go

@@ -2,15 +2,18 @@ package rpc
 
 import (
 	"fmt"
+	"log"
 	"net"
+	"time"
 
 	"google.golang.org/grpc"
 
+	"github.com/naiba/nezha/model"
 	pb "github.com/naiba/nezha/proto"
+	"github.com/naiba/nezha/service/dao"
 	rpcService "github.com/naiba/nezha/service/rpc"
 )
 
-// ServeRPC ...
 func ServeRPC(port uint) {
 	server := grpc.NewServer()
 	pb.RegisterNezhaServiceServer(server, &rpcService.NezhaHandler{
@@ -22,3 +25,34 @@ func ServeRPC(port uint) {
 	}
 	server.Serve(listen)
 }
+
+func DispatchTask(duration time.Duration) {
+	var index uint64 = 0
+	for {
+		var tasks []model.Monitor
+		var hasAliveAgent bool
+		dao.DB.Find(&tasks)
+		dao.ServerLock.RLock()
+		for i := 0; i < len(tasks); i++ {
+			if index >= uint64(len(dao.SortedServerList)) {
+				index = 0
+				if !hasAliveAgent {
+					break
+				}
+				hasAliveAgent = false
+			}
+			if dao.SortedServerList[index].TaskStream == nil {
+				i--
+				index++
+				continue
+			}
+			hasAliveAgent = true
+			log.Println("DispatchTask 确认派发 >>>>>", i, index)
+			dao.SortedServerList[index].TaskStream.Send(tasks[i].PB())
+			log.Println("DispatchTask 确认派发 <<<<<", i, index)
+			index++
+		}
+		dao.ServerLock.RUnlock()
+		time.Sleep(duration)
+	}
+}