feat: batch size

README.md

@@ -156,6 +156,7 @@ auto_master_merge: false  # Auto-update branch from target branch
 stale_branches_deletion:
   enabled: false  # Clean up stale branches after merge
   days: 90  # Consider branches stale after N days
+  batch_size: 5 # Number of branches can be deleted at once
 ```
 
 #### Example Configuration
@@ -186,6 +187,7 @@ auto_master_merge: true
 stale_branches_deletion:
   enabled: true
   days: 30
+  batch_size: 2
 ```
 
 ## Features

handlers/gitlab/gitlab.go

@@ -227,19 +227,22 @@ func (g *GitlabProvider) GetVar(projectId int, varName string) (string, error) {
 	return secretVar.Value, nil
 }
 
-func (g *GitlabProvider) ListBranches(projectId int) ([]handlers.Branch, error) {
+func (g *GitlabProvider) ListBranches(projectId, size int) ([]handlers.Branch, error) {
 	branches, _, err := g.client.Branches.ListBranches(projectId, &gitlab.ListBranchesOptions{})
 	if err != nil {
 		return nil, err
 	}
 
-	staleBranches := []handlers.Branch{}
+	staleBranches := make([]handlers.Branch, 0, size)
 	for _, b := range branches {
 		if b.Default || b.Protected {
 			continue
 		}
 
 		staleBranches = append(staleBranches, handlers.Branch{Name: b.Name, LastUpdated: *b.Commit.CreatedAt})
+		if len(staleBranches) == size {
+			break
+		}
 	}
 	return staleBranches, nil
 }

handlers/provider.go

@@ -46,7 +46,7 @@ type RequestProvider interface {
 	LeaveComment(projectId, mergeId int, message string) error
 	GetMRInfo(projectId, mergeId int, path string) (*MrInfo, error)
 	UpdateFromMaster(projectId, mergeId int) error
-	ListBranches(projectId int) ([]Branch, error)
+	ListBranches(projectId, size int) ([]Branch, error)
 	DeleteBranch(projectId int, name string) error
 	GetVar(projectId int, varName string) (string, error)
 }
@@ -71,8 +71,9 @@ type Config struct {
 	AutoMasterMerge bool `yaml:"auto_master_merge"`
 
 	StaleBranchesDeletion struct {
-		Enabled bool `yaml:"enabled"`
-		Days    int  `yaml:"days"`
+		Enabled   bool `yaml:"enabled"`
+		Days      int  `yaml:"days"`
+		BatchSize int  `yaml:"batch_size"`
 	} `yaml:"stale_branches_deletion"`
 }
 

handlers/request.go

@@ -77,11 +77,13 @@ func (r *Request) ParseConfig(content string) (*Config, error) {
 		},
 		AutoMasterMerge: false,
 		StaleBranchesDeletion: struct {
-			Enabled bool `yaml:"enabled"`
-			Days    int  `yaml:"days"`
+			Enabled   bool `yaml:"enabled"`
+			Days      int  `yaml:"days"`
+			BatchSize int  `yaml:"batch_size"`
 		}{
-			Enabled: false,
-			Days:    90,
+			Enabled:   false,
+			Days:      90,
+			BatchSize: 5,
 		},
 	}
 

handlers/request_test.go

@@ -41,7 +41,7 @@ func (p *testProvider) Merge(projectId, id int, message string) error {
 	return p.err
 }
 
-func (p *testProvider) ListBranches(projectId int) ([]Branch, error) {
+func (p *testProvider) ListBranches(projectId, size int) ([]Branch, error) {
 	return nil, nil
 }
 

handlers/stalebranches.go

@@ -2,20 +2,28 @@ package handlers
 
 import (
 	"fmt"
+	"sync"
 	"time"
 
 	"github.com/Gasoid/mergebot/logger"
 )
 
+var (
+	cleanStaleBranchesLock sync.Mutex
+)
+
 type Branch struct {
 	Name        string
 	LastUpdated time.Time
 }
 
 func (r *Request) cleanStaleBranches(projectId int) error {
+	cleanStaleBranchesLock.Lock()
+	defer cleanStaleBranchesLock.Unlock()
+
 	logger.Debug("deletion of stale branches has been run")
 
-	candidates, err := r.provider.ListBranches(projectId)
+	candidates, err := r.provider.ListBranches(projectId, r.config.StaleBranchesDeletion.BatchSize)
 	if err != nil {
 		return fmt.Errorf("ListBranches returns error: %w", err)
 	}