mirror of
				https://github.com/go-gitea/gitea.git
				synced 2025-10-25 17:44:32 +02: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)
 | |
| 	})
 | |
| }
 |