mirror of
				https://codeberg.org/forgejo/forgejo.git
				synced 2025-10-25 03:22:36 +00:00 
			
		
		
		
	Related #14180 Related #25233 Related #22639 Close #19786 Related #12763 This PR will change all the branches retrieve method from reading git data to read database to reduce git read operations. - [x] Sync git branches information into database when push git data - [x] Create a new table `Branch`, merge some columns of `DeletedBranch` into `Branch` table and drop the table `DeletedBranch`. - [x] Read `Branch` table when visit `code` -> `branch` page - [x] Read `Branch` table when list branch names in `code` page dropdown - [x] Read `Branch` table when list git ref compare page - [x] Provide a button in admin page to manually sync all branches. - [x] Sync branches if repository is not empty but database branches are empty when visiting pages with branches list - [x] Use `commit_time desc` as the default FindBranch order by to keep consistent as before and deleted branches will be always at the end. --------- Co-authored-by: Jason Song <i@wolfogre.com>
		
			
				
	
	
		
			95 lines
		
	
	
	
		
			2.5 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			95 lines
		
	
	
	
		
			2.5 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| // Copyright 2022 The Gitea Authors. All rights reserved.
 | |
| // SPDX-License-Identifier: MIT
 | |
| 
 | |
| package git
 | |
| 
 | |
| import (
 | |
| 	"context"
 | |
| 	"sort"
 | |
| 
 | |
| 	"code.gitea.io/gitea/models/db"
 | |
| 	"code.gitea.io/gitea/modules/util"
 | |
| 
 | |
| 	"github.com/gobwas/glob"
 | |
| )
 | |
| 
 | |
| type ProtectedBranchRules []*ProtectedBranch
 | |
| 
 | |
| func (rules ProtectedBranchRules) GetFirstMatched(branchName string) *ProtectedBranch {
 | |
| 	for _, rule := range rules {
 | |
| 		if rule.Match(branchName) {
 | |
| 			return rule
 | |
| 		}
 | |
| 	}
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| func (rules ProtectedBranchRules) sort() {
 | |
| 	sort.Slice(rules, func(i, j int) bool {
 | |
| 		rules[i].loadGlob()
 | |
| 		rules[j].loadGlob()
 | |
| 		if rules[i].isPlainName != rules[j].isPlainName {
 | |
| 			return rules[i].isPlainName // plain name comes first, so plain name means "less"
 | |
| 		}
 | |
| 		return rules[i].CreatedUnix < rules[j].CreatedUnix
 | |
| 	})
 | |
| }
 | |
| 
 | |
| // FindRepoProtectedBranchRules load all repository's protected rules
 | |
| func FindRepoProtectedBranchRules(ctx context.Context, repoID int64) (ProtectedBranchRules, error) {
 | |
| 	var rules ProtectedBranchRules
 | |
| 	err := db.GetEngine(ctx).Where("repo_id = ?", repoID).Asc("created_unix").Find(&rules)
 | |
| 	if err != nil {
 | |
| 		return nil, err
 | |
| 	}
 | |
| 	rules.sort() // to make non-glob rules have higher priority, and for same glob/non-glob rules, first created rules have higher priority
 | |
| 	return rules, nil
 | |
| }
 | |
| 
 | |
| // FindAllMatchedBranches find all matched branches
 | |
| func FindAllMatchedBranches(ctx context.Context, repoID int64, ruleName string) ([]string, error) {
 | |
| 	results := make([]string, 0, 10)
 | |
| 	for page := 1; ; page++ {
 | |
| 		brancheNames, err := FindBranchNames(ctx, FindBranchOptions{
 | |
| 			ListOptions: db.ListOptions{
 | |
| 				PageSize: 100,
 | |
| 				Page:     page,
 | |
| 			},
 | |
| 			RepoID:          repoID,
 | |
| 			IsDeletedBranch: util.OptionalBoolFalse,
 | |
| 		})
 | |
| 		if err != nil {
 | |
| 			return nil, err
 | |
| 		}
 | |
| 		rule := glob.MustCompile(ruleName)
 | |
| 
 | |
| 		for _, branch := range brancheNames {
 | |
| 			if rule.Match(branch) {
 | |
| 				results = append(results, branch)
 | |
| 			}
 | |
| 		}
 | |
| 		if len(brancheNames) < 100 {
 | |
| 			break
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	return results, nil
 | |
| }
 | |
| 
 | |
| // GetFirstMatchProtectedBranchRule returns the first matched rules
 | |
| func GetFirstMatchProtectedBranchRule(ctx context.Context, repoID int64, branchName string) (*ProtectedBranch, error) {
 | |
| 	rules, err := FindRepoProtectedBranchRules(ctx, repoID)
 | |
| 	if err != nil {
 | |
| 		return nil, err
 | |
| 	}
 | |
| 	return rules.GetFirstMatched(branchName), nil
 | |
| }
 | |
| 
 | |
| // IsBranchProtected checks if branch is protected
 | |
| func IsBranchProtected(ctx context.Context, repoID int64, branchName string) (bool, error) {
 | |
| 	rule, err := GetFirstMatchProtectedBranchRule(ctx, repoID, branchName)
 | |
| 	if err != nil {
 | |
| 		return false, err
 | |
| 	}
 | |
| 	return rule != nil, nil
 | |
| }
 |