feat: add worker limit for gemini

server/internal/config.go

@@ -21,6 +21,8 @@ const (
 	tokenExpirationInSeconds = uint32(86400)
 
 	WORKER_TIMEOUT_INFINITY = -1
+
+	MAX_GEMINI_WORKER_COUNT = 3
 )
 
 var (

server/internal/http_server.go

@@ -212,6 +212,25 @@ func (s *HttpServer) handlerStart(c *gin.Context) {
 		return
 	}
 
+	// Check if the graphName contains "gemini"
+	if strings.Contains(req.GraphName, "gemini") {
+		// Count existing workers with the same graphName
+		graphNameCount := 0
+		for _, channelName := range workers.Keys() {
+			worker := workers.Get(channelName).(*Worker)
+			if worker.GraphName == req.GraphName {
+				graphNameCount++
+			}
+		}
+
+		// Reject if more than 3 workers are using the same graphName
+		if graphNameCount >= MAX_GEMINI_WORKER_COUNT {
+			slog.Error("handlerStart graphName workers exceed limit", "graphName", req.GraphName, "graphNameCount", graphNameCount, logTag)
+			s.output(c, codeErrWorkersLimit, http.StatusTooManyRequests)
+			return
+		}
+	}
+
 	req.WorkerHttpServerPort = getHttpServerPort()
 	propertyJsonFile, logFile, err := s.processProperty(&req)
 	if err != nil {
@@ -222,6 +241,7 @@ func (s *HttpServer) handlerStart(c *gin.Context) {
 
 	worker := newWorker(req.ChannelName, logFile, s.config.Log2Stdout, propertyJsonFile)
 	worker.HttpServerPort = req.WorkerHttpServerPort
+	worker.GraphName = req.GraphName // Save graphName in the Worker instance
 
 	if req.QuitTimeoutSeconds > 0 {
 		worker.QuitTimeoutSeconds = req.QuitTimeoutSeconds

server/internal/worker.go

@@ -26,6 +26,7 @@ type Worker struct {
 	LogFile            string
 	Log2Stdout         bool
 	PropertyJsonFile   string
+	GraphName          string // New field to store the graphName
 	Pid                int
 	QuitTimeoutSeconds int
 	CreateTs           int64