Skip to content

Commit c99e7e1

Browse files
ethantkoeniglunny
authored andcommitted
Simplify RepositoryList.loadAttributes() (#1211)
1 parent 3803f25 commit c99e7e1

File tree

1 file changed

+5
-12
lines changed

1 file changed

+5
-12
lines changed

models/repo_list.go

Lines changed: 5 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -21,26 +21,19 @@ func (repos RepositoryList) loadAttributes(e Engine) error {
2121
}
2222

2323
// Load owners.
24-
set := make(map[int64]*User)
24+
set := make(map[int64]struct{})
2525
for i := range repos {
26-
set[repos[i].OwnerID] = nil
26+
set[repos[i].OwnerID] = struct{}{}
2727
}
28-
userIDs := make([]int64, 0, len(set))
29-
for userID := range set {
30-
userIDs = append(userIDs, userID)
31-
}
32-
users := make([]*User, 0, len(userIDs))
28+
users := make(map[int64]*User, len(set))
3329
if err := e.
3430
Where("id > 0").
35-
In("id", userIDs).
31+
In("id", keysInt64(set)).
3632
Find(&users); err != nil {
3733
return fmt.Errorf("find users: %v", err)
3834
}
39-
for i := range users {
40-
set[users[i].ID] = users[i]
41-
}
4235
for i := range repos {
43-
repos[i].Owner = set[repos[i].OwnerID]
36+
repos[i].Owner = users[repos[i].OwnerID]
4437
}
4538
return nil
4639
}

0 commit comments

Comments
 (0)