Fix all the bugs in issues and pulls on dashboard (#943)

* fix all the bugs in issues and pulls on dashboard

* small fix and refactor

* add method getRepoIDs for IssueList

1 files changed, 59 insertions, 0 deletions

diff --git a/models/issue.go b/models/issue.go
index 3ae34009bf..b160004fd5 100644
--- a/models/issue.go
+++ b/models/issue.go
@@ -1399,3 +1399,62 @@ func updateIssue(e Engine, issue *Issue) error {
 func UpdateIssue(issue *Issue) error {
 	return updateIssue(x, issue)
 }
+
+// IssueList defines a list of issues
+type IssueList []*Issue
+
+func (issues IssueList) getRepoIDs() []int64 {
+	repoIDs := make([]int64, 0, len(issues))
+	for _, issue := range issues {
+		var has bool
+		for _, repoID := range repoIDs {
+			if repoID == issue.RepoID {
+				has = true
+				break
+			}
+		}
+		if !has {
+			repoIDs = append(repoIDs, issue.RepoID)
+		}
+	}
+	return repoIDs
+}
+
+func (issues IssueList) loadRepositories(e Engine) ([]*Repository, error) {
+	if len(issues) == 0 {
+		return nil, nil
+	}
+
+	repoIDs := issues.getRepoIDs()
+	rows, err := e.
+		Where("id > 0").
+		In("id", repoIDs).
+		Rows(new(Repository))
+	if err != nil {
+		return nil, fmt.Errorf("find repository: %v", err)
+	}
+	defer rows.Close()
+
+	repositories := make([]*Repository, 0, len(repoIDs))
+	repoMaps := make(map[int64]*Repository, len(repoIDs))
+	for rows.Next() {
+		var repo Repository
+		err = rows.Scan(&repo)
+		if err != nil {
+			return nil, fmt.Errorf("find repository: %v", err)
+		}
+
+		repositories = append(repositories, &repo)
+		repoMaps[repo.ID] = &repo
+	}
+
+	for _, issue := range issues {
+		issue.Repo = repoMaps[issue.RepoID]
+	}
+	return repositories, nil
+}
+
+// LoadRepositories loads issues' all repositories
+func (issues IssueList) LoadRepositories() ([]*Repository, error) {
+	return issues.loadRepositories(x)
+}