From dab9fded20bcd8864de68906b435ec72db26dbd3 Mon Sep 17 00:00:00 2001
From: Josip Stojak <Josip.Stojak@infobip.com>
Date: Tue, 11 Jun 2024 16:47:21 +0200
Subject: [PATCH] refactoring to use simple ApiCollector

---
 backend/plugins/bitbucket_server/tasks/pr_collector.go | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/backend/plugins/bitbucket_server/tasks/pr_collector.go b/backend/plugins/bitbucket_server/tasks/pr_collector.go
index 3a72ec8ba0b..d9444258865 100644
--- a/backend/plugins/bitbucket_server/tasks/pr_collector.go
+++ b/backend/plugins/bitbucket_server/tasks/pr_collector.go
@@ -39,12 +39,8 @@ var CollectApiPullRequestsMeta = plugin.SubTaskMeta{
 
 func CollectApiPullRequests(taskCtx plugin.SubTaskContext) errors.Error {
 	rawDataSubTaskArgs, data := CreateRawDataSubTaskArgs(taskCtx, RAW_PULL_REQUEST_TABLE)
-	collectorWithState, err := helper.NewStatefulApiCollector(*rawDataSubTaskArgs)
-	if err != nil {
-		return err
-	}
 
-	err = collectorWithState.InitCollector(helper.ApiCollectorArgs{
+	collector, err := helper.NewApiCollector(helper.ApiCollectorArgs{
 		RawDataSubTaskArgs:    *rawDataSubTaskArgs,
 		ApiClient:             data.ApiClient,
 		PageSize:              100,
@@ -57,5 +53,5 @@ func CollectApiPullRequests(taskCtx plugin.SubTaskContext) errors.Error {
 		return err
 	}
 
-	return collectorWithState.Execute()
+	return collector.Execute()
 }