mirror of
				https://gitee.com/gitea/gitea
				synced 2025-11-04 08:30:25 +08:00 
			
		
		
		
	Refactor process package and introduce ProcessManager{} with tests (#75)
* Add a process.Manager singleton with process.GetManager() * Use process.GetManager everywhere * Fix godoc comments for process module * Increment process counter id after locking the mutex
This commit is contained in:
		
				
					committed by
					
						
						Lunny Xiao
					
				
			
			
				
	
			
			
			
						parent
						
							6dd096b7f0
						
					
				
				
					commit
					d1006150fb
				
			@@ -483,8 +483,8 @@ func GetDiffRange(repoPath, beforeCommitID, afterCommitID string, maxLines, maxL
 | 
				
			|||||||
		return nil, fmt.Errorf("Start: %v", err)
 | 
							return nil, fmt.Errorf("Start: %v", err)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	pid := process.Add(fmt.Sprintf("GetDiffRange [repo_path: %s]", repoPath), cmd)
 | 
						pid := process.GetManager().Add(fmt.Sprintf("GetDiffRange [repo_path: %s]", repoPath), cmd)
 | 
				
			||||||
	defer process.Remove(pid)
 | 
						defer process.GetManager().Remove(pid)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	diff, err := ParsePatch(maxLines, maxLineCharacters, maxFiles, stdout)
 | 
						diff, err := ParsePatch(maxLines, maxLineCharacters, maxFiles, stdout)
 | 
				
			||||||
	if err != nil {
 | 
						if err != nil {
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -281,41 +281,41 @@ func (pr *PullRequest) Merge(doer *User, baseGitRepo *git.Repository) (err error
 | 
				
			|||||||
	defer os.RemoveAll(path.Dir(tmpBasePath))
 | 
						defer os.RemoveAll(path.Dir(tmpBasePath))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	var stderr string
 | 
						var stderr string
 | 
				
			||||||
	if _, stderr, err = process.ExecTimeout(5*time.Minute,
 | 
						if _, stderr, err = process.GetManager().ExecTimeout(5*time.Minute,
 | 
				
			||||||
		fmt.Sprintf("PullRequest.Merge (git clone): %s", tmpBasePath),
 | 
							fmt.Sprintf("PullRequest.Merge (git clone): %s", tmpBasePath),
 | 
				
			||||||
		"git", "clone", baseGitRepo.Path, tmpBasePath); err != nil {
 | 
							"git", "clone", baseGitRepo.Path, tmpBasePath); err != nil {
 | 
				
			||||||
		return fmt.Errorf("git clone: %s", stderr)
 | 
							return fmt.Errorf("git clone: %s", stderr)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	// Check out base branch.
 | 
						// Check out base branch.
 | 
				
			||||||
	if _, stderr, err = process.ExecDir(-1, tmpBasePath,
 | 
						if _, stderr, err = process.GetManager().ExecDir(-1, tmpBasePath,
 | 
				
			||||||
		fmt.Sprintf("PullRequest.Merge (git checkout): %s", tmpBasePath),
 | 
							fmt.Sprintf("PullRequest.Merge (git checkout): %s", tmpBasePath),
 | 
				
			||||||
		"git", "checkout", pr.BaseBranch); err != nil {
 | 
							"git", "checkout", pr.BaseBranch); err != nil {
 | 
				
			||||||
		return fmt.Errorf("git checkout: %s", stderr)
 | 
							return fmt.Errorf("git checkout: %s", stderr)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	// Add head repo remote.
 | 
						// Add head repo remote.
 | 
				
			||||||
	if _, stderr, err = process.ExecDir(-1, tmpBasePath,
 | 
						if _, stderr, err = process.GetManager().ExecDir(-1, tmpBasePath,
 | 
				
			||||||
		fmt.Sprintf("PullRequest.Merge (git remote add): %s", tmpBasePath),
 | 
							fmt.Sprintf("PullRequest.Merge (git remote add): %s", tmpBasePath),
 | 
				
			||||||
		"git", "remote", "add", "head_repo", headRepoPath); err != nil {
 | 
							"git", "remote", "add", "head_repo", headRepoPath); err != nil {
 | 
				
			||||||
		return fmt.Errorf("git remote add [%s -> %s]: %s", headRepoPath, tmpBasePath, stderr)
 | 
							return fmt.Errorf("git remote add [%s -> %s]: %s", headRepoPath, tmpBasePath, stderr)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	// Merge commits.
 | 
						// Merge commits.
 | 
				
			||||||
	if _, stderr, err = process.ExecDir(-1, tmpBasePath,
 | 
						if _, stderr, err = process.GetManager().ExecDir(-1, tmpBasePath,
 | 
				
			||||||
		fmt.Sprintf("PullRequest.Merge (git fetch): %s", tmpBasePath),
 | 
							fmt.Sprintf("PullRequest.Merge (git fetch): %s", tmpBasePath),
 | 
				
			||||||
		"git", "fetch", "head_repo"); err != nil {
 | 
							"git", "fetch", "head_repo"); err != nil {
 | 
				
			||||||
		return fmt.Errorf("git fetch [%s -> %s]: %s", headRepoPath, tmpBasePath, stderr)
 | 
							return fmt.Errorf("git fetch [%s -> %s]: %s", headRepoPath, tmpBasePath, stderr)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	if _, stderr, err = process.ExecDir(-1, tmpBasePath,
 | 
						if _, stderr, err = process.GetManager().ExecDir(-1, tmpBasePath,
 | 
				
			||||||
		fmt.Sprintf("PullRequest.Merge (git merge --no-ff --no-commit): %s", tmpBasePath),
 | 
							fmt.Sprintf("PullRequest.Merge (git merge --no-ff --no-commit): %s", tmpBasePath),
 | 
				
			||||||
		"git", "merge", "--no-ff", "--no-commit", "head_repo/"+pr.HeadBranch); err != nil {
 | 
							"git", "merge", "--no-ff", "--no-commit", "head_repo/"+pr.HeadBranch); err != nil {
 | 
				
			||||||
		return fmt.Errorf("git merge --no-ff --no-commit [%s]: %v - %s", tmpBasePath, err, stderr)
 | 
							return fmt.Errorf("git merge --no-ff --no-commit [%s]: %v - %s", tmpBasePath, err, stderr)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	sig := doer.NewGitSig()
 | 
						sig := doer.NewGitSig()
 | 
				
			||||||
	if _, stderr, err = process.ExecDir(-1, tmpBasePath,
 | 
						if _, stderr, err = process.GetManager().ExecDir(-1, tmpBasePath,
 | 
				
			||||||
		fmt.Sprintf("PullRequest.Merge (git merge): %s", tmpBasePath),
 | 
							fmt.Sprintf("PullRequest.Merge (git merge): %s", tmpBasePath),
 | 
				
			||||||
		"git", "commit", fmt.Sprintf("--author='%s <%s>'", sig.Name, sig.Email),
 | 
							"git", "commit", fmt.Sprintf("--author='%s <%s>'", sig.Name, sig.Email),
 | 
				
			||||||
		"-m", fmt.Sprintf("Merge branch '%s' of %s/%s into %s", pr.HeadBranch, pr.HeadUserName, pr.HeadRepo.Name, pr.BaseBranch)); err != nil {
 | 
							"-m", fmt.Sprintf("Merge branch '%s' of %s/%s into %s", pr.HeadBranch, pr.HeadUserName, pr.HeadRepo.Name, pr.BaseBranch)); err != nil {
 | 
				
			||||||
@@ -323,7 +323,7 @@ func (pr *PullRequest) Merge(doer *User, baseGitRepo *git.Repository) (err error
 | 
				
			|||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	// Push back to upstream.
 | 
						// Push back to upstream.
 | 
				
			||||||
	if _, stderr, err = process.ExecDir(-1, tmpBasePath,
 | 
						if _, stderr, err = process.GetManager().ExecDir(-1, tmpBasePath,
 | 
				
			||||||
		fmt.Sprintf("PullRequest.Merge (git push): %s", tmpBasePath),
 | 
							fmt.Sprintf("PullRequest.Merge (git push): %s", tmpBasePath),
 | 
				
			||||||
		"git", "push", baseGitRepo.Path, pr.BaseBranch); err != nil {
 | 
							"git", "push", baseGitRepo.Path, pr.BaseBranch); err != nil {
 | 
				
			||||||
		return fmt.Errorf("git push: %s", stderr)
 | 
							return fmt.Errorf("git push: %s", stderr)
 | 
				
			||||||
@@ -437,14 +437,14 @@ func (pr *PullRequest) testPatch() (err error) {
 | 
				
			|||||||
	defer os.Remove(indexTmpPath)
 | 
						defer os.Remove(indexTmpPath)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	var stderr string
 | 
						var stderr string
 | 
				
			||||||
	_, stderr, err = process.ExecDirEnv(-1, "", fmt.Sprintf("testPatch (git read-tree): %d", pr.BaseRepo.ID),
 | 
						_, stderr, err = process.GetManager().ExecDirEnv(-1, "", fmt.Sprintf("testPatch (git read-tree): %d", pr.BaseRepo.ID),
 | 
				
			||||||
		[]string{"GIT_DIR=" + pr.BaseRepo.RepoPath(), "GIT_INDEX_FILE=" + indexTmpPath},
 | 
							[]string{"GIT_DIR=" + pr.BaseRepo.RepoPath(), "GIT_INDEX_FILE=" + indexTmpPath},
 | 
				
			||||||
		"git", "read-tree", pr.BaseBranch)
 | 
							"git", "read-tree", pr.BaseBranch)
 | 
				
			||||||
	if err != nil {
 | 
						if err != nil {
 | 
				
			||||||
		return fmt.Errorf("git read-tree --index-output=%s %s: %v - %s", indexTmpPath, pr.BaseBranch, err, stderr)
 | 
							return fmt.Errorf("git read-tree --index-output=%s %s: %v - %s", indexTmpPath, pr.BaseBranch, err, stderr)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	_, stderr, err = process.ExecDirEnv(-1, "", fmt.Sprintf("testPatch (git apply --check): %d", pr.BaseRepo.ID),
 | 
						_, stderr, err = process.GetManager().ExecDirEnv(-1, "", fmt.Sprintf("testPatch (git apply --check): %d", pr.BaseRepo.ID),
 | 
				
			||||||
		[]string{"GIT_INDEX_FILE=" + indexTmpPath, "GIT_DIR=" + pr.BaseRepo.RepoPath()},
 | 
							[]string{"GIT_INDEX_FILE=" + indexTmpPath, "GIT_DIR=" + pr.BaseRepo.RepoPath()},
 | 
				
			||||||
		"git", "apply", "--check", "--cached", patchPath)
 | 
							"git", "apply", "--check", "--cached", patchPath)
 | 
				
			||||||
	if err != nil {
 | 
						if err != nil {
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -10,14 +10,11 @@ import (
 | 
				
			|||||||
	"strings"
 | 
						"strings"
 | 
				
			||||||
	"time"
 | 
						"time"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	"github.com/go-xorm/xorm"
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	"code.gitea.io/git"
 | 
						"code.gitea.io/git"
 | 
				
			||||||
 | 
					 | 
				
			||||||
	"code.gitea.io/gitea/modules/process"
 | 
						"code.gitea.io/gitea/modules/process"
 | 
				
			||||||
	"code.gitea.io/gitea/modules/setting"
 | 
						"code.gitea.io/gitea/modules/setting"
 | 
				
			||||||
 | 
					 | 
				
			||||||
	api "code.gitea.io/sdk/gitea"
 | 
						api "code.gitea.io/sdk/gitea"
 | 
				
			||||||
 | 
						"github.com/go-xorm/xorm"
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// Release represents a release of repository.
 | 
					// Release represents a release of repository.
 | 
				
			||||||
@@ -159,7 +156,7 @@ func createTag(gitRepo *git.Repository, rel *Release) error {
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
func addReleaseAttachments(releaseID int64, attachmentUUIDs []string) (err error) {
 | 
					func addReleaseAttachments(releaseID int64, attachmentUUIDs []string) (err error) {
 | 
				
			||||||
	// Check attachments
 | 
						// Check attachments
 | 
				
			||||||
	var attachments = make([]*Attachment,0)
 | 
						var attachments = make([]*Attachment, 0)
 | 
				
			||||||
	for _, uuid := range attachmentUUIDs {
 | 
						for _, uuid := range attachmentUUIDs {
 | 
				
			||||||
		attach, err := getAttachmentByUUID(x, uuid)
 | 
							attach, err := getAttachmentByUUID(x, uuid)
 | 
				
			||||||
		if err != nil {
 | 
							if err != nil {
 | 
				
			||||||
@@ -257,9 +254,10 @@ func GetReleasesByRepoIDAndNames(repoID int64, tagNames []string) (rels []*Relea
 | 
				
			|||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
type releaseMetaSearch struct {
 | 
					type releaseMetaSearch struct {
 | 
				
			||||||
	ID [] int64
 | 
						ID  []int64
 | 
				
			||||||
	Rel [] *Release
 | 
						Rel []*Release
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (s releaseMetaSearch) Len() int {
 | 
					func (s releaseMetaSearch) Len() int {
 | 
				
			||||||
	return len(s.ID)
 | 
						return len(s.ID)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
@@ -272,7 +270,7 @@ func (s releaseMetaSearch) Less(i, j int) bool {
 | 
				
			|||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// GetReleaseAttachments retrieves the attachments for releases
 | 
					// GetReleaseAttachments retrieves the attachments for releases
 | 
				
			||||||
func GetReleaseAttachments(rels ... *Release) (err error){
 | 
					func GetReleaseAttachments(rels ...*Release) (err error) {
 | 
				
			||||||
	if len(rels) == 0 {
 | 
						if len(rels) == 0 {
 | 
				
			||||||
		return
 | 
							return
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
@@ -283,7 +281,7 @@ func GetReleaseAttachments(rels ... *Release) (err error){
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
	// Sort
 | 
						// Sort
 | 
				
			||||||
	var sortedRels = releaseMetaSearch{ID: make([]int64, len(rels)), Rel: make([]*Release, len(rels))}
 | 
						var sortedRels = releaseMetaSearch{ID: make([]int64, len(rels)), Rel: make([]*Release, len(rels))}
 | 
				
			||||||
	var attachments [] *Attachment
 | 
						var attachments []*Attachment
 | 
				
			||||||
	for index, element := range rels {
 | 
						for index, element := range rels {
 | 
				
			||||||
		element.Attachments = []*Attachment{}
 | 
							element.Attachments = []*Attachment{}
 | 
				
			||||||
		sortedRels.ID[index] = element.ID
 | 
							sortedRels.ID[index] = element.ID
 | 
				
			||||||
@@ -375,7 +373,7 @@ func DeleteReleaseByID(id int64, u *User, delTag bool) error {
 | 
				
			|||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	if delTag {
 | 
						if delTag {
 | 
				
			||||||
		_, stderr, err := process.ExecDir(-1, repo.RepoPath(),
 | 
							_, stderr, err := process.GetManager().ExecDir(-1, repo.RepoPath(),
 | 
				
			||||||
			fmt.Sprintf("DeleteReleaseByID (git tag -d): %d", rel.ID),
 | 
								fmt.Sprintf("DeleteReleaseByID (git tag -d): %d", rel.ID),
 | 
				
			||||||
			"git", "tag", "-d", rel.TagName)
 | 
								"git", "tag", "-d", rel.TagName)
 | 
				
			||||||
		if err != nil && !strings.Contains(stderr, "not found") {
 | 
							if err != nil && !strings.Contains(stderr, "not found") {
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -147,10 +147,10 @@ func NewRepoContext() {
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
	// Git requires setting user.name and user.email in order to commit changes.
 | 
						// Git requires setting user.name and user.email in order to commit changes.
 | 
				
			||||||
	for configKey, defaultValue := range map[string]string{"user.name": "Gitea", "user.email": "gitea@fake.local"} {
 | 
						for configKey, defaultValue := range map[string]string{"user.name": "Gitea", "user.email": "gitea@fake.local"} {
 | 
				
			||||||
		if stdout, stderr, err := process.Exec("NewRepoContext(get setting)", "git", "config", "--get", configKey); err != nil || strings.TrimSpace(stdout) == "" {
 | 
							if stdout, stderr, err := process.GetManager().Exec("NewRepoContext(get setting)", "git", "config", "--get", configKey); err != nil || strings.TrimSpace(stdout) == "" {
 | 
				
			||||||
			// ExitError indicates this config is not set
 | 
								// ExitError indicates this config is not set
 | 
				
			||||||
			if _, ok := err.(*exec.ExitError); ok || strings.TrimSpace(stdout) == "" {
 | 
								if _, ok := err.(*exec.ExitError); ok || strings.TrimSpace(stdout) == "" {
 | 
				
			||||||
				if _, stderr, gerr := process.Exec("NewRepoContext(set "+configKey+")", "git", "config", "--global", configKey, defaultValue); gerr != nil {
 | 
									if _, stderr, gerr := process.GetManager().Exec("NewRepoContext(set "+configKey+")", "git", "config", "--global", configKey, defaultValue); gerr != nil {
 | 
				
			||||||
					log.Fatal(4, "Fail to set git %s(%s): %s", configKey, gerr, stderr)
 | 
										log.Fatal(4, "Fail to set git %s(%s): %s", configKey, gerr, stderr)
 | 
				
			||||||
				}
 | 
									}
 | 
				
			||||||
				log.Info("Git config %s set to %s", configKey, defaultValue)
 | 
									log.Info("Git config %s set to %s", configKey, defaultValue)
 | 
				
			||||||
@@ -161,7 +161,7 @@ func NewRepoContext() {
 | 
				
			|||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	// Set git some configurations.
 | 
						// Set git some configurations.
 | 
				
			||||||
	if _, stderr, err := process.Exec("NewRepoContext(git config --global core.quotepath false)",
 | 
						if _, stderr, err := process.GetManager().Exec("NewRepoContext(git config --global core.quotepath false)",
 | 
				
			||||||
		"git", "config", "--global", "core.quotepath", "false"); err != nil {
 | 
							"git", "config", "--global", "core.quotepath", "false"); err != nil {
 | 
				
			||||||
		log.Fatal(4, "Fail to execute 'git config --global core.quotepath false': %s", stderr)
 | 
							log.Fatal(4, "Fail to execute 'git config --global core.quotepath false': %s", stderr)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
@@ -797,20 +797,20 @@ func CleanUpMigrateInfo(repo *Repository) (*Repository, error) {
 | 
				
			|||||||
// initRepoCommit temporarily changes with work directory.
 | 
					// initRepoCommit temporarily changes with work directory.
 | 
				
			||||||
func initRepoCommit(tmpPath string, sig *git.Signature) (err error) {
 | 
					func initRepoCommit(tmpPath string, sig *git.Signature) (err error) {
 | 
				
			||||||
	var stderr string
 | 
						var stderr string
 | 
				
			||||||
	if _, stderr, err = process.ExecDir(-1,
 | 
						if _, stderr, err = process.GetManager().ExecDir(-1,
 | 
				
			||||||
		tmpPath, fmt.Sprintf("initRepoCommit (git add): %s", tmpPath),
 | 
							tmpPath, fmt.Sprintf("initRepoCommit (git add): %s", tmpPath),
 | 
				
			||||||
		"git", "add", "--all"); err != nil {
 | 
							"git", "add", "--all"); err != nil {
 | 
				
			||||||
		return fmt.Errorf("git add: %s", stderr)
 | 
							return fmt.Errorf("git add: %s", stderr)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	if _, stderr, err = process.ExecDir(-1,
 | 
						if _, stderr, err = process.GetManager().ExecDir(-1,
 | 
				
			||||||
		tmpPath, fmt.Sprintf("initRepoCommit (git commit): %s", tmpPath),
 | 
							tmpPath, fmt.Sprintf("initRepoCommit (git commit): %s", tmpPath),
 | 
				
			||||||
		"git", "commit", fmt.Sprintf("--author='%s <%s>'", sig.Name, sig.Email),
 | 
							"git", "commit", fmt.Sprintf("--author='%s <%s>'", sig.Name, sig.Email),
 | 
				
			||||||
		"-m", "Initial commit"); err != nil {
 | 
							"-m", "Initial commit"); err != nil {
 | 
				
			||||||
		return fmt.Errorf("git commit: %s", stderr)
 | 
							return fmt.Errorf("git commit: %s", stderr)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	if _, stderr, err = process.ExecDir(-1,
 | 
						if _, stderr, err = process.GetManager().ExecDir(-1,
 | 
				
			||||||
		tmpPath, fmt.Sprintf("initRepoCommit (git push): %s", tmpPath),
 | 
							tmpPath, fmt.Sprintf("initRepoCommit (git push): %s", tmpPath),
 | 
				
			||||||
		"git", "push", "origin", "master"); err != nil {
 | 
							"git", "push", "origin", "master"); err != nil {
 | 
				
			||||||
		return fmt.Errorf("git push: %s", stderr)
 | 
							return fmt.Errorf("git push: %s", stderr)
 | 
				
			||||||
@@ -856,8 +856,10 @@ func getRepoInitFile(tp, name string) ([]byte, error) {
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
func prepareRepoCommit(repo *Repository, tmpDir, repoPath string, opts CreateRepoOptions) error {
 | 
					func prepareRepoCommit(repo *Repository, tmpDir, repoPath string, opts CreateRepoOptions) error {
 | 
				
			||||||
	// Clone to temporary path and do the init commit.
 | 
						// Clone to temporary path and do the init commit.
 | 
				
			||||||
	_, stderr, err := process.Exec(
 | 
						_, stderr, err := process.GetManager().Exec(
 | 
				
			||||||
		fmt.Sprintf("initRepository(git clone): %s", repoPath), "git", "clone", repoPath, tmpDir)
 | 
							fmt.Sprintf("initRepository(git clone): %s", repoPath),
 | 
				
			||||||
 | 
							"git", "clone", repoPath, tmpDir,
 | 
				
			||||||
 | 
						)
 | 
				
			||||||
	if err != nil {
 | 
						if err != nil {
 | 
				
			||||||
		return fmt.Errorf("git clone: %v - %s", err, stderr)
 | 
							return fmt.Errorf("git clone: %v - %s", err, stderr)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
@@ -1066,7 +1068,7 @@ func CreateRepository(u *User, opts CreateRepoOptions) (_ *Repository, err error
 | 
				
			|||||||
			return nil, fmt.Errorf("initRepository: %v", err)
 | 
								return nil, fmt.Errorf("initRepository: %v", err)
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		_, stderr, err := process.ExecDir(-1,
 | 
							_, stderr, err := process.GetManager().ExecDir(-1,
 | 
				
			||||||
			repoPath, fmt.Sprintf("CreateRepository(git update-server-info): %s", repoPath),
 | 
								repoPath, fmt.Sprintf("CreateRepository(git update-server-info): %s", repoPath),
 | 
				
			||||||
			"git", "update-server-info")
 | 
								"git", "update-server-info")
 | 
				
			||||||
		if err != nil {
 | 
							if err != nil {
 | 
				
			||||||
@@ -1839,7 +1841,7 @@ func GitGcRepos() error {
 | 
				
			|||||||
				if err := repo.GetOwner(); err != nil {
 | 
									if err := repo.GetOwner(); err != nil {
 | 
				
			||||||
					return err
 | 
										return err
 | 
				
			||||||
				}
 | 
									}
 | 
				
			||||||
				_, stderr, err := process.ExecDir(
 | 
									_, stderr, err := process.GetManager().ExecDir(
 | 
				
			||||||
					time.Duration(setting.Git.Timeout.GC)*time.Second,
 | 
										time.Duration(setting.Git.Timeout.GC)*time.Second,
 | 
				
			||||||
					RepoPath(repo.Owner.Name, repo.Name), "Repository garbage collection",
 | 
										RepoPath(repo.Owner.Name, repo.Name), "Repository garbage collection",
 | 
				
			||||||
					"git", args...)
 | 
										"git", args...)
 | 
				
			||||||
@@ -2192,14 +2194,14 @@ func ForkRepository(u *User, oldRepo *Repository, name, desc string) (_ *Reposit
 | 
				
			|||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	repoPath := RepoPath(u.Name, repo.Name)
 | 
						repoPath := RepoPath(u.Name, repo.Name)
 | 
				
			||||||
	_, stderr, err := process.ExecTimeout(10*time.Minute,
 | 
						_, stderr, err := process.GetManager().ExecTimeout(10*time.Minute,
 | 
				
			||||||
		fmt.Sprintf("ForkRepository(git clone): %s/%s", u.Name, repo.Name),
 | 
							fmt.Sprintf("ForkRepository(git clone): %s/%s", u.Name, repo.Name),
 | 
				
			||||||
		"git", "clone", "--bare", oldRepo.RepoPath(), repoPath)
 | 
							"git", "clone", "--bare", oldRepo.RepoPath(), repoPath)
 | 
				
			||||||
	if err != nil {
 | 
						if err != nil {
 | 
				
			||||||
		return nil, fmt.Errorf("git clone: %v", stderr)
 | 
							return nil, fmt.Errorf("git clone: %v", stderr)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	_, stderr, err = process.ExecDir(-1,
 | 
						_, stderr, err = process.GetManager().ExecDir(-1,
 | 
				
			||||||
		repoPath, fmt.Sprintf("ForkRepository(git update-server-info): %s", repoPath),
 | 
							repoPath, fmt.Sprintf("ForkRepository(git update-server-info): %s", repoPath),
 | 
				
			||||||
		"git", "update-server-info")
 | 
							"git", "update-server-info")
 | 
				
			||||||
	if err != nil {
 | 
						if err != nil {
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -212,8 +212,8 @@ func (repo *Repository) GetDiffPreview(branch, treePath, content string) (diff *
 | 
				
			|||||||
		return nil, fmt.Errorf("Start: %v", err)
 | 
							return nil, fmt.Errorf("Start: %v", err)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	pid := process.Add(fmt.Sprintf("GetDiffPreview [repo_path: %s]", repo.RepoPath()), cmd)
 | 
						pid := process.GetManager().Add(fmt.Sprintf("GetDiffPreview [repo_path: %s]", repo.RepoPath()), cmd)
 | 
				
			||||||
	defer process.Remove(pid)
 | 
						defer process.GetManager().Remove(pid)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	diff, err = ParsePatch(setting.Git.MaxGitDiffLines, setting.Git.MaxGitDiffLineCharacters, setting.Git.MaxGitDiffFiles, stdout)
 | 
						diff, err = ParsePatch(setting.Git.MaxGitDiffLines, setting.Git.MaxGitDiffLineCharacters, setting.Git.MaxGitDiffFiles, stdout)
 | 
				
			||||||
	if err != nil {
 | 
						if err != nil {
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -137,7 +137,7 @@ func (m *Mirror) runSync() bool {
 | 
				
			|||||||
		gitArgs = append(gitArgs, "--prune")
 | 
							gitArgs = append(gitArgs, "--prune")
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	if _, stderr, err := process.ExecDir(
 | 
						if _, stderr, err := process.GetManager().ExecDir(
 | 
				
			||||||
		timeout, repoPath, fmt.Sprintf("Mirror.runSync: %s", repoPath),
 | 
							timeout, repoPath, fmt.Sprintf("Mirror.runSync: %s", repoPath),
 | 
				
			||||||
		"git", gitArgs...); err != nil {
 | 
							"git", gitArgs...); err != nil {
 | 
				
			||||||
		desc := fmt.Sprintf("Fail to update mirror repository '%s': %s", repoPath, stderr)
 | 
							desc := fmt.Sprintf("Fail to update mirror repository '%s': %s", repoPath, stderr)
 | 
				
			||||||
@@ -148,7 +148,7 @@ func (m *Mirror) runSync() bool {
 | 
				
			|||||||
		return false
 | 
							return false
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	if m.Repo.HasWiki() {
 | 
						if m.Repo.HasWiki() {
 | 
				
			||||||
		if _, stderr, err := process.ExecDir(
 | 
							if _, stderr, err := process.GetManager().ExecDir(
 | 
				
			||||||
			timeout, wikiPath, fmt.Sprintf("Mirror.runSync: %s", wikiPath),
 | 
								timeout, wikiPath, fmt.Sprintf("Mirror.runSync: %s", wikiPath),
 | 
				
			||||||
			"git", "remote", "update", "--prune"); err != nil {
 | 
								"git", "remote", "update", "--prune"); err != nil {
 | 
				
			||||||
			desc := fmt.Sprintf("Fail to update mirror wiki repository '%s': %s", wikiPath, stderr)
 | 
								desc := fmt.Sprintf("Fail to update mirror wiki repository '%s': %s", wikiPath, stderr)
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -197,7 +197,7 @@ func SSHKeyGenParsePublicKey(key string) (string, int, error) {
 | 
				
			|||||||
	}
 | 
						}
 | 
				
			||||||
	defer os.Remove(tmpName)
 | 
						defer os.Remove(tmpName)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	stdout, stderr, err := process.Exec("SSHKeyGenParsePublicKey", setting.SSH.KeygenPath, "-lf", tmpName)
 | 
						stdout, stderr, err := process.GetManager().Exec("SSHKeyGenParsePublicKey", setting.SSH.KeygenPath, "-lf", tmpName)
 | 
				
			||||||
	if err != nil {
 | 
						if err != nil {
 | 
				
			||||||
		return "", 0, fmt.Errorf("fail to parse public key: %s - %s", err, stderr)
 | 
							return "", 0, fmt.Errorf("fail to parse public key: %s - %s", err, stderr)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
@@ -382,7 +382,7 @@ func addKey(e Engine, key *PublicKey) (err error) {
 | 
				
			|||||||
	if err = ioutil.WriteFile(tmpPath, []byte(key.Content), 0644); err != nil {
 | 
						if err = ioutil.WriteFile(tmpPath, []byte(key.Content), 0644); err != nil {
 | 
				
			||||||
		return err
 | 
							return err
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	stdout, stderr, err := process.Exec("AddPublicKey", "ssh-keygen", "-lf", tmpPath)
 | 
						stdout, stderr, err := process.GetManager().Exec("AddPublicKey", "ssh-keygen", "-lf", tmpPath)
 | 
				
			||||||
	if err != nil {
 | 
						if err != nil {
 | 
				
			||||||
		return fmt.Errorf("'ssh-keygen -lf %s' failed with error '%s': %s", tmpPath, err, stderr)
 | 
							return fmt.Errorf("'ssh-keygen -lf %s' failed with error '%s': %s", tmpPath, err, stderr)
 | 
				
			||||||
	} else if len(stdout) < 2 {
 | 
						} else if len(stdout) < 2 {
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -9,71 +9,108 @@ import (
 | 
				
			|||||||
	"errors"
 | 
						"errors"
 | 
				
			||||||
	"fmt"
 | 
						"fmt"
 | 
				
			||||||
	"os/exec"
 | 
						"os/exec"
 | 
				
			||||||
 | 
						"sync"
 | 
				
			||||||
	"time"
 | 
						"time"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	"code.gitea.io/gitea/modules/log"
 | 
						"code.gitea.io/gitea/modules/log"
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// TODO: This packages still uses a singleton for the Manager.
 | 
				
			||||||
 | 
					// Once there's a decent web framework and dependencies are passed around like they should,
 | 
				
			||||||
 | 
					// then we delete the singleton.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
var (
 | 
					var (
 | 
				
			||||||
	// ErrExecTimeout represent a timeout error
 | 
						// ErrExecTimeout represent a timeout error
 | 
				
			||||||
	ErrExecTimeout = errors.New("Process execution timeout")
 | 
						ErrExecTimeout = errors.New("Process execution timeout")
 | 
				
			||||||
 | 
						manager        *Manager
 | 
				
			||||||
	// DefaultTimeout is the timeout used by Exec* family
 | 
					 | 
				
			||||||
	// of function when timeout parameter is omitted or
 | 
					 | 
				
			||||||
	// passed as -1
 | 
					 | 
				
			||||||
	// NOTE: could be custom in config file for default.
 | 
					 | 
				
			||||||
	DefaultTimeout = 60 * time.Second
 | 
					 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// Process represents a working process inherit from Gogs.
 | 
					// Process represents a working process inherit from Gogs.
 | 
				
			||||||
type Process struct {
 | 
					type Process struct {
 | 
				
			||||||
	Pid         int64 // Process ID, not system one.
 | 
						PID         int64 // Process ID, not system one.
 | 
				
			||||||
	Description string
 | 
						Description string
 | 
				
			||||||
	Start       time.Time
 | 
						Start       time.Time
 | 
				
			||||||
	Cmd         *exec.Cmd
 | 
						Cmd         *exec.Cmd
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// List of existing processes.
 | 
					// Manager knows about all processes and counts PIDs.
 | 
				
			||||||
var (
 | 
					type Manager struct {
 | 
				
			||||||
	curPid    int64 = 1
 | 
						mutex sync.Mutex
 | 
				
			||||||
	Processes []*Process
 | 
					 | 
				
			||||||
)
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
// Add adds a existing process and returns its PID.
 | 
						counter   int64
 | 
				
			||||||
func Add(desc string, cmd *exec.Cmd) int64 {
 | 
						Processes map[int64]*Process
 | 
				
			||||||
	pid := curPid
 | 
					}
 | 
				
			||||||
	Processes = append(Processes, &Process{
 | 
					
 | 
				
			||||||
		Pid:         pid,
 | 
					// GetManager returns a Manager and initializes one as singleton if there's none yet
 | 
				
			||||||
		Description: desc,
 | 
					func GetManager() *Manager {
 | 
				
			||||||
 | 
						if manager == nil {
 | 
				
			||||||
 | 
							manager = &Manager{
 | 
				
			||||||
 | 
								Processes: make(map[int64]*Process),
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return manager
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// Add a process to the ProcessManager and returns its PID.
 | 
				
			||||||
 | 
					func (pm *Manager) Add(description string, cmd *exec.Cmd) int64 {
 | 
				
			||||||
 | 
						pm.mutex.Lock()
 | 
				
			||||||
 | 
						pid := pm.counter + 1
 | 
				
			||||||
 | 
						pm.Processes[pid] = &Process{
 | 
				
			||||||
 | 
							PID:         pid,
 | 
				
			||||||
 | 
							Description: description,
 | 
				
			||||||
		Start:       time.Now(),
 | 
							Start:       time.Now(),
 | 
				
			||||||
		Cmd:         cmd,
 | 
							Cmd:         cmd,
 | 
				
			||||||
	})
 | 
						}
 | 
				
			||||||
	curPid++
 | 
						pm.counter = pid
 | 
				
			||||||
 | 
						pm.mutex.Unlock()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	return pid
 | 
						return pid
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// Remove a process from the ProcessManager.
 | 
				
			||||||
 | 
					func (pm *Manager) Remove(pid int64) {
 | 
				
			||||||
 | 
						pm.mutex.Lock()
 | 
				
			||||||
 | 
						delete(pm.Processes, pid)
 | 
				
			||||||
 | 
						pm.mutex.Unlock()
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// Exec a command and use the default timeout.
 | 
				
			||||||
 | 
					func (pm *Manager) Exec(desc, cmdName string, args ...string) (string, string, error) {
 | 
				
			||||||
 | 
						return pm.ExecDir(-1, "", desc, cmdName, args...)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// ExecTimeout a command and use a specific timeout duration.
 | 
				
			||||||
 | 
					func (pm *Manager) ExecTimeout(timeout time.Duration, desc, cmdName string, args ...string) (string, string, error) {
 | 
				
			||||||
 | 
						return pm.ExecDir(timeout, "", desc, cmdName, args...)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// ExecDir a command and use the default timeout.
 | 
				
			||||||
 | 
					func (pm *Manager) ExecDir(timeout time.Duration, dir, desc, cmdName string, args ...string) (string, string, error) {
 | 
				
			||||||
 | 
						return pm.ExecDirEnv(timeout, dir, desc, nil, cmdName, args...)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// ExecDirEnv runs a command in given path and environment variables, and waits for its completion
 | 
					// ExecDirEnv runs a command in given path and environment variables, and waits for its completion
 | 
				
			||||||
// up to the given timeout (or DefaultTimeout if -1 is given).
 | 
					// up to the given timeout (or DefaultTimeout if -1 is given).
 | 
				
			||||||
// Returns its complete stdout and stderr
 | 
					// Returns its complete stdout and stderr
 | 
				
			||||||
// outputs and an error, if any (including timeout)
 | 
					// outputs and an error, if any (including timeout)
 | 
				
			||||||
func ExecDirEnv(timeout time.Duration, dir, desc string, env []string, cmdName string, args ...string) (string, string, error) {
 | 
					func (pm *Manager) ExecDirEnv(timeout time.Duration, dir, desc string, env []string, cmdName string, args ...string) (string, string, error) {
 | 
				
			||||||
	if timeout == -1 {
 | 
						if timeout == -1 {
 | 
				
			||||||
		timeout = DefaultTimeout
 | 
							timeout = 60 * time.Second
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	bufOut := new(bytes.Buffer)
 | 
						stdOut := new(bytes.Buffer)
 | 
				
			||||||
	bufErr := new(bytes.Buffer)
 | 
						stdErr := new(bytes.Buffer)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	cmd := exec.Command(cmdName, args...)
 | 
						cmd := exec.Command(cmdName, args...)
 | 
				
			||||||
	cmd.Dir = dir
 | 
						cmd.Dir = dir
 | 
				
			||||||
	cmd.Env = env
 | 
						cmd.Env = env
 | 
				
			||||||
	cmd.Stdout = bufOut
 | 
						cmd.Stdout = stdOut
 | 
				
			||||||
	cmd.Stderr = bufErr
 | 
						cmd.Stderr = stdErr
 | 
				
			||||||
	if err := cmd.Start(); err != nil {
 | 
						if err := cmd.Start(); err != nil {
 | 
				
			||||||
		return "", err.Error(), err
 | 
							return "", "", err
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	pid := Add(desc, cmd)
 | 
						pid := pm.Add(desc, cmd)
 | 
				
			||||||
	done := make(chan error)
 | 
						done := make(chan error)
 | 
				
			||||||
	go func() {
 | 
						go func() {
 | 
				
			||||||
		done <- cmd.Wait()
 | 
							done <- cmd.Wait()
 | 
				
			||||||
@@ -82,61 +119,39 @@ func ExecDirEnv(timeout time.Duration, dir, desc string, env []string, cmdName s
 | 
				
			|||||||
	var err error
 | 
						var err error
 | 
				
			||||||
	select {
 | 
						select {
 | 
				
			||||||
	case <-time.After(timeout):
 | 
						case <-time.After(timeout):
 | 
				
			||||||
		if errKill := Kill(pid); errKill != nil {
 | 
							if errKill := pm.Kill(pid); errKill != nil {
 | 
				
			||||||
			log.Error(4, "Exec(%d:%s): %v", pid, desc, errKill)
 | 
								log.Error(4, "exec(%d:%s) failed to kill: %v", pid, desc, errKill)
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
		<-done
 | 
							<-done
 | 
				
			||||||
		return "", ErrExecTimeout.Error(), ErrExecTimeout
 | 
							return "", "", ErrExecTimeout
 | 
				
			||||||
	case err = <-done:
 | 
						case err = <-done:
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	Remove(pid)
 | 
						pm.Remove(pid)
 | 
				
			||||||
	return bufOut.String(), bufErr.String(), err
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
// ExecDir works exactly like ExecDirEnv except no environment variable is provided.
 | 
						if err != nil {
 | 
				
			||||||
func ExecDir(timeout time.Duration, dir, desc, cmdName string, args ...string) (string, string, error) {
 | 
							out := fmt.Errorf("exec(%d:%s) failed: %v stdout: %v stderr: %v", pid, desc, err, stdOut, stdErr)
 | 
				
			||||||
	return ExecDirEnv(timeout, dir, desc, nil, cmdName, args...)
 | 
							return stdOut.String(), stdErr.String(), out
 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
// ExecTimeout runs a command and waits for its completion
 | 
					 | 
				
			||||||
// up to the given timeout (or DefaultTimeout if -1 is given).
 | 
					 | 
				
			||||||
// Returns its complete stdout and stderr
 | 
					 | 
				
			||||||
// outputs and an error, if any (including timeout)
 | 
					 | 
				
			||||||
func ExecTimeout(timeout time.Duration, desc, cmdName string, args ...string) (string, string, error) {
 | 
					 | 
				
			||||||
	return ExecDir(timeout, "", desc, cmdName, args...)
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
// Exec runs a command and waits for its completion
 | 
					 | 
				
			||||||
// up to DefaultTimeout. Returns its complete stdout and stderr
 | 
					 | 
				
			||||||
// outputs and an error, if any (including timeout)
 | 
					 | 
				
			||||||
func Exec(desc, cmdName string, args ...string) (string, string, error) {
 | 
					 | 
				
			||||||
	return ExecDir(-1, "", desc, cmdName, args...)
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
// Remove removes a process from list.
 | 
					 | 
				
			||||||
func Remove(pid int64) {
 | 
					 | 
				
			||||||
	for i, proc := range Processes {
 | 
					 | 
				
			||||||
		if proc.Pid == pid {
 | 
					 | 
				
			||||||
			Processes = append(Processes[:i], Processes[i+1:]...)
 | 
					 | 
				
			||||||
			return
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return stdOut.String(), stdErr.String(), nil
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// Kill kills and removes a process from list.
 | 
					// Kill and remove a process from list.
 | 
				
			||||||
func Kill(pid int64) error {
 | 
					func (pm *Manager) Kill(pid int64) error {
 | 
				
			||||||
	for i, proc := range Processes {
 | 
						if proc, exists := pm.Processes[pid]; exists {
 | 
				
			||||||
		if proc.Pid == pid {
 | 
							pm.mutex.Lock()
 | 
				
			||||||
			if proc.Cmd != nil && proc.Cmd.Process != nil &&
 | 
							if proc.Cmd != nil &&
 | 
				
			||||||
				proc.Cmd.ProcessState != nil && !proc.Cmd.ProcessState.Exited() {
 | 
								proc.Cmd.Process != nil &&
 | 
				
			||||||
				if err := proc.Cmd.Process.Kill(); err != nil {
 | 
								proc.Cmd.ProcessState != nil &&
 | 
				
			||||||
					return fmt.Errorf("fail to kill process(%d/%s): %v", proc.Pid, proc.Description, err)
 | 
								!proc.Cmd.ProcessState.Exited() {
 | 
				
			||||||
				}
 | 
								if err := proc.Cmd.Process.Kill(); err != nil {
 | 
				
			||||||
 | 
									return fmt.Errorf("failed to kill process(%d/%s): %v", pid, proc.Description, err)
 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
			Processes = append(Processes[:i], Processes[i+1:]...)
 | 
					 | 
				
			||||||
			return nil
 | 
					 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
							delete(pm.Processes, pid)
 | 
				
			||||||
 | 
							pm.mutex.Unlock()
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	return nil
 | 
						return nil
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										33
									
								
								modules/process/manager_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										33
									
								
								modules/process/manager_test.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,33 @@
 | 
				
			|||||||
 | 
					package process
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"os/exec"
 | 
				
			||||||
 | 
						"testing"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						"github.com/stretchr/testify/assert"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func TestManager_Add(t *testing.T) {
 | 
				
			||||||
 | 
						pm := Manager{Processes: make(map[int64]*Process)}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						pid := pm.Add("foo", exec.Command("foo"))
 | 
				
			||||||
 | 
						assert.Equal(t, int64(1), pid, "expected to get pid 1 got %d", pid)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						pid = pm.Add("bar", exec.Command("bar"))
 | 
				
			||||||
 | 
						assert.Equal(t, int64(2), pid, "expected to get pid 2 got %d", pid)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func TestManager_Remove(t *testing.T) {
 | 
				
			||||||
 | 
						pm := Manager{Processes: make(map[int64]*Process)}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						pid1 := pm.Add("foo", exec.Command("foo"))
 | 
				
			||||||
 | 
						assert.Equal(t, int64(1), pid1, "expected to get pid 1 got %d", pid1)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						pid2 := pm.Add("bar", exec.Command("bar"))
 | 
				
			||||||
 | 
						assert.Equal(t, int64(2), pid2, "expected to get pid 2 got %d", pid2)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						pm.Remove(pid2)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						_, exists := pm.Processes[pid2]
 | 
				
			||||||
 | 
						assert.False(t, exists, "PID %d is in the list but shouldn't", pid2)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
@@ -246,7 +246,7 @@ func Monitor(ctx *context.Context) {
 | 
				
			|||||||
	ctx.Data["Title"] = ctx.Tr("admin.monitor")
 | 
						ctx.Data["Title"] = ctx.Tr("admin.monitor")
 | 
				
			||||||
	ctx.Data["PageIsAdmin"] = true
 | 
						ctx.Data["PageIsAdmin"] = true
 | 
				
			||||||
	ctx.Data["PageIsAdminMonitor"] = true
 | 
						ctx.Data["PageIsAdminMonitor"] = true
 | 
				
			||||||
	ctx.Data["Processes"] = process.Processes
 | 
						ctx.Data["Processes"] = process.GetManager().Processes
 | 
				
			||||||
	ctx.Data["Entries"] = cron.ListTasks()
 | 
						ctx.Data["Entries"] = cron.ListTasks()
 | 
				
			||||||
	ctx.HTML(200, tplMonitor)
 | 
						ctx.HTML(200, tplMonitor)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
		Reference in New Issue
	
	Block a user