refactoring to use simple ApiCollector (#7609) (#7743)

Co-authored-by: sstojak1 <[email protected]> Co-authored-by: Josip Stojak <[email protected]>
apache · Jul 16, 2024 · 52433a8 · 52433a8
1 parent 5fa4200
commit 52433a8
Showing 1 changed file with 2 additions and 6 deletions.
diff --git 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()
 }