Prefetch the available protocols on the mirror overview page

Otherwise we are doing one query per mirror, which at this point is over
100 separate queries.

Signed-off-by: Dan McGee <dan@archlinux.org>

2 files changed, 8 insertions, 5 deletions

diff --git a/mirrors/models.py b/mirrors/models.py
index 384668b..0179d5b 100644
--- a/mirrors/models.py
+++ b/mirrors/models.py
@@ -33,11 +33,6 @@ class Mirror(models.Model):
     def __unicode__(self):
         return self.name
 
-    def supported_protocols(self):
-        protocols = MirrorProtocol.objects.filter(
-                urls__mirror=self).order_by('protocol').distinct()
-        return sorted(protocols)
-
     def downstream(self):
         return Mirror.objects.filter(upstream=self).order_by('name')
 
diff --git a/mirrors/views.py b/mirrors/views.py
index 5e374b4..2e1e83b 100644
--- a/mirrors/views.py
+++ b/mirrors/views.py
@@ -159,8 +159,16 @@ def find_mirrors_simple(request, protocol):
 
 def mirrors(request):
     mirror_list = Mirror.objects.select_related().order_by('tier', 'country')
+    protos = MirrorUrl.objects.values_list(
+            'mirror_id', 'protocol__protocol').order_by(
+            'mirror__id', 'protocol__protocol')
     if not request.user.is_authenticated():
         mirror_list = mirror_list.filter(public=True, active=True)
+        protos = protos.filter(mirror__public=True, mirror__active=True)
+    protos = dict((k, list(v)) for k, v in groupby(protos, key=itemgetter(0)))
+    for mirror in mirror_list:
+        items = protos.get(mirror.id, [])
+        mirror.protocols = [item[1] for item in items]
     return render(request, 'mirrors/mirrors.html',
             {'mirror_list': mirror_list})