mirror of
				https://github.com/go-gitea/gitea
				synced 2025-11-03 21:08:25 +00:00 
			
		
		
		
	Backport #31754 by @lunny When opening a repository, it will call `ensureValidRepository` and also `CatFileBatch`. But sometimes these will not be used until repository closed. So it's a waste of CPU to invoke 3 times git command for every open repository. This PR removed all of these from `OpenRepository` but only kept checking whether the folder exists. When a batch is necessary, the necessary functions will be invoked. Co-authored-by: Lunny Xiao <xiaolunwen@gmail.com>
		
			
				
	
	
		
			175 lines
		
	
	
		
			4.3 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			175 lines
		
	
	
		
			4.3 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
// Copyright 2020 The Gitea Authors. All rights reserved.
 | 
						|
// SPDX-License-Identifier: MIT
 | 
						|
 | 
						|
//go:build !gogit
 | 
						|
 | 
						|
package git
 | 
						|
 | 
						|
import (
 | 
						|
	"bufio"
 | 
						|
	"errors"
 | 
						|
	"io"
 | 
						|
	"strings"
 | 
						|
 | 
						|
	"code.gitea.io/gitea/modules/log"
 | 
						|
)
 | 
						|
 | 
						|
// ResolveReference resolves a name to a reference
 | 
						|
func (repo *Repository) ResolveReference(name string) (string, error) {
 | 
						|
	stdout, _, err := NewCommand(repo.Ctx, "show-ref", "--hash").AddDynamicArguments(name).RunStdString(&RunOpts{Dir: repo.Path})
 | 
						|
	if err != nil {
 | 
						|
		if strings.Contains(err.Error(), "not a valid ref") {
 | 
						|
			return "", ErrNotExist{name, ""}
 | 
						|
		}
 | 
						|
		return "", err
 | 
						|
	}
 | 
						|
	stdout = strings.TrimSpace(stdout)
 | 
						|
	if stdout == "" {
 | 
						|
		return "", ErrNotExist{name, ""}
 | 
						|
	}
 | 
						|
 | 
						|
	return stdout, nil
 | 
						|
}
 | 
						|
 | 
						|
// GetRefCommitID returns the last commit ID string of given reference (branch or tag).
 | 
						|
func (repo *Repository) GetRefCommitID(name string) (string, error) {
 | 
						|
	wr, rd, cancel, err := repo.CatFileBatchCheck(repo.Ctx)
 | 
						|
	if err != nil {
 | 
						|
		return "", err
 | 
						|
	}
 | 
						|
	defer cancel()
 | 
						|
	_, err = wr.Write([]byte(name + "\n"))
 | 
						|
	if err != nil {
 | 
						|
		return "", err
 | 
						|
	}
 | 
						|
	shaBs, _, _, err := ReadBatchLine(rd)
 | 
						|
	if IsErrNotExist(err) {
 | 
						|
		return "", ErrNotExist{name, ""}
 | 
						|
	}
 | 
						|
 | 
						|
	return string(shaBs), nil
 | 
						|
}
 | 
						|
 | 
						|
// SetReference sets the commit ID string of given reference (e.g. branch or tag).
 | 
						|
func (repo *Repository) SetReference(name, commitID string) error {
 | 
						|
	_, _, err := NewCommand(repo.Ctx, "update-ref").AddDynamicArguments(name, commitID).RunStdString(&RunOpts{Dir: repo.Path})
 | 
						|
	return err
 | 
						|
}
 | 
						|
 | 
						|
// RemoveReference removes the given reference (e.g. branch or tag).
 | 
						|
func (repo *Repository) RemoveReference(name string) error {
 | 
						|
	_, _, err := NewCommand(repo.Ctx, "update-ref", "--no-deref", "-d").AddDynamicArguments(name).RunStdString(&RunOpts{Dir: repo.Path})
 | 
						|
	return err
 | 
						|
}
 | 
						|
 | 
						|
// IsCommitExist returns true if given commit exists in current repository.
 | 
						|
func (repo *Repository) IsCommitExist(name string) bool {
 | 
						|
	if err := ensureValidGitRepository(repo.Ctx, repo.Path); err != nil {
 | 
						|
		log.Error("IsCommitExist: %v", err)
 | 
						|
		return false
 | 
						|
	}
 | 
						|
	_, _, err := NewCommand(repo.Ctx, "cat-file", "-e").AddDynamicArguments(name).RunStdString(&RunOpts{Dir: repo.Path})
 | 
						|
	return err == nil
 | 
						|
}
 | 
						|
 | 
						|
func (repo *Repository) getCommit(id ObjectID) (*Commit, error) {
 | 
						|
	wr, rd, cancel, err := repo.CatFileBatch(repo.Ctx)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	defer cancel()
 | 
						|
 | 
						|
	_, _ = wr.Write([]byte(id.String() + "\n"))
 | 
						|
 | 
						|
	return repo.getCommitFromBatchReader(rd, id)
 | 
						|
}
 | 
						|
 | 
						|
func (repo *Repository) getCommitFromBatchReader(rd *bufio.Reader, id ObjectID) (*Commit, error) {
 | 
						|
	_, typ, size, err := ReadBatchLine(rd)
 | 
						|
	if err != nil {
 | 
						|
		if errors.Is(err, io.EOF) || IsErrNotExist(err) {
 | 
						|
			return nil, ErrNotExist{ID: id.String()}
 | 
						|
		}
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	switch typ {
 | 
						|
	case "missing":
 | 
						|
		return nil, ErrNotExist{ID: id.String()}
 | 
						|
	case "tag":
 | 
						|
		// then we need to parse the tag
 | 
						|
		// and load the commit
 | 
						|
		data, err := io.ReadAll(io.LimitReader(rd, size))
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
		_, err = rd.Discard(1)
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
		tag, err := parseTagData(id.Type(), data)
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
 | 
						|
		commit, err := tag.Commit(repo)
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
 | 
						|
		return commit, nil
 | 
						|
	case "commit":
 | 
						|
		commit, err := CommitFromReader(repo, id, io.LimitReader(rd, size))
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
		_, err = rd.Discard(1)
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
 | 
						|
		return commit, nil
 | 
						|
	default:
 | 
						|
		log.Debug("Unknown typ: %s", typ)
 | 
						|
		if err := DiscardFull(rd, size+1); err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
		return nil, ErrNotExist{
 | 
						|
			ID: id.String(),
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
// ConvertToGitID returns a GitHash object from a potential ID string
 | 
						|
func (repo *Repository) ConvertToGitID(commitID string) (ObjectID, error) {
 | 
						|
	objectFormat, err := repo.GetObjectFormat()
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	if len(commitID) == objectFormat.FullLength() && objectFormat.IsValid(commitID) {
 | 
						|
		ID, err := NewIDFromString(commitID)
 | 
						|
		if err == nil {
 | 
						|
			return ID, nil
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	wr, rd, cancel, err := repo.CatFileBatchCheck(repo.Ctx)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	defer cancel()
 | 
						|
	_, err = wr.Write([]byte(commitID + "\n"))
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	sha, _, _, err := ReadBatchLine(rd)
 | 
						|
	if err != nil {
 | 
						|
		if IsErrNotExist(err) {
 | 
						|
			return nil, ErrNotExist{commitID, ""}
 | 
						|
		}
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	return MustIDFromString(string(sha)), nil
 | 
						|
}
 |