mirror of
				https://github.com/go-gitea/gitea.git
				synced 2025-11-04 13:34:43 +01:00 
			
		
		
		
	Fix #19513 This PR introduce a new db method `InTransaction(context.Context)`, and also builtin check on `db.TxContext` and `db.WithTx`. There is also a new method `db.AutoTx` has been introduced but could be used by other PRs. `WithTx` will always open a new transaction, if a transaction exist in context, return an error. `AutoTx` will try to open a new transaction if no transaction exist in context. That means it will always enter a transaction if there is no error. Co-authored-by: delvh <dev.lh@web.de> Co-authored-by: 6543 <6543@obermui.de>
		
			
				
	
	
		
			44 lines
		
	
	
		
			1.1 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			44 lines
		
	
	
		
			1.1 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
// Copyright 2022 The Gitea Authors. All rights reserved.
 | 
						|
// Use of this source code is governed by a MIT-style
 | 
						|
// license that can be found in the LICENSE file.
 | 
						|
 | 
						|
package repository
 | 
						|
 | 
						|
import (
 | 
						|
	"context"
 | 
						|
 | 
						|
	"code.gitea.io/gitea/models/db"
 | 
						|
	"code.gitea.io/gitea/models/perm"
 | 
						|
	access_model "code.gitea.io/gitea/models/perm/access"
 | 
						|
	repo_model "code.gitea.io/gitea/models/repo"
 | 
						|
	user_model "code.gitea.io/gitea/models/user"
 | 
						|
)
 | 
						|
 | 
						|
func addCollaborator(ctx context.Context, repo *repo_model.Repository, u *user_model.User) error {
 | 
						|
	collaboration := &repo_model.Collaboration{
 | 
						|
		RepoID: repo.ID,
 | 
						|
		UserID: u.ID,
 | 
						|
	}
 | 
						|
 | 
						|
	has, err := db.GetByBean(ctx, collaboration)
 | 
						|
	if err != nil {
 | 
						|
		return err
 | 
						|
	} else if has {
 | 
						|
		return nil
 | 
						|
	}
 | 
						|
	collaboration.Mode = perm.AccessModeWrite
 | 
						|
 | 
						|
	if err = db.Insert(ctx, collaboration); err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
 | 
						|
	return access_model.RecalculateUserAccess(ctx, repo, u.ID)
 | 
						|
}
 | 
						|
 | 
						|
// AddCollaborator adds new collaboration to a repository with default access mode.
 | 
						|
func AddCollaborator(repo *repo_model.Repository, u *user_model.User) error {
 | 
						|
	return db.WithTx(db.DefaultContext, func(ctx context.Context) error {
 | 
						|
		return addCollaborator(ctx, repo, u)
 | 
						|
	})
 | 
						|
}
 |