mirror of
https://github.com/go-gitea/gitea.git
synced 2026-07-16 13:46:28 +02:00
wxiaoguangandGitHub
82edc3da01
1. Storing "ctx" in a long-living object is wrong 2. Make the commit & tree cacheable (for the future performance optimization) 3. Also fix some bad designs like `// FIXME: bad design, this field can be nil if the commit is from "last commit cache"` ref: * #33893
83 lines
2.3 KiB
Go
83 lines
2.3 KiB
Go
// Copyright 2015 The Gogs Authors. All rights reserved.
|
|
// Copyright 2019 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
//go:build gogit
|
|
|
|
package git
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
|
|
"github.com/go-git/go-git/v5/plumbing"
|
|
"github.com/go-git/go-git/v5/plumbing/object"
|
|
)
|
|
|
|
// Tree represents a flat directory listing.
|
|
type Tree struct {
|
|
TreeCommon
|
|
|
|
resolvedGogitTreeObject *object.Tree
|
|
}
|
|
|
|
func (t *Tree) gogitTreeObject(gitRepo *Repository) (_ *object.Tree, err error) {
|
|
if t.resolvedGogitTreeObject == nil {
|
|
t.resolvedGogitTreeObject, err = gitRepo.gogitRepo.TreeObject(plumbing.Hash(t.ID.RawValue()))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
return t.resolvedGogitTreeObject, nil
|
|
}
|
|
|
|
// ListEntries returns all entries of current tree.
|
|
func (t *Tree) ListEntries(_ context.Context, gitRepo *Repository) (Entries, error) {
|
|
gogitTree, err := t.gogitTreeObject(gitRepo)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
entries := make([]*TreeEntry, len(gogitTree.Entries))
|
|
for i, gogitTreeEntry := range gogitTree.Entries {
|
|
entries[i] = &TreeEntry{
|
|
ID: ParseGogitHash(gogitTreeEntry.Hash),
|
|
ptree: t,
|
|
name: gogitTreeEntry.Name,
|
|
entryMode: gogitFileModeToEntryMode(gogitTreeEntry.Mode),
|
|
}
|
|
}
|
|
|
|
return entries, nil
|
|
}
|
|
|
|
// ListEntriesRecursiveWithSize returns all entries of current tree recursively including all subtrees
|
|
func (t *Tree) ListEntriesRecursiveWithSize(_ context.Context, gitRepo *Repository) (entries Entries, _ error) {
|
|
gogitTree, err := t.gogitTreeObject(gitRepo)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
walker := object.NewTreeWalker(gogitTree, true, nil)
|
|
for {
|
|
fullName, gogitTreeEntry, err := walker.Next()
|
|
if err == io.EOF {
|
|
break
|
|
} else if err != nil {
|
|
return nil, err
|
|
}
|
|
convertedEntry := &TreeEntry{
|
|
ID: ParseGogitHash(gogitTreeEntry.Hash),
|
|
name: fullName, // FIXME: the "name" field is abused, here it is a full path
|
|
ptree: t, // FIXME: this ptree is not right, fortunately it isn't really used
|
|
entryMode: gogitFileModeToEntryMode(gogitTreeEntry.Mode),
|
|
}
|
|
entries = append(entries, convertedEntry)
|
|
}
|
|
return entries, nil
|
|
}
|
|
|
|
// ListEntriesRecursiveFast is the alias of ListEntriesRecursiveWithSize for the gogit version
|
|
func (t *Tree) ListEntriesRecursiveFast(ctx context.Context, gitRepo *Repository) (Entries, error) {
|
|
return t.ListEntriesRecursiveWithSize(ctx, gitRepo)
|
|
}
|