123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543 |
- // Copyright 2018 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 models
-
- // CommentList defines a list of comments
- type CommentList []*Comment
-
- func (comments CommentList) getPosterIDs() []int64 {
- posterIDs := make(map[int64]struct{}, len(comments))
- for _, comment := range comments {
- if _, ok := posterIDs[comment.PosterID]; !ok {
- posterIDs[comment.PosterID] = struct{}{}
- }
- }
- return keysInt64(posterIDs)
- }
-
- func (comments CommentList) loadPosters(e Engine) error {
- if len(comments) == 0 {
- return nil
- }
-
- posterIDs := comments.getPosterIDs()
- posterMaps := make(map[int64]*User, len(posterIDs))
- var left = len(posterIDs)
- for left > 0 {
- var limit = defaultMaxInSize
- if left < limit {
- limit = left
- }
- err := e.
- In("id", posterIDs[:limit]).
- Find(&posterMaps)
- if err != nil {
- return err
- }
- left -= limit
- posterIDs = posterIDs[limit:]
- }
-
- for _, comment := range comments {
- if comment.PosterID <= 0 {
- continue
- }
- var ok bool
- if comment.Poster, ok = posterMaps[comment.PosterID]; !ok {
- comment.Poster = NewGhostUser()
- }
- }
- return nil
- }
-
- func (comments CommentList) getCommentIDs() []int64 {
- var ids = make([]int64, 0, len(comments))
- for _, comment := range comments {
- ids = append(ids, comment.ID)
- }
- return ids
- }
-
- func (comments CommentList) getLabelIDs() []int64 {
- var ids = make(map[int64]struct{}, len(comments))
- for _, comment := range comments {
- if _, ok := ids[comment.LabelID]; !ok {
- ids[comment.LabelID] = struct{}{}
- }
- }
- return keysInt64(ids)
- }
-
- func (comments CommentList) loadLabels(e Engine) error {
- if len(comments) == 0 {
- return nil
- }
-
- var labelIDs = comments.getLabelIDs()
- var commentLabels = make(map[int64]*Label, len(labelIDs))
- var left = len(labelIDs)
- for left > 0 {
- var limit = defaultMaxInSize
- if left < limit {
- limit = left
- }
- rows, err := e.
- In("id", labelIDs[:limit]).
- Rows(new(Label))
- if err != nil {
- return err
- }
-
- for rows.Next() {
- var label Label
- err = rows.Scan(&label)
- if err != nil {
- _ = rows.Close()
- return err
- }
- commentLabels[label.ID] = &label
- }
- _ = rows.Close()
- left -= limit
- labelIDs = labelIDs[limit:]
- }
-
- for _, comment := range comments {
- comment.Label = commentLabels[comment.ID]
- }
- return nil
- }
-
- func (comments CommentList) getMilestoneIDs() []int64 {
- var ids = make(map[int64]struct{}, len(comments))
- for _, comment := range comments {
- if _, ok := ids[comment.MilestoneID]; !ok {
- ids[comment.MilestoneID] = struct{}{}
- }
- }
- return keysInt64(ids)
- }
-
- func (comments CommentList) loadMilestones(e Engine) error {
- if len(comments) == 0 {
- return nil
- }
-
- milestoneIDs := comments.getMilestoneIDs()
- if len(milestoneIDs) == 0 {
- return nil
- }
-
- milestoneMaps := make(map[int64]*Milestone, len(milestoneIDs))
- var left = len(milestoneIDs)
- for left > 0 {
- var limit = defaultMaxInSize
- if left < limit {
- limit = left
- }
- err := e.
- In("id", milestoneIDs[:limit]).
- Find(&milestoneMaps)
- if err != nil {
- return err
- }
- left -= limit
- milestoneIDs = milestoneIDs[limit:]
- }
-
- for _, issue := range comments {
- issue.Milestone = milestoneMaps[issue.MilestoneID]
- }
- return nil
- }
-
- func (comments CommentList) getOldMilestoneIDs() []int64 {
- var ids = make(map[int64]struct{}, len(comments))
- for _, comment := range comments {
- if _, ok := ids[comment.OldMilestoneID]; !ok {
- ids[comment.OldMilestoneID] = struct{}{}
- }
- }
- return keysInt64(ids)
- }
-
- func (comments CommentList) loadOldMilestones(e Engine) error {
- if len(comments) == 0 {
- return nil
- }
-
- milestoneIDs := comments.getOldMilestoneIDs()
- if len(milestoneIDs) == 0 {
- return nil
- }
-
- milestoneMaps := make(map[int64]*Milestone, len(milestoneIDs))
- var left = len(milestoneIDs)
- for left > 0 {
- var limit = defaultMaxInSize
- if left < limit {
- limit = left
- }
- err := e.
- In("id", milestoneIDs[:limit]).
- Find(&milestoneMaps)
- if err != nil {
- return err
- }
- left -= limit
- milestoneIDs = milestoneIDs[limit:]
- }
-
- for _, issue := range comments {
- issue.OldMilestone = milestoneMaps[issue.MilestoneID]
- }
- return nil
- }
-
- func (comments CommentList) getAssigneeIDs() []int64 {
- var ids = make(map[int64]struct{}, len(comments))
- for _, comment := range comments {
- if _, ok := ids[comment.AssigneeID]; !ok {
- ids[comment.AssigneeID] = struct{}{}
- }
- }
- return keysInt64(ids)
- }
-
- func (comments CommentList) loadAssignees(e Engine) error {
- if len(comments) == 0 {
- return nil
- }
-
- var assigneeIDs = comments.getAssigneeIDs()
- var assignees = make(map[int64]*User, len(assigneeIDs))
- var left = len(assigneeIDs)
- for left > 0 {
- var limit = defaultMaxInSize
- if left < limit {
- limit = left
- }
- rows, err := e.
- In("id", assigneeIDs[:limit]).
- Rows(new(User))
- if err != nil {
- return err
- }
-
- for rows.Next() {
- var user User
- err = rows.Scan(&user)
- if err != nil {
- rows.Close()
- return err
- }
-
- assignees[user.ID] = &user
- }
- _ = rows.Close()
-
- left -= limit
- assigneeIDs = assigneeIDs[limit:]
- }
-
- for _, comment := range comments {
- comment.Assignee = assignees[comment.AssigneeID]
- }
- return nil
- }
-
- // getIssueIDs returns all the issue ids on this comment list which issue hasn't been loaded
- func (comments CommentList) getIssueIDs() []int64 {
- var ids = make(map[int64]struct{}, len(comments))
- for _, comment := range comments {
- if comment.Issue != nil {
- continue
- }
- if _, ok := ids[comment.IssueID]; !ok {
- ids[comment.IssueID] = struct{}{}
- }
- }
- return keysInt64(ids)
- }
-
- // Issues returns all the issues of comments
- func (comments CommentList) Issues() IssueList {
- var issues = make(map[int64]*Issue, len(comments))
- for _, comment := range comments {
- if comment.Issue != nil {
- if _, ok := issues[comment.Issue.ID]; !ok {
- issues[comment.Issue.ID] = comment.Issue
- }
- }
- }
-
- var issueList = make([]*Issue, 0, len(issues))
- for _, issue := range issues {
- issueList = append(issueList, issue)
- }
- return issueList
- }
-
- func (comments CommentList) loadIssues(e Engine) error {
- if len(comments) == 0 {
- return nil
- }
-
- var issueIDs = comments.getIssueIDs()
- var issues = make(map[int64]*Issue, len(issueIDs))
- var left = len(issueIDs)
- for left > 0 {
- var limit = defaultMaxInSize
- if left < limit {
- limit = left
- }
- rows, err := e.
- In("id", issueIDs[:limit]).
- Rows(new(Issue))
- if err != nil {
- return err
- }
-
- for rows.Next() {
- var issue Issue
- err = rows.Scan(&issue)
- if err != nil {
- rows.Close()
- return err
- }
-
- issues[issue.ID] = &issue
- }
- _ = rows.Close()
-
- left -= limit
- issueIDs = issueIDs[limit:]
- }
-
- for _, comment := range comments {
- if comment.Issue == nil {
- comment.Issue = issues[comment.IssueID]
- }
- }
- return nil
- }
-
- func (comments CommentList) getDependentIssueIDs() []int64 {
- var ids = make(map[int64]struct{}, len(comments))
- for _, comment := range comments {
- if comment.DependentIssue != nil {
- continue
- }
- if _, ok := ids[comment.DependentIssueID]; !ok {
- ids[comment.DependentIssueID] = struct{}{}
- }
- }
- return keysInt64(ids)
- }
-
- func (comments CommentList) loadDependentIssues(e Engine) error {
- if len(comments) == 0 {
- return nil
- }
-
- var issueIDs = comments.getDependentIssueIDs()
- var issues = make(map[int64]*Issue, len(issueIDs))
- var left = len(issueIDs)
- for left > 0 {
- var limit = defaultMaxInSize
- if left < limit {
- limit = left
- }
- rows, err := e.
- In("id", issueIDs[:limit]).
- Rows(new(Issue))
- if err != nil {
- return err
- }
-
- for rows.Next() {
- var issue Issue
- err = rows.Scan(&issue)
- if err != nil {
- _ = rows.Close()
- return err
- }
-
- issues[issue.ID] = &issue
- }
- _ = rows.Close()
-
- left -= limit
- issueIDs = issueIDs[limit:]
- }
-
- for _, comment := range comments {
- if comment.DependentIssue == nil {
- comment.DependentIssue = issues[comment.DependentIssueID]
- if comment.DependentIssue != nil {
- if err := comment.DependentIssue.loadRepo(e); err != nil {
- return err
- }
- }
- }
- }
- return nil
- }
-
- func (comments CommentList) loadAttachments(e Engine) (err error) {
- if len(comments) == 0 {
- return nil
- }
-
- var attachments = make(map[int64][]*Attachment, len(comments))
- var commentsIDs = comments.getCommentIDs()
- var left = len(commentsIDs)
- for left > 0 {
- var limit = defaultMaxInSize
- if left < limit {
- limit = left
- }
- rows, err := e.Table("attachment").
- Join("INNER", "comment", "comment.id = attachment.comment_id").
- In("comment.id", commentsIDs[:limit]).
- Rows(new(Attachment))
- if err != nil {
- return err
- }
-
- for rows.Next() {
- var attachment Attachment
- err = rows.Scan(&attachment)
- if err != nil {
- _ = rows.Close()
- return err
- }
- attachments[attachment.CommentID] = append(attachments[attachment.CommentID], &attachment)
- }
-
- _ = rows.Close()
- left -= limit
- commentsIDs = commentsIDs[limit:]
- }
-
- for _, comment := range comments {
- comment.Attachments = attachments[comment.ID]
- }
- return nil
- }
-
- func (comments CommentList) getReviewIDs() []int64 {
- var ids = make(map[int64]struct{}, len(comments))
- for _, comment := range comments {
- if _, ok := ids[comment.ReviewID]; !ok {
- ids[comment.ReviewID] = struct{}{}
- }
- }
- return keysInt64(ids)
- }
-
- func (comments CommentList) loadReviews(e Engine) error {
- if len(comments) == 0 {
- return nil
- }
-
- var reviewIDs = comments.getReviewIDs()
- var reviews = make(map[int64]*Review, len(reviewIDs))
- var left = len(reviewIDs)
- for left > 0 {
- var limit = defaultMaxInSize
- if left < limit {
- limit = left
- }
- rows, err := e.
- In("id", reviewIDs[:limit]).
- Rows(new(Review))
- if err != nil {
- return err
- }
-
- for rows.Next() {
- var review Review
- err = rows.Scan(&review)
- if err != nil {
- _ = rows.Close()
- return err
- }
-
- reviews[review.ID] = &review
- }
- _ = rows.Close()
-
- left -= limit
- reviewIDs = reviewIDs[limit:]
- }
-
- for _, comment := range comments {
- comment.Review = reviews[comment.ReviewID]
- }
- return nil
- }
-
- // loadAttributes loads all attributes
- func (comments CommentList) loadAttributes(e Engine) (err error) {
- if err = comments.loadPosters(e); err != nil {
- return
- }
-
- if err = comments.loadLabels(e); err != nil {
- return
- }
-
- if err = comments.loadMilestones(e); err != nil {
- return
- }
-
- if err = comments.loadOldMilestones(e); err != nil {
- return
- }
-
- if err = comments.loadAssignees(e); err != nil {
- return
- }
-
- if err = comments.loadAttachments(e); err != nil {
- return
- }
-
- if err = comments.loadReviews(e); err != nil {
- return
- }
-
- if err = comments.loadIssues(e); err != nil {
- return
- }
-
- if err = comments.loadDependentIssues(e); err != nil {
- return
- }
-
- return nil
- }
-
- // LoadAttributes loads attributes of the comments, except for attachments and
- // comments
- func (comments CommentList) LoadAttributes() error {
- return comments.loadAttributes(x)
- }
-
- // LoadAttachments loads attachments
- func (comments CommentList) LoadAttachments() error {
- return comments.loadAttachments(x)
- }
-
- // LoadPosters loads posters
- func (comments CommentList) LoadPosters() error {
- return comments.loadPosters(x)
- }
-
- // LoadIssues loads issues of comments
- func (comments CommentList) LoadIssues() error {
- return comments.loadIssues(x)
- }
|