Merge pull request #2392 from coreos-inc/search-optimization

Optimize repository search by changing our lookup strategy

endpoints/api/search.py

@@ -18,6 +18,12 @@ from util.names import parse_robot_username
 import anunidecode # Don't listen to pylint's lies. This import is required.
 import math
 
+
+ENTITY_SEARCH_SCORE = 1
+TEAM_SEARCH_SCORE = 2
+REPOSITORY_SEARCH_SCORE = 4
+
+
 @resource('/v1/entities/link/<username>')
 @internal_only
 class LinkExternalEntity(ApiResource):
@@ -179,7 +185,7 @@ def search_entity_view(username, entity, get_short_name=None):
     'kind': kind,
     'avatar': avatar_data,
     'name': entity.username,
-    'score': 1,
+    'score': ENTITY_SEARCH_SCORE,
     'href': href
   }
 
@@ -203,7 +209,7 @@ def conduct_team_search(username, query, encountered_teams, results):
       'name': team.name,
       'organization': search_entity_view(username, team.organization),
       'avatar': avatar.get_data_for_team(team),
-      'score': 2,
+      'score': TEAM_SEARCH_SCORE,
       'href': '/organization/' + team.organization.username + '/teams/' + team.name
     })
 
@@ -222,38 +228,23 @@ def conduct_admined_team_search(username, query, encountered_teams, results):
       'name': team.name,
       'organization': search_entity_view(username, team.organization),
       'avatar': avatar.get_data_for_team(team),
-      'score': 2,
+      'score': TEAM_SEARCH_SCORE,
       'href': '/organization/' + team.organization.username + '/teams/' + team.name
     })
 
 
 def conduct_repo_search(username, query, results):
   """ Finds matching repositories. """
-  def can_read(repo):
-    if repo.is_public:
-      return True
-
-    return ReadRepositoryPermission(repo.namespace_user.username, repo.name).can()
-
-  only_public = username is None
-  matching_repos = model.repository.get_sorted_matching_repositories(query, only_public, can_read,
-                                                                     limit=5)
+  matching_repos = model.repository.get_filtered_matching_repositories(query, username, limit=5)
 
   for repo in matching_repos:
-    repo_score = math.log(repo.count or 1, 10) or 1
-
-    # If the repository is under the user's namespace, give it 20% more weight.
-    namespace = repo.namespace_user.username
-    if OrganizationMemberPermission(namespace).can() or namespace == username:
-      repo_score = repo_score * 1.2
-
     results.append({
       'kind': 'repository',
       'namespace': search_entity_view(username, repo.namespace_user),
       'name': repo.name,
       'description': repo.description,
-      'is_public': repo.is_public,
-      'score': repo_score,
+      'is_public': model.repository.is_repository_public(repo),
+      'score': REPOSITORY_SEARCH_SCORE,
       'href': '/repository/' + repo.namespace_user.username + '/' + repo.name
     })