summaryrefslogtreecommitdiffstats
path: root/fastestmirror-repomd-download-speed-timing.patch
blob: 167422f06bbd2c0d235e44795494468395eaf0da (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
diff -ruN yum-utils-1.1.18.old/plugins/fastestmirror/fastestmirror.py yum-utils-1.1.18/plugins/fastestmirror/fastestmirror.py
--- yum-utils-1.1.18.old/plugins/fastestmirror/fastestmirror.py	2008-12-25 16:45:42.000000000 +0800
+++ yum-utils-1.1.18/plugins/fastestmirror/fastestmirror.py	2008-12-25 16:46:30.000000000 +0800
@@ -45,6 +45,7 @@
 import urlparse
 import datetime
 import threading
+import urllib2
 
 from yum.plugins import TYPE_CORE
 
@@ -464,15 +465,19 @@
                 if self.host in ("127.0.0.1", "::1", "localhost"):
                     result = 0
                 else:
-                    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-                    uPort = string.find(self.host,":")
-                    if uPort > 0:
-                        self.port = int(self.host[uPort+1:])
-                        self.host = self.host[:uPort]
+                    headers = {'Pragma': 'no-cache',
+                    'Cache-Control': 'no-cache'   }
+
+                    url = "%s/repodata/repomd.xml" % self.mirror
+                    request = urllib2.Request(url, None, headers)
                     time_before = time.time()
-                    sock.connect((self.host, self.port))
-                    result = time.time() - time_before
-                    sock.close()
+                    usock = urllib2.urlopen(request)
+                    fsize =  len(usock.read())
+                    elapsed = time.time() - time_before
+                    result = (elapsed*10000) / fsize
+                    if verbose:
+                       print "Timed %s (score: %.1f)" % (self.host,fsize/elapsed/1024.0)
+
             self.parent._add_result(self.mirror, self.host, result)
         except:
             if verbose: