mirror of
https://codeberg.org/forgejo/forgejo.git
synced 2024-09-05 21:03:11 +00:00
* Implementation for calculating language statistics Impement saving code language statistics to database Implement rendering langauge stats Add primary laguage to show in repository list Implement repository stats indexer queue Add indexer test Refactor to use queue module * Do not timeout for queues
141 lines
4.1 KiB
Go
141 lines
4.1 KiB
Go
// Copyright 2019 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 indexer
|
|
|
|
import (
|
|
"code.gitea.io/gitea/models"
|
|
"code.gitea.io/gitea/modules/git"
|
|
code_indexer "code.gitea.io/gitea/modules/indexer/code"
|
|
issue_indexer "code.gitea.io/gitea/modules/indexer/issues"
|
|
stats_indexer "code.gitea.io/gitea/modules/indexer/stats"
|
|
"code.gitea.io/gitea/modules/log"
|
|
"code.gitea.io/gitea/modules/notification/base"
|
|
"code.gitea.io/gitea/modules/repository"
|
|
"code.gitea.io/gitea/modules/setting"
|
|
)
|
|
|
|
type indexerNotifier struct {
|
|
base.NullNotifier
|
|
}
|
|
|
|
var (
|
|
_ base.Notifier = &indexerNotifier{}
|
|
)
|
|
|
|
// NewNotifier create a new indexerNotifier notifier
|
|
func NewNotifier() base.Notifier {
|
|
return &indexerNotifier{}
|
|
}
|
|
|
|
func (r *indexerNotifier) NotifyCreateIssueComment(doer *models.User, repo *models.Repository,
|
|
issue *models.Issue, comment *models.Comment) {
|
|
if comment.Type == models.CommentTypeComment {
|
|
if issue.Comments == nil {
|
|
if err := issue.LoadDiscussComments(); err != nil {
|
|
log.Error("LoadComments failed: %v", err)
|
|
return
|
|
}
|
|
} else {
|
|
issue.Comments = append(issue.Comments, comment)
|
|
}
|
|
|
|
issue_indexer.UpdateIssueIndexer(issue)
|
|
}
|
|
}
|
|
|
|
func (r *indexerNotifier) NotifyNewIssue(issue *models.Issue) {
|
|
issue_indexer.UpdateIssueIndexer(issue)
|
|
}
|
|
|
|
func (r *indexerNotifier) NotifyNewPullRequest(pr *models.PullRequest) {
|
|
issue_indexer.UpdateIssueIndexer(pr.Issue)
|
|
}
|
|
|
|
func (r *indexerNotifier) NotifyUpdateComment(doer *models.User, c *models.Comment, oldContent string) {
|
|
if c.Type == models.CommentTypeComment {
|
|
var found bool
|
|
if c.Issue.Comments != nil {
|
|
for i := 0; i < len(c.Issue.Comments); i++ {
|
|
if c.Issue.Comments[i].ID == c.ID {
|
|
c.Issue.Comments[i] = c
|
|
found = true
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
if !found {
|
|
if err := c.Issue.LoadDiscussComments(); err != nil {
|
|
log.Error("LoadComments failed: %v", err)
|
|
return
|
|
}
|
|
}
|
|
|
|
issue_indexer.UpdateIssueIndexer(c.Issue)
|
|
}
|
|
}
|
|
|
|
func (r *indexerNotifier) NotifyDeleteComment(doer *models.User, comment *models.Comment) {
|
|
if comment.Type == models.CommentTypeComment {
|
|
if err := comment.LoadIssue(); err != nil {
|
|
log.Error("LoadIssue: %v", err)
|
|
return
|
|
}
|
|
|
|
var found bool
|
|
if comment.Issue.Comments != nil {
|
|
for i := 0; i < len(comment.Issue.Comments); i++ {
|
|
if comment.Issue.Comments[i].ID == comment.ID {
|
|
comment.Issue.Comments = append(comment.Issue.Comments[:i], comment.Issue.Comments[i+1:]...)
|
|
found = true
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
if !found {
|
|
if err := comment.Issue.LoadDiscussComments(); err != nil {
|
|
log.Error("LoadComments failed: %v", err)
|
|
return
|
|
}
|
|
}
|
|
// reload comments to delete the old comment
|
|
issue_indexer.UpdateIssueIndexer(comment.Issue)
|
|
}
|
|
}
|
|
|
|
func (r *indexerNotifier) NotifyDeleteRepository(doer *models.User, repo *models.Repository) {
|
|
issue_indexer.DeleteRepoIssueIndexer(repo)
|
|
if setting.Indexer.RepoIndexerEnabled {
|
|
code_indexer.DeleteRepoFromIndexer(repo)
|
|
}
|
|
}
|
|
|
|
func (r *indexerNotifier) NotifyMigrateRepository(doer *models.User, u *models.User, repo *models.Repository) {
|
|
issue_indexer.UpdateRepoIndexer(repo)
|
|
if setting.Indexer.RepoIndexerEnabled && !repo.IsEmpty {
|
|
code_indexer.UpdateRepoIndexer(repo)
|
|
}
|
|
if err := stats_indexer.UpdateRepoIndexer(repo); err != nil {
|
|
log.Error("stats_indexer.UpdateRepoIndexer(%d) failed: %v", repo.ID, err)
|
|
}
|
|
}
|
|
|
|
func (r *indexerNotifier) NotifyPushCommits(pusher *models.User, repo *models.Repository, refName, oldCommitID, newCommitID string, commits *repository.PushCommits) {
|
|
if setting.Indexer.RepoIndexerEnabled && refName == git.BranchPrefix+repo.DefaultBranch {
|
|
code_indexer.UpdateRepoIndexer(repo)
|
|
}
|
|
if err := stats_indexer.UpdateRepoIndexer(repo); err != nil {
|
|
log.Error("stats_indexer.UpdateRepoIndexer(%d) failed: %v", repo.ID, err)
|
|
}
|
|
}
|
|
|
|
func (r *indexerNotifier) NotifyIssueChangeContent(doer *models.User, issue *models.Issue, oldContent string) {
|
|
issue_indexer.UpdateIssueIndexer(issue)
|
|
}
|
|
|
|
func (r *indexerNotifier) NotifyIssueChangeTitle(doer *models.User, issue *models.Issue, oldTitle string) {
|
|
issue_indexer.UpdateIssueIndexer(issue)
|
|
}
|