refactor: reduce sql query in retrieveFeeds (#3547)
parent
d27d720f05
commit
04b3e8cbdc
@ -0,0 +1,98 @@ |
|||||||
|
// 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 |
||||||
|
|
||||||
|
import "fmt" |
||||||
|
|
||||||
|
// ActionList defines a list of actions
|
||||||
|
type ActionList []*Action |
||||||
|
|
||||||
|
func (actions ActionList) getUserIDs() []int64 { |
||||||
|
userIDs := make(map[int64]struct{}, len(actions)) |
||||||
|
for _, action := range actions { |
||||||
|
if _, ok := userIDs[action.ActUserID]; !ok { |
||||||
|
userIDs[action.ActUserID] = struct{}{} |
||||||
|
} |
||||||
|
} |
||||||
|
return keysInt64(userIDs) |
||||||
|
} |
||||||
|
|
||||||
|
func (actions ActionList) loadUsers(e Engine) ([]*User, error) { |
||||||
|
if len(actions) == 0 { |
||||||
|
return nil, nil |
||||||
|
} |
||||||
|
|
||||||
|
userIDs := actions.getUserIDs() |
||||||
|
userMaps := make(map[int64]*User, len(userIDs)) |
||||||
|
err := e. |
||||||
|
In("id", userIDs). |
||||||
|
Find(&userMaps) |
||||||
|
if err != nil { |
||||||
|
return nil, fmt.Errorf("find user: %v", err) |
||||||
|
} |
||||||
|
|
||||||
|
for _, action := range actions { |
||||||
|
action.ActUser = userMaps[action.ActUserID] |
||||||
|
} |
||||||
|
return valuesUser(userMaps), nil |
||||||
|
} |
||||||
|
|
||||||
|
// LoadUsers loads actions' all users
|
||||||
|
func (actions ActionList) LoadUsers() ([]*User, error) { |
||||||
|
return actions.loadUsers(x) |
||||||
|
} |
||||||
|
|
||||||
|
func (actions ActionList) getRepoIDs() []int64 { |
||||||
|
repoIDs := make(map[int64]struct{}, len(actions)) |
||||||
|
for _, action := range actions { |
||||||
|
if _, ok := repoIDs[action.RepoID]; !ok { |
||||||
|
repoIDs[action.RepoID] = struct{}{} |
||||||
|
} |
||||||
|
} |
||||||
|
return keysInt64(repoIDs) |
||||||
|
} |
||||||
|
|
||||||
|
func (actions ActionList) loadRepositories(e Engine) ([]*Repository, error) { |
||||||
|
if len(actions) == 0 { |
||||||
|
return nil, nil |
||||||
|
} |
||||||
|
|
||||||
|
repoIDs := actions.getRepoIDs() |
||||||
|
repoMaps := make(map[int64]*Repository, len(repoIDs)) |
||||||
|
err := e. |
||||||
|
In("id", repoIDs). |
||||||
|
Find(&repoMaps) |
||||||
|
if err != nil { |
||||||
|
return nil, fmt.Errorf("find repository: %v", err) |
||||||
|
} |
||||||
|
|
||||||
|
for _, action := range actions { |
||||||
|
action.Repo = repoMaps[action.RepoID] |
||||||
|
} |
||||||
|
return valuesRepository(repoMaps), nil |
||||||
|
} |
||||||
|
|
||||||
|
// LoadRepositories loads actions' all repositories
|
||||||
|
func (actions ActionList) LoadRepositories() ([]*Repository, error) { |
||||||
|
return actions.loadRepositories(x) |
||||||
|
} |
||||||
|
|
||||||
|
// loadAttributes loads all attributes
|
||||||
|
func (actions ActionList) loadAttributes(e Engine) (err error) { |
||||||
|
if _, err = actions.loadUsers(e); err != nil { |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
if _, err = actions.loadRepositories(e); err != nil { |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
// LoadAttributes loads attributes of the actions
|
||||||
|
func (actions ActionList) LoadAttributes() error { |
||||||
|
return actions.loadAttributes(x) |
||||||
|
} |
Loading…
Reference in new issue