Use remote.get_tar_stream in misc.pull_directory

Avoids remote and local temp files, and subsequent read/write
thrashing of local dir.  Since some of those directories are
multiple tens or hundreds of gigabytes, the extra file load
is important.

Signed-off-by: Dan Mick <dan.mick@redhat.com>

diff --git a/teuthology/misc.py b/teuthology/misc.py
index 18ac78fab78bf9410251bf3069e17cb477b3f5ed..dd5629c272cb995c1bea85d8811a4df44b064a1a 100644 (file)
--- a/teuthology/misc.py
+++ b/teuthology/misc.py
@@ -19,7 +19,6 @@ import urlparse
 import yaml
 import json
 import re
-import tempfile
 import pprint
 
 from teuthology import safepath
@@ -778,33 +777,31 @@ def pull_directory(remote, remotedir, localdir):
               remote.shortname, remotedir, localdir)
     if not os.path.exists(localdir):
         os.mkdir(localdir)
-    _, local_tarfile = tempfile.mkstemp(dir=localdir)
-    remote.get_tar(remotedir, local_tarfile, sudo=True)
-    with open(local_tarfile, 'r+') as fb1:
-        tar = tarfile.open(mode='r|gz', fileobj=fb1)
-        while True:
-            ti = tar.next()
-            if ti is None:
-                break
+    r = remote.get_tar_stream(remotedir, sudo=True)
+    tar = tarfile.open(mode='r|gz', fileobj=r.stdout)
+    while True:
+        ti = tar.next()
+        if ti is None:
+            break
 
-            if ti.isdir():
-                # ignore silently; easier to just create leading dirs below
-                pass
-            elif ti.isfile():
-                sub = safepath.munge(ti.name)
-                safepath.makedirs(root=localdir, path=os.path.dirname(sub))
-                tar.makefile(ti, targetpath=os.path.join(localdir, sub))
+        if ti.isdir():
+            # ignore silently; easier to just create leading dirs below
+            # XXX this mean empty dirs are not transferred
+            pass
+        elif ti.isfile():
+            sub = safepath.munge(ti.name)
+            safepath.makedirs(root=localdir, path=os.path.dirname(sub))
+            tar.makefile(ti, targetpath=os.path.join(localdir, sub))
+        else:
+            if ti.isdev():
+                type_ = 'device'
+            elif ti.issym():
+                type_ = 'symlink'
+            elif ti.islnk():
+                type_ = 'hard link'
             else:
-                if ti.isdev():
-                    type_ = 'device'
-                elif ti.issym():
-                    type_ = 'symlink'
-                elif ti.islnk():
-                    type_ = 'hard link'
-                else:
-                    type_ = 'unknown'
-                log.info('Ignoring tar entry: %r type %r', ti.name, type_)
-    os.remove(local_tarfile)
+                type_ = 'unknown'
+            log.info('Ignoring tar entry: %r type %r', ti.name, type_)
 
 
 def pull_directory_tarball(remote, remotedir, localfile):