worker notice limit config

2026-04-12 21:03:34 +03:30
parent 6aa6022da0
commit a6a1ad51b1
3 changed files with 28 additions and 20 deletions
@@ -18,24 +18,29 @@ import (
 )

 type NoticeWorker struct {
-	Mongo      *mongo.ConnectionManager
-	Logger     logger.Logger
-	Notify     *notification.SDK
-	NoticeRepo notice.Repository
-	TenderRepo tender.TenderRepository
-	GLM        *glm.SDK
-	Scraper    scraper.SDK
+	Mongo           *mongo.ConnectionManager
+	Logger          logger.Logger
+	Notify          *notification.SDK
+	NoticeRepo      notice.Repository
+	TenderRepo      tender.TenderRepository
+	GLM             *glm.SDK
+	Scraper         scraper.SDK
+	ProcessingLimit int
 }

-func NewNoticeWorker(mongo *mongo.ConnectionManager, logger logger.Logger, notify *notification.SDK, noticeRepo notice.Repository, tenderRepo tender.TenderRepository, glmService *glm.SDK, scraper scraper.SDK) *NoticeWorker {
+func NewNoticeWorker(mongo *mongo.ConnectionManager, logger logger.Logger, notify *notification.SDK, noticeRepo notice.Repository, tenderRepo tender.TenderRepository, glmService *glm.SDK, scraper scraper.SDK, processingLimit int) *NoticeWorker {
+	if processingLimit <= 0 {
+		processingLimit = 5 // Default limit
+	}
 	return &NoticeWorker{
-		Mongo:      mongo,
-		Logger:     logger,
-		Notify:     notify,
-		NoticeRepo: noticeRepo,
-		TenderRepo: tenderRepo,
-		GLM:        glmService,
-		Scraper:    scraper,
+		Mongo:           mongo,
+		Logger:          logger,
+		Notify:          notify,
+		NoticeRepo:      noticeRepo,
+		TenderRepo:      tenderRepo,
+		GLM:             glmService,
+		Scraper:         scraper,
+		ProcessingLimit: processingLimit,
 	}
 }

@@ -100,7 +105,7 @@ func (w *NoticeWorker) Run() {
 	limit := 10
 	skip := 0
 	processedCount := 0
-	maxToProcess := 5 // Limit processing to allow scraper worker to run
+	maxToProcess := w.ProcessingLimit

 	for processedCount < maxToProcess {
 		notices, _, err := w.NoticeRepo.GetUnProcessedNotices(context.Background(), limit, skip)