* Fix session bugs (#16552) * fix deadlog bug * Fix models/issue_stopwatch.go * Update models/issue_stopwatch.go Co-authored-by: zeripath <art27@cantab.net> * fix getLatestCommitStatus Co-authored-by: zeripath <art27@cantab.net>
This commit is contained in:
		
							parent
							
								
									693275455e
								
							
						
					
					
						commit
						e10cd3da1e
					
				
					 3 changed files with 47 additions and 17 deletions
				
			
		|  | @ -160,7 +160,7 @@ func getLatestCommitStatus(e Engine, repoID int64, sha string, listOptions ListO | ||||||
| 	if len(ids) == 0 { | 	if len(ids) == 0 { | ||||||
| 		return statuses, nil | 		return statuses, nil | ||||||
| 	} | 	} | ||||||
| 	return statuses, x.In("id", ids).Find(&statuses) | 	return statuses, e.In("id", ids).Find(&statuses) | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| // FindRepoRecentCommitStatusContexts returns repository's recent commit status contexts
 | // FindRepoRecentCommitStatusContexts returns repository's recent commit status contexts
 | ||||||
|  |  | ||||||
|  | @ -9,6 +9,8 @@ import ( | ||||||
| 	"time" | 	"time" | ||||||
| 
 | 
 | ||||||
| 	"code.gitea.io/gitea/modules/timeutil" | 	"code.gitea.io/gitea/modules/timeutil" | ||||||
|  | 
 | ||||||
|  | 	"xorm.io/xorm" | ||||||
| ) | ) | ||||||
| 
 | 
 | ||||||
| // Stopwatch represents a stopwatch for time tracking.
 | // Stopwatch represents a stopwatch for time tracking.
 | ||||||
|  | @ -61,8 +63,12 @@ func StopwatchExists(userID, issueID int64) bool { | ||||||
| 
 | 
 | ||||||
| // HasUserStopwatch returns true if the user has a stopwatch
 | // HasUserStopwatch returns true if the user has a stopwatch
 | ||||||
| func HasUserStopwatch(userID int64) (exists bool, sw *Stopwatch, err error) { | func HasUserStopwatch(userID int64) (exists bool, sw *Stopwatch, err error) { | ||||||
|  | 	return hasUserStopwatch(x, userID) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func hasUserStopwatch(e Engine, userID int64) (exists bool, sw *Stopwatch, err error) { | ||||||
| 	sw = new(Stopwatch) | 	sw = new(Stopwatch) | ||||||
| 	exists, err = x. | 	exists, err = e. | ||||||
| 		Where("user_id = ?", userID). | 		Where("user_id = ?", userID). | ||||||
| 		Get(sw) | 		Get(sw) | ||||||
| 	return | 	return | ||||||
|  | @ -70,11 +76,23 @@ func HasUserStopwatch(userID int64) (exists bool, sw *Stopwatch, err error) { | ||||||
| 
 | 
 | ||||||
| // CreateOrStopIssueStopwatch will create or remove a stopwatch and will log it into issue's timeline.
 | // CreateOrStopIssueStopwatch will create or remove a stopwatch and will log it into issue's timeline.
 | ||||||
| func CreateOrStopIssueStopwatch(user *User, issue *Issue) error { | func CreateOrStopIssueStopwatch(user *User, issue *Issue) error { | ||||||
| 	sw, exists, err := getStopwatch(x, user.ID, issue.ID) | 	sess := x.NewSession() | ||||||
|  | 	defer sess.Close() | ||||||
|  | 	if err := sess.Begin(); err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	if err := createOrStopIssueStopwatch(sess, user, issue); err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	return sess.Commit() | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func createOrStopIssueStopwatch(e *xorm.Session, user *User, issue *Issue) error { | ||||||
|  | 	sw, exists, err := getStopwatch(e, user.ID, issue.ID) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| 	if err := issue.loadRepo(x); err != nil { | 	if err := issue.loadRepo(e); err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
|  | @ -90,11 +108,11 @@ func CreateOrStopIssueStopwatch(user *User, issue *Issue) error { | ||||||
| 			Time:    timediff, | 			Time:    timediff, | ||||||
| 		} | 		} | ||||||
| 
 | 
 | ||||||
| 		if _, err := x.Insert(tt); err != nil { | 		if _, err := e.Insert(tt); err != nil { | ||||||
| 			return err | 			return err | ||||||
| 		} | 		} | ||||||
| 
 | 
 | ||||||
| 		if _, err := CreateComment(&CreateCommentOptions{ | 		if _, err := createComment(e, &CreateCommentOptions{ | ||||||
| 			Doer:    user, | 			Doer:    user, | ||||||
| 			Issue:   issue, | 			Issue:   issue, | ||||||
| 			Repo:    issue.Repo, | 			Repo:    issue.Repo, | ||||||
|  | @ -104,21 +122,21 @@ func CreateOrStopIssueStopwatch(user *User, issue *Issue) error { | ||||||
| 		}); err != nil { | 		}); err != nil { | ||||||
| 			return err | 			return err | ||||||
| 		} | 		} | ||||||
| 		if _, err := x.Delete(sw); err != nil { | 		if _, err := e.Delete(sw); err != nil { | ||||||
| 			return err | 			return err | ||||||
| 		} | 		} | ||||||
| 	} else { | 	} else { | ||||||
| 		// if another stopwatch is running: stop it
 | 		// if another stopwatch is running: stop it
 | ||||||
| 		exists, sw, err := HasUserStopwatch(user.ID) | 		exists, sw, err := hasUserStopwatch(e, user.ID) | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			return err | 			return err | ||||||
| 		} | 		} | ||||||
| 		if exists { | 		if exists { | ||||||
| 			issue, err := getIssueByID(x, sw.IssueID) | 			issue, err := getIssueByID(e, sw.IssueID) | ||||||
| 			if err != nil { | 			if err != nil { | ||||||
| 				return err | 				return err | ||||||
| 			} | 			} | ||||||
| 			if err := CreateOrStopIssueStopwatch(user, issue); err != nil { | 			if err := createOrStopIssueStopwatch(e, user, issue); err != nil { | ||||||
| 				return err | 				return err | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
|  | @ -129,11 +147,11 @@ func CreateOrStopIssueStopwatch(user *User, issue *Issue) error { | ||||||
| 			IssueID: issue.ID, | 			IssueID: issue.ID, | ||||||
| 		} | 		} | ||||||
| 
 | 
 | ||||||
| 		if _, err := x.Insert(sw); err != nil { | 		if _, err := e.Insert(sw); err != nil { | ||||||
| 			return err | 			return err | ||||||
| 		} | 		} | ||||||
| 
 | 
 | ||||||
| 		if _, err := CreateComment(&CreateCommentOptions{ | 		if _, err := createComment(e, &CreateCommentOptions{ | ||||||
| 			Doer:  user, | 			Doer:  user, | ||||||
| 			Issue: issue, | 			Issue: issue, | ||||||
| 			Repo:  issue.Repo, | 			Repo:  issue.Repo, | ||||||
|  | @ -147,21 +165,33 @@ func CreateOrStopIssueStopwatch(user *User, issue *Issue) error { | ||||||
| 
 | 
 | ||||||
| // CancelStopwatch removes the given stopwatch and logs it into issue's timeline.
 | // CancelStopwatch removes the given stopwatch and logs it into issue's timeline.
 | ||||||
| func CancelStopwatch(user *User, issue *Issue) error { | func CancelStopwatch(user *User, issue *Issue) error { | ||||||
| 	sw, exists, err := getStopwatch(x, user.ID, issue.ID) | 	sess := x.NewSession() | ||||||
|  | 	defer sess.Close() | ||||||
|  | 	if err := sess.Begin(); err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	if err := cancelStopwatch(sess, user, issue); err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	return sess.Commit() | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func cancelStopwatch(e *xorm.Session, user *User, issue *Issue) error { | ||||||
|  | 	sw, exists, err := getStopwatch(e, user.ID, issue.ID) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	if exists { | 	if exists { | ||||||
| 		if _, err := x.Delete(sw); err != nil { | 		if _, err := e.Delete(sw); err != nil { | ||||||
| 			return err | 			return err | ||||||
| 		} | 		} | ||||||
| 
 | 
 | ||||||
| 		if err := issue.loadRepo(x); err != nil { | 		if err := issue.loadRepo(e); err != nil { | ||||||
| 			return err | 			return err | ||||||
| 		} | 		} | ||||||
| 
 | 
 | ||||||
| 		if _, err := CreateComment(&CreateCommentOptions{ | 		if _, err := createComment(e, &CreateCommentOptions{ | ||||||
| 			Doer:  user, | 			Doer:  user, | ||||||
| 			Issue: issue, | 			Issue: issue, | ||||||
| 			Repo:  issue.Repo, | 			Repo:  issue.Repo, | ||||||
|  |  | ||||||
|  | @ -1125,7 +1125,7 @@ func CreateRepository(ctx DBContext, doer, u *User, repo *Repository, overwriteO | ||||||
| 
 | 
 | ||||||
| 	// Give access to all members in teams with access to all repositories.
 | 	// Give access to all members in teams with access to all repositories.
 | ||||||
| 	if u.IsOrganization() { | 	if u.IsOrganization() { | ||||||
| 		if err := u.GetTeams(&SearchTeamOptions{}); err != nil { | 		if err := u.getTeams(ctx.e); err != nil { | ||||||
| 			return fmt.Errorf("GetTeams: %v", err) | 			return fmt.Errorf("GetTeams: %v", err) | ||||||
| 		} | 		} | ||||||
| 		for _, t := range u.Teams { | 		for _, t := range u.Teams { | ||||||
|  |  | ||||||
		Loading…
	
		Reference in a new issue