mirror of
https://github.com/go-gitea/gitea.git
synced 2024-08-25 21:44:20 -04:00
GetActiveStopwatch & HasUserStopwatch is a hot piece of code that is repeatedly called and on examination of the cpu profile for TestGit it represents 0.44 seconds of CPU time. This PR reduces this time to 80ms. --------- Signed-off-by: Andrew Thornton <art27@cantab.net> Co-authored-by: KN4CK3R <admin@oldschoolhack.me> Co-authored-by: Lunny Xiao <xiaolunwen@gmail.com> Co-authored-by: delvh <leon@kske.dev>
114 lines
2.4 KiB
Go
114 lines
2.4 KiB
Go
// Copyright 2017 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package repo
|
|
|
|
import (
|
|
"net/http"
|
|
"strings"
|
|
|
|
"code.gitea.io/gitea/models/db"
|
|
issues_model "code.gitea.io/gitea/models/issues"
|
|
"code.gitea.io/gitea/modules/context"
|
|
"code.gitea.io/gitea/modules/eventsource"
|
|
)
|
|
|
|
// IssueStopwatch creates or stops a stopwatch for the given issue.
|
|
func IssueStopwatch(c *context.Context) {
|
|
issue := GetActionIssue(c)
|
|
if c.Written() {
|
|
return
|
|
}
|
|
|
|
var showSuccessMessage bool
|
|
|
|
if !issues_model.StopwatchExists(c.Doer.ID, issue.ID) {
|
|
showSuccessMessage = true
|
|
}
|
|
|
|
if !c.Repo.CanUseTimetracker(issue, c.Doer) {
|
|
c.NotFound("CanUseTimetracker", nil)
|
|
return
|
|
}
|
|
|
|
if err := issues_model.CreateOrStopIssueStopwatch(c.Doer, issue); err != nil {
|
|
c.ServerError("CreateOrStopIssueStopwatch", err)
|
|
return
|
|
}
|
|
|
|
if showSuccessMessage {
|
|
c.Flash.Success(c.Tr("repo.issues.tracker_auto_close"))
|
|
}
|
|
|
|
url := issue.Link()
|
|
c.Redirect(url, http.StatusSeeOther)
|
|
}
|
|
|
|
// CancelStopwatch cancel the stopwatch
|
|
func CancelStopwatch(c *context.Context) {
|
|
issue := GetActionIssue(c)
|
|
if c.Written() {
|
|
return
|
|
}
|
|
if !c.Repo.CanUseTimetracker(issue, c.Doer) {
|
|
c.NotFound("CanUseTimetracker", nil)
|
|
return
|
|
}
|
|
|
|
if err := issues_model.CancelStopwatch(c.Doer, issue); err != nil {
|
|
c.ServerError("CancelStopwatch", err)
|
|
return
|
|
}
|
|
|
|
stopwatches, err := issues_model.GetUserStopwatches(c.Doer.ID, db.ListOptions{})
|
|
if err != nil {
|
|
c.ServerError("GetUserStopwatches", err)
|
|
return
|
|
}
|
|
if len(stopwatches) == 0 {
|
|
eventsource.GetManager().SendMessage(c.Doer.ID, &eventsource.Event{
|
|
Name: "stopwatches",
|
|
Data: "{}",
|
|
})
|
|
}
|
|
|
|
url := issue.Link()
|
|
c.Redirect(url, http.StatusSeeOther)
|
|
}
|
|
|
|
// GetActiveStopwatch is the middleware that sets .ActiveStopwatch on context
|
|
func GetActiveStopwatch(ctx *context.Context) {
|
|
if strings.HasPrefix(ctx.Req.URL.Path, "/api") {
|
|
return
|
|
}
|
|
|
|
if !ctx.IsSigned {
|
|
return
|
|
}
|
|
|
|
_, sw, issue, err := issues_model.HasUserStopwatch(ctx, ctx.Doer.ID)
|
|
if err != nil {
|
|
ctx.ServerError("HasUserStopwatch", err)
|
|
return
|
|
}
|
|
|
|
if sw == nil || sw.ID == 0 {
|
|
return
|
|
}
|
|
|
|
ctx.Data["ActiveStopwatch"] = StopwatchTmplInfo{
|
|
issue.Link(),
|
|
issue.Repo.FullName(),
|
|
issue.Index,
|
|
sw.Seconds() + 1, // ensure time is never zero in ui
|
|
}
|
|
}
|
|
|
|
// StopwatchTmplInfo is a view on a stopwatch specifically for template rendering
|
|
type StopwatchTmplInfo struct {
|
|
IssueLink string
|
|
RepoSlug string
|
|
IssueIndex int64
|
|
Seconds int64
|
|
}
|