Merge f34e907 into cede7c6

discovery.go

@@ -83,6 +83,8 @@ type Server struct {
 	getMap          sync.Map
 	countMap        map[int]string
 	getLoad         int
+	getMapB         map[string]int
+	mapLock         *sync.Mutex
 }
 
 type httpGetter interface {
@@ -138,6 +140,8 @@ func InitServer() *Server {
 	s.lockNames = make(map[string]string)
 	s.elector = &prodElector{dial: s.DoDial}
 	s.countMap = make(map[int]string)
+	s.getMapB = make(map[string]int)
+	s.mapLock = &sync.Mutex{}
 	return s
 }
 

discoveryv2api.go

@@ -107,8 +107,22 @@ func (s *Server) Get(ctx context.Context, req *pb.GetRequest) (*pb.GetResponse,
 		s.getLoad--
 	}()
 
+	jobName := "unknown"
+	if len(req.GetJob()) > 0 {
+		jobName = req.GetJob()
+	}
+	s.mapLock.Lock()
+	s.getMapB[jobName]++
+	s.mapLock.Unlock()
+
+	defer func() {
+		s.mapLock.Lock()
+		s.getMapB[jobName]--
+		s.mapLock.Unlock()
+	}()
+
 	if s.getLoad > 50 {
-		s.RaiseIssue(ctx, "Overload", fmt.Sprintf("Discover on %v is recording %v get calls", s.Registry, s.getLoad), false)
+		s.RaiseIssue(ctx, "Overload", fmt.Sprintf("Discover on %v is recording %v get calls: %v", s.Registry, s.getLoad, s.getMapB), false)
 	}
 
 	if s.getLoad > 100 {